Backup_Repos/leech
1
0
Fork 0
mirror of https://github.com/kemayo/leech synced 2025-12-06 16:33:16 +01:00
Code Issues Projects Releases Wiki Activity

Allow AO3 logins for member-only stories

Browse source
This commit is contained in:
David Lynch 2018-01-19 14:15:43 -06:00
parent f8d494283c
commit 2042f813d0
1 changed files with 21 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

21
sites/ao3.py
View file

@ -4,6 +4,8 @@ import logging
import datetime import datetime
import re import re
import urllib import urllib
import requests_cache
from bs4 import BeautifulSoup
from . import register, Site, Section, Chapter from . import register, Site, Section, Chapter
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
@ -19,6 +21,25 @@ class ArchiveOfOurOwn(Site):
if match: if match:
return match.group(1) + '/' return match.group(1) + '/'
def login(self, login_details):
with requests_cache.disabled():
login = self.session.get('http://archiveofourown.org/login')
soup = BeautifulSoup(login.text, 'html5lib')
form = soup.find(id='new_user_session')
post = {
'user_session[login]': login_details[0],
'user_session[password]': login_details[1],
# standard fields:
'user_session[remember_me]': '1',
'utf8': form.find(attrs={'name': 'utf8'})['value'],
'authenticity_token': form.find(attrs={'name': 'authenticity_token'})['value'],
'commit': 'Log In',
}
# I feel the session *should* handle this cookies bit for me. But
# it doesn't. And I don't know why.
self.session.post('https://archiveofourown.org/user_sessions', data=post, cookies=login.cookies)
logger.info("Logged in as %s", login_details[0])
def extract(self, url): def extract(self, url):
workid = re.match(r'^https?://archiveofourown\.org/works/(\d+)/?.*', url).group(1) workid = re.match(r'^https?://archiveofourown\.org/works/(\d+)/?.*', url).group(1)
return self._extract_work(workid) return self._extract_work(workid)