request redirect counter
parent
3d1af2fe8b
commit
08af800b52
|
@ -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
|
||||
|
|
Loading…
Reference in New Issue