Update logic for finding consent in url

dpguthrie · Dec 16, 2023 · 399284b · 399284b
1 parent 2047ddc
commit 399284b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/yahooquery/utils/__init__.py b/yahooquery/utils/__init__.py
@@ -1385,7 +1385,7 @@ def setup_session(session: requests.Session, url: str = None):
         response = response.result()
 
     # check for and handle consent page:w
-    if response.url.find("consent"):
+    if response.url.find("consent") >= 0:
         logger.debug(f'Redirected to consent page: "{response.url}"')
 
         soup = BeautifulSoup(response.content, "html.parser")
@@ -1398,7 +1398,7 @@ def setup_session(session: requests.Session, url: str = None):
                 logger.critical(
                     f'Failed to find or extract "{param}" from response. Exception={exc}'
                 )
-                return
+                return session
 
         logger.debug(f"params: {params}")