Merge pull request #503 from ArjixWasTaken/patch-16

added a cache system for SeleScrape and Improve SeleScrape generally
anime-dl · Aug 20, 2021 · 0fc9613 · 0fc9613
2 parents 7b6d9e7 + 4e184b3
commit 0fc9613
Show file tree

Hide file tree

Showing 2 changed files with 214 additions and 110 deletions.
diff --git a/anime_downloader/sites/helpers/request.py b/anime_downloader/sites/helpers/request.py
@@ -46,6 +46,8 @@ def setup_func(url: str,
         cf : bool
             cf if True performs the request through cfscrape.
             For cloudflare protected sites.
+        sel : bool
+            sel if True perfroms the request through selescrape (selenium).
         referer : str
             a url sent as referer in request headers
         '''
@@ -57,6 +59,7 @@ def setup_func(url: str,
                 from selenium import webdriver
                 from anime_downloader.sites.helpers import selescrape
                 sess = selescrape
+                sess.cache = cache
             except ImportError:
                 sess = cf_session
                 logger.warning("This provider may not work correctly because it requires selenium to work.\nIf you want to install it then run:  'pip install selenium' .")
@@ -107,6 +110,8 @@ def get(url: str,
     cf : bool
         cf if True performs the request through cfscrape.
         For cloudflare protected sites.
+    sel : bool
+        sel if True perfroms the request through selescrape (selenium).
     referer : str
         a url sent as referer in request headers
     '''