From: Yossi Gottlieb <yossigo@gmail.com>
Date: Tue, 7 Oct 2014 21:06:01 +0000 (+0300)
Subject: Clean up, support all redirects, fix potential endless 401 loop.
X-Git-Tag: v2.5.0~13^2~1
X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=6ff6f756c034ad0a946c711d0e136cfbd077ab62;p=services%2Fpython-requests.git

Clean up, support all redirects, fix potential endless 401 loop.
---

diff --git a/requests/auth.py b/requests/auth.py
index 41be982..b2b341f 100644
--- a/requests/auth.py
+++ b/requests/auth.py
@@ -17,6 +17,7 @@ from base64 import b64encode
 from .compat import urlparse, str
 from .cookies import extract_cookies_to_jar
 from .utils import parse_dict_header, to_native_string
+from .status_codes import codes
 
 CONTENT_TYPE_FORM_URLENCODED = 'application/x-www-form-urlencoded'
 CONTENT_TYPE_MULTI_PART = 'multipart/form-data'
@@ -150,13 +151,12 @@ class HTTPDigestAuth(AuthBase):
 
         return 'Digest %s' % (base)
 
-    def handle_302(self, r, **kwargs):
+    def handle_redirect(self, r, **kwargs):
         """Reset num_401_calls counter on redirects."""
-        try:
-            delattr(self, 'num_401_calls')
-        except AttributeError:
-            pass
-        return r
+        if r.status_code in (
+                codes.temporary_redirect,
+                codes.permanent_redirect):
+            setattr(self, 'num_401_calls', 1)
 
     def handle_401(self, r, **kwargs):
         """Takes the given response and tries digest-auth, if needed."""
@@ -190,7 +190,7 @@ class HTTPDigestAuth(AuthBase):
 
             return _r
 
-        setattr(self, 'num_401_calls', 1)
+        setattr(self, 'num_401_calls', num_401_calls + 1)
         return r
 
     def __call__(self, r):
@@ -202,5 +202,5 @@ class HTTPDigestAuth(AuthBase):
         except AttributeError:
             pass
         r.register_hook('response', self.handle_401)
-        r.register_hook('response', self.handle_302)
+        r.register_hook('response', self.handle_redirect)
         return r