From 0c7398882bb565abc705e9b99860a17be111612e Mon Sep 17 00:00:00 2001 From: Eduardo M KALINOWSKI Date: Sat, 25 May 2024 08:41:55 -0300 Subject: [PATCH] Detect errors when checking if the user is logged in --- instaloader/instaloadercontext.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/instaloader/instaloadercontext.py b/instaloader/instaloadercontext.py index 8ee87d0..6fb1ef6 100644 --- a/instaloader/instaloadercontext.py +++ b/instaloader/instaloadercontext.py @@ -243,8 +243,12 @@ class InstaloaderContext: def test_login(self) -> Optional[str]: """Not meant to be used directly, use :meth:`Instaloader.test_login`.""" - data = self.graphql_query("d6f4427fbe92d846298cf93df0b937d3", {}) - return data["data"]["user"]["username"] if data["data"]["user"] is not None else None + try: + data = self.graphql_query("d6f4427fbe92d846298cf93df0b937d3", {}) + return data["data"]["user"]["username"] if data["data"]["user"] is not None else None + except (AbortDownloadException, ConnectionException) as err: + self.error(f"Error when checking if logged in: {err}") + return None def login(self, user, passwd): """Not meant to be used directly, use :meth:`Instaloader.login`.