From 08af800b525cac887cf56c9f99cf06f2c9d73303 Mon Sep 17 00:00:00 2001 From: Laura Liberda Date: Thu, 15 Oct 2020 12:51:57 +0200 Subject: [PATCH] request redirect counter --- librefi/fxckers/_common.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/librefi/fxckers/_common.py b/librefi/fxckers/_common.py index 44f8fda..a6025c2 100644 --- a/librefi/fxckers/_common.py +++ b/librefi/fxckers/_common.py @@ -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