From 99050710d78b22b5c605b08d85c1513d9db44323 Mon Sep 17 00:00:00 2001 From: JustAnotherArchivist Date: Fri, 3 Mar 2023 01:34:53 +0000 Subject: [PATCH] Fix AttributeError crashes on resolving user IDs to usernames or vice-versa --- snscrape/modules/twitter.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/snscrape/modules/twitter.py b/snscrape/modules/twitter.py index efa5655..d63f854 100644 --- a/snscrape/modules/twitter.py +++ b/snscrape/modules/twitter.py @@ -1743,6 +1743,8 @@ class TwitterUserScraper(TwitterSearchScraper): def get_items(self): if self._isUserId: # Resolve user ID to username + if self.entity is None: + raise snscrape.base.ScraperException(f'Could not resolve user ID {self._user!r} to username') self._user = self.entity.username self._isUserId = False self._query = f'from:{self._user}' @@ -1772,6 +1774,8 @@ class TwitterProfileScraper(TwitterUserScraper): def get_items(self): if not self._isUserId: + if self.entity is None: + raise snscrape.base.ScraperException(f'Could not resolve username {self._user!r} to ID') userId = self.entity.id else: userId = self._user