def redirect_request(self, request, fp, code, msg, headers, new_url):
request_method = request.get_method()
if str(code) in self.redirect_codes and request_method in self.valid_methods:
new_url = new_url.replace(' ', '%20')
request = Request(new_url,
data=request.data,
headers=request.headers,
origin_req_host=request.get_origin_req_host(),
unverifiable=True)
if self.method in self.valid_methods:
if request.get_method() != self.method:
request.get_method = lambda: self.method
return request
else:
HTTPRedirectHandler.redirect_request(request, fp, code, msg, headers, new_url)
def redirect_request(self, request, fp, code, msg, headers, new_url):
request_method = request.get_method()
if str(
code
) in self.redirect_codes and request_method in self.valid_methods:
new_url = new_url.replace(' ', '%20')
request = Request(new_url,
data=request.data,
headers=request.headers,
origin_req_host=request.get_origin_req_host(),
unverifiable=True)
if self.method in self.valid_methods:
if request.get_method() != self.method:
request.get_method = lambda: self.method
return request
else:
HTTPRedirectHandler.redirect_request(request, fp, code, msg,
headers, new_url)
def http_error_301(self, req, fp, code, msg, headers):
"""
Wraps the :func:`urllib2.HTTPRedirectHandler.http_error_301` handler,
setting the ``result.status`` to ``301`` in case a http ``301`` error
is encountered.
"""
result = HTTPRedirectHandler.http_error_301(self, req, fp, code, msg,
headers)
result.status = code
return result
def http_error_302(self, req, fp, code, msg, headers):
"""
Wraps the :func:`urllib2.HTTPRedirectHandler.http_error_302` handler,
setting the ``result.status`` to ``302`` in case a http ``302`` error
is encountered.
"""
result = HTTPRedirectHandler.http_error_302(
self, req, fp, code, msg, headers)
result.status = code
return result
def http_error_302(self, req, fp, code, msg, headers):
# return reference
result = None
result = HTTPRedirectHandler.http_error_302( self, req, fp, code, msg, headers )
# log redirect info.
self.log_redirect_info( result, code )
return result
def http_error_302(self, req, fp, code, msg, headers):
# Jenkins can send several Set-Cookie values sometimes
# The valid one is the last one
for header, value in headers.items():
if header.lower() == 'set-cookie':
cookie = self.extract_cookie(value)
req.headers['Cookie'] = cookie
result = HTTPRedirectHandler.http_error_302(
self, req, fp, code, msg, headers)
result.orig_status = code
result.orig_headers = headers
result.cookie = cookie
return result
def http_error_302(self, req, fp, code, msg, headers):
# Jenkins can send several Set-Cookie values sometimes
# The valid one is the last one
for header, value in headers.items():
if header.lower() == 'set-cookie':
cookie = self.extract_cookie(value)
req.headers['Cookie'] = cookie
result = HTTPRedirectHandler.http_error_302(self, req, fp,
code, msg,
headers)
result.orig_status = code
result.orig_headers = headers
result.cookie = cookie
return result
def redirect_request(self, req, fp, code, msg, hdrs, newurl):
self.redirects.add(newurl)
return HTTPRedirectHandler.redirect_request(
self, req, fp, code, msg, hdrs, newurl)
def redirect_request(self, req, fp, code, msg, headers, newurl):
new_req = HTTPRedirectHandler.redirect_request(self, req, fp, code,
msg, headers, newurl)
req.redirect_code = code
return new_req
def redirect_request(self, req, fp, code, msg, headers, newurl):
new_req = HTTPRedirectHandler.redirect_request(self, req, fp, code,
msg, headers, newurl)
req.redirect_code = code
return new_req
