From edac5f38cbb5de6fcd5368bccb42c30b13a0a199 Mon Sep 17 00:00:00 2001 From: JustAnotherArchivist Date: Wed, 15 Mar 2023 01:19:53 +0000 Subject: [PATCH] Validate mode parameter on TwitterSearchScraper --- snscrape/modules/twitter.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/snscrape/modules/twitter.py b/snscrape/modules/twitter.py index 00c17ef..593eb42 100644 --- a/snscrape/modules/twitter.py +++ b/snscrape/modules/twitter.py @@ -1583,6 +1583,8 @@ class TwitterSearchScraper(_TwitterAPIScraper): def __init__(self, query, *, cursor = None, mode = TwitterSearchScraperMode.LIVE, top = None, maxEmptyPages = 20, **kwargs): if not query.strip(): raise ValueError('empty query') + if mode not in tuple(TwitterSearchScraperMode): + raise ValueError('invalid mode, must be a TwitterSearchScraperMode') kwargs['maxEmptyPages'] = maxEmptyPages super().__init__(baseUrl = 'https://twitter.com/search?' + urllib.parse.urlencode({'f': 'live', 'lang': 'en', 'q': query, 'src': 'spelling_expansion_revert_click'}), **kwargs) self._query = query # Note: may get replaced by subclasses when using user ID resolution