From c067bcad78233a0466d89c16161c1f60fd0b7937 Mon Sep 17 00:00:00 2001
From: Arjix <53124886+ArjixGamer@users.noreply.github.com>
Date: Sat, 24 Oct 2020 22:39:35 +0300
Subject: [PATCH] fixed a bug

---
 anime_downloader/sites/helpers/selescrape.py | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

diff --git a/anime_downloader/sites/helpers/selescrape.py b/anime_downloader/sites/helpers/selescrape.py
index 2b6dc13..88ba4d0 100644
--- a/anime_downloader/sites/helpers/selescrape.py
+++ b/anime_downloader/sites/helpers/selescrape.py
@@ -195,11 +195,12 @@ def cloudflare_wait(driver):
         if delta >= abort_after:
             logger.error(f'Timeout:\tCouldnt bypass cloudflare. \
             See the screenshot for more info:\t{get_data_dir()}/screenshot.png')
-            break
+            return 1
         title = driver.title
         if not title == "Just a moment...":
             break
-    time.sleep(1)  # This is necessary to make sure everything has loaded fine.
+    time.sleep(2)  # This is necessary to make sure everything has loaded fine.
+    return 0
 
 
 def request(request_type, url, **kwargs):  # Headers not yet supported , headers={}
@@ -221,11 +222,15 @@ def request(request_type, url, **kwargs):  # Headers not yet supported , headers
 
         try:
 
-            cloudflare_wait(driver)
+            exit_code = cloudflare_wait(driver)
             user_agent = driver.execute_script("return navigator.userAgent;")
             cookies = driver.get_cookies()
             text = driver.page_source
             driver.close()
+            if exit_code == 0:
+                pass
+            else:
+                return SeleResponse(url, request_type, None, cookies, user_agent)
 
             seleResponse = SeleResponse(url, request_type, text, cookies, user_agent)
             cache_request(seleResponse)