Basic success-testing on logins

kemayo · May 15, 2024 · 53727fd · 53727fd
1 parent ef9309e
commit 53727fd
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/sites/ao3.py b/sites/ao3.py
@@ -29,11 +29,14 @@ def login(self, login_details):
             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
@@ -73,7 +73,10 @@ def login(self, login_details):
                 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)