From ddd92476a853fc2a014798b516b9375bab4870f3 Mon Sep 17 00:00:00 2001
From: Miroslav Stampar <miroslav.stampar@gmail.com>
Date: Thu, 15 Mar 2012 15:58:25 +0000
Subject: [PATCH] minor fix

---
 lib/request/connect.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/lib/request/connect.py b/lib/request/connect.py
index d0301da3e..7d5aec74b 100644
--- a/lib/request/connect.py
+++ b/lib/request/connect.py
@@ -165,6 +165,7 @@ class Connect:
         # url splitted with space char while urlencoding it in the later phase
         url = url.replace(" ", "%20")
 
+        code = None
         page = None
         requestMsg = u"HTTP request [#%d]:\n%s " % (threadData.lastRequestUID, method or (HTTPMETHOD.POST if post else HTTPMETHOD.GET))
         requestMsg += "%s" % urlparse.urlsplit(url)[2] or "/"
@@ -308,10 +309,11 @@ class Connect:
                   else kb.originalPage if kb.redirectChoice == REDIRECTION.ORIGINAL\
                   else conn.read()
                 skipLogTraffic = kb.redirectChoice != REDIRECTION.FOLLOW
+                code = conn.redcode
             else:
                 page = conn.read()
 
-            code = conn.code
+            code = code or conn.code
             responseHeaders = conn.info()
             responseHeaders[URI_HTTP_HEADER] = conn.geturl()
             page = decodePage(page, responseHeaders.get(HTTPHEADER.CONTENT_ENCODING), responseHeaders.get(HTTPHEADER.CONTENT_TYPE))