adjust login methods to a specific style

This commit is contained in:
Mike Fährmann
2017-01-08 17:33:25 +01:00
parent 98d6d131a3
commit 4a8d74973c
3 changed files with 24 additions and 19 deletions

View File

@@ -25,10 +25,7 @@ class NijieExtractor(AsynchronousExtractor):
self.artist_id = ""
def items(self):
self.session.cookies = self.login(
config.interpolate(("extractor", self.category, "username")),
config.interpolate(("extractor", self.category, "password"))
)
self.login()
data = self.get_job_metadata()
images = self.get_image_ids()
yield Message.Version, 1
@@ -62,9 +59,15 @@ class NijieExtractor(AsynchronousExtractor):
"image-id": image_id,
})
@cache(maxage=30*24*60*60, keyarg=1)
def login(self, username, password):
def login(self):
"""Login and obtain session cookie"""
username = config.interpolate(("extractor", "nijie", "username"))
password = config.interpolate(("extractor", "nijie", "password"))
self.session.cookies = self._login_impl(username, password)
@cache(maxage=30*24*60*60, keyarg=1)
def _login_impl(self, username, password):
"""Actual login implementation"""
params = {"email": username, "password": password}
page = self.session.post("https://nijie.info/login_int.php",
data=params).text