Add delay as parameter

Anorov · Feb 5, 2018 · 265b65b · 265b65b
1 parent 428e147
commit 265b65b
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/cfscrape/__init__.py b/cfscrape/__init__.py
@@ -30,6 +30,8 @@
 
 class CloudflareScraper(Session):
     def __init__(self, *args, **kwargs):
+        self.delay = kwargs.pop('delay', 5)
+
         super(CloudflareScraper, self).__init__(*args, **kwargs)
 
         if "requests" in self.headers["User-Agent"]:
@@ -51,7 +53,7 @@ def request(self, method, url, *args, **kwargs):
         return resp
 
     def solve_cf_challenge(self, resp, **original_kwargs):
-        sleep(5)  # Cloudflare requires a delay before solving the challenge
+        sleep(self.delay)  # Cloudflare requires a delay before solving the challenge
 
         body = resp.text
         parsed_url = urlparse(resp.url)