|
123456789101112131415161718 |
-
- from urllib2 import HTTPRedirectHandler
-
- class CookieAddingHTTPRedirectHandler(HTTPRedirectHandler):
- """The standard python HttpRedirectHandler doesn't add a cookie to the new request after a 302. This handler does."""
-
- def __init__(self, cookie_jar):
- self.cookie_jar = cookie_jar
-
- # This really ought to call the superclasses init method, but there doesn't seem to be one.
-
-
- def redirect_request(self, *args):
- new_request = HTTPRedirectHandler.redirect_request(self, *args)
- # We need to add a cookie from the cookie_jar
- self.cookie_jar.add_cookie_header(new_request)
-
- return new_request
|