139 lines
4.0 KiB
Python
139 lines
4.0 KiB
Python
"""
|
|
Mandaye HTTP object: Response, Request and Headers
|
|
"""
|
|
|
|
import copy
|
|
import Cookie
|
|
|
|
class HTTPHeader(dict):
|
|
""" Mandaye class to manage HTTP header
|
|
"""
|
|
|
|
def load_httpmessage(self, headers):
|
|
""" headers: HTTPMessage (httplib)
|
|
"""
|
|
for key in headers.keys():
|
|
self[key.lower()] = headers.getheaders(key)
|
|
|
|
def load_from_list(self, headers):
|
|
""" headers: list with the headers
|
|
"""
|
|
for header in headers:
|
|
self[header[0].lower()] = [header[1]]
|
|
|
|
def load_from_dict(self, headers):
|
|
"""
|
|
"""
|
|
for key, value in headers.iteritems():
|
|
self.addheader(key, value)
|
|
|
|
def addheader(self, name, value):
|
|
""" name: field name
|
|
value: string with the field value """
|
|
if isinstance(value, str):
|
|
self[name.lower()] = [value]
|
|
else:
|
|
raise TypeError('Header value must be a string')
|
|
|
|
def addheaders(self, name, values):
|
|
""" name: field name
|
|
values: list of field values """
|
|
self[name.lower()] = value
|
|
|
|
def getheader(self, name):
|
|
if self.has_key(name.lower()):
|
|
return self[name.lower()][0]
|
|
else:
|
|
return None
|
|
|
|
def getheaders(self):
|
|
res = {}
|
|
for key, values in self.iteritems():
|
|
res[key.lower()] = ', '.join(values)
|
|
return res
|
|
|
|
def addsetcookies(self, cookies):
|
|
""" cookies: Cookie.SimpleCookie object """
|
|
self['set-cookie'] = []
|
|
for cookie in cookies.values():
|
|
self['set-cookie'].append(cookie.output(header=""))
|
|
|
|
def items(self):
|
|
items = super(HTTPHeader, self).items()
|
|
new_items = []
|
|
for (key, values) in items:
|
|
for value in values:
|
|
new_items.append((key, value))
|
|
return new_items
|
|
|
|
def copy(self):
|
|
return copy.deepcopy(self)
|
|
|
|
class HTTPRequest(object):
|
|
""" Mandaye HTTP Request
|
|
"""
|
|
|
|
def __init__(self, cookies=None, headers=None, req_method="GET", msg=None, target=None):
|
|
"""
|
|
cookies: BaseCookie object (http://docs.python.org/library/cookie.html)
|
|
headers: HTTPHeader object
|
|
req_method: GET or POST
|
|
msg: message body str
|
|
"""
|
|
self.cookies = cookies
|
|
self.headers = headers
|
|
self.msg = msg
|
|
self.req_method = req_method
|
|
if not headers:
|
|
self.headers = HTTPHeader()
|
|
if not cookies:
|
|
self.cookies = Cookie.BaseCookie()
|
|
self.target = target
|
|
|
|
def __repr__(self):
|
|
d = self.__dict__
|
|
return '<%s %s>' % (self.__class__.__name__, d)
|
|
|
|
def copy(self):
|
|
return copy.deepcopy(self)
|
|
|
|
class HTTPResponse(object):
|
|
""" Mandaye HTTP Response
|
|
"""
|
|
|
|
def __init__(self, code=None, reason=None, headers=None, msg=None, cookies=None):
|
|
"""
|
|
code: integer return code (200, 302, 404, ..)
|
|
reason: string with http status (ex: OK, Not Found, ...)
|
|
headers: dictionary with the headers
|
|
msg: message body str
|
|
cookies: BaseCookie object (http://docs.python.org/library/cookie.html)
|
|
"""
|
|
self.code = code
|
|
self.reason = reason
|
|
self.headers = headers
|
|
self.msg = msg
|
|
self.cookies = cookies
|
|
if not headers:
|
|
self.headers = HTTPHeader()
|
|
if not cookies:
|
|
self.cookies = Cookie.BaseCookie()
|
|
|
|
def load_from_urllib(self, response):
|
|
""" response: urrlib.addinfourl
|
|
"""
|
|
self.headers.load_httpmessage(response.headers)
|
|
for setcookie in response.headers.getheaders('set-cookie'):
|
|
cookie = Cookie.SimpleCookie("Set-Cookie: %s" % setcookie)
|
|
cookie = cookie.values()[0]
|
|
if self.cookies.has_key(cookie.key):
|
|
del self.cookies[cookie.key]
|
|
self.cookies.load("Set-Cookie: %s" % setcookie)
|
|
self.code = response.code
|
|
self.reason = response.msg
|
|
self.msg = response.read()
|
|
|
|
def __repr__(self):
|
|
d = self.__dict__
|
|
return '<%s %s>' % (self.__class__.__name__, d)
|