From 1f57cd6f071517de8cbff501948e27c8645acfb6 Mon Sep 17 00:00:00 2001 From: David Lynch Date: Tue, 14 May 2024 22:05:16 -0500 Subject: [PATCH] Basic success-testing on logins --- sites/ao3.py | 7 +++++-- sites/xenforo.py | 7 +++++-- 2 files changed, 10 insertions(+), 4 deletions(-) diff --git a/sites/ao3.py b/sites/ao3.py index b672e06..6d0e905 100644 --- a/sites/ao3.py +++ b/sites/ao3.py @@ -29,11 +29,14 @@ class ArchiveOfOurOwn(Site): post['user[password]'] = login_details[1] # I feel the session *should* handle this cookies bit for me. But # it doesn't. And I don't know why. - self.session.post( + result = self.session.post( self._join_url(login.url, action), data=post, cookies=login.cookies ) - logger.info("Logged in as %s", login_details[0]) + if result.ok: + logger.info("Logged in as %s", login_details[0]) + else: + logger.error("Failed to log in as %s", login_details[0]) def extract(self, url): workid = re.match(r'^https?://(?:www\.)?archiveofourown\.org/works/(\d+)/?.*', url).group(1) diff --git a/sites/xenforo.py b/sites/xenforo.py index 720cf00..39aad7c 100644 --- a/sites/xenforo.py +++ b/sites/xenforo.py @@ -69,11 +69,14 @@ class XenForo(Site): post['password'] = login_details[1] # I feel the session *should* handle this cookies bit for me. But # it doesn't. And I don't know why. - self.session.post( + result = self.session.post( self._join_url(login.url, action), data=post, cookies=login.cookies ) - logger.info("Logged in as %s", login_details[0]) + if result.ok: + logger.info("Logged in as %s", login_details[0]) + else: + logger.error("Failed to log in as %s", login_details[0]) def extract(self, url): soup = self._soup(url)