request redirect counter

master
Laura Liberda 2020-10-15 12:51:57 +02:00
parent 3d1af2fe8b
commit 08af800b52
1 changed files with 7 additions and 3 deletions

View File

@ -14,7 +14,7 @@ class BaseFxcker:
return self.__class__.__name__[:-6]
def request(self, method, url, resource=None,
follow_redirects=True, **kwargs):
follow_redirects=True, redirect_count=0, **kwargs):
kwargs["cookies"] = self.cookie_jar
if not kwargs.get("headers"):
kwargs["headers"] = {}
@ -23,8 +23,11 @@ class BaseFxcker:
if not kwargs.get("allow_redirects"):
kwargs["allow_redirects"] = False
self.log.info("Requesting " +
resource if resource is not None
else (str(method) + " " + str(url)))
(resource if resource is not None
else (str(method) + " " + str(url)))
+ (" (redirect #%d)" % (redirect_count)
if redirect_count > 0
else ""))
sess = requests.Session()
req = sess.request(method, url, **kwargs)
self.cookie_jar.update(sess.cookies)
@ -35,5 +38,6 @@ class BaseFxcker:
return self.request("GET", new_url,
resource=resource,
follow_redirects=follow_redirects,
redirect_count=(redirect_count + 1),
**kwargs)
return req