From 0d183c41a9ea3afd22e789932d43c7c111eb8ef5 Mon Sep 17 00:00:00 2001 From: Kenneth Reitz Date: Mon, 17 Dec 2012 02:27:07 -0500 Subject: [PATCH] epic redirect generator is epic --- requests/sessions.py | 15 +++++++-------- 1 file changed, 7 insertions(+), 8 deletions(-) diff --git a/requests/sessions.py b/requests/sessions.py index f8f83fb..23e4631 100644 --- a/requests/sessions.py +++ b/requests/sessions.py @@ -68,10 +68,6 @@ class SessionMixin(object): def resolve_redirects(self, resp, req, prefetch=True, timeout=None, verify=True, cert=None): """Receives a Response. Returns a generator of Responses.""" - print resp - print req - print resp.headers['location'] - print resp.status_code # ((resp.status_code is codes.see_other)) while (('location' in resp.headers and resp.status_code in REDIRECT_STATI)): @@ -257,12 +253,15 @@ class Session(SessionMixin): resp = self.send(prep) - if allow_redirects: - gen = self.resolve_redirects(resp, req, prefetch, timeout, verify, cert) + # Redirect generator. + gen = self.resolve_redirects(resp, req, prefetch, timeout, verify, cert) - history = [r for r in gen] + history = [r for r in gen] if allow_redirects else [] - print history + if history: + history.insert(0, resp) + resp = history.pop() + resp.history = history return resp -- 2.34.1