mirror of
https://github.com/kevin1024/vcrpy.git
synced 2025-12-08 16:53:23 +00:00
e84cd6f059
So the stubs were getting out of hand, and while trying to add support for the putrequest and putheader methods, I had an idea for a cleaner way to handle the stubs using the VCRHTTPConnection more as a proxy object. So VCRHTTPConnection and VCRHTTPSConnection no longer inherit from HTTPConnection and HTTPSConnection. This allowed me to get rid of quite a bit of copy-and-pasted stdlib code.
58 lines
1.4 KiB
Python
58 lines
1.4 KiB
Python
class Request(object):
|
|
|
|
def __init__(self, protocol, host, port, method, path, body, headers):
|
|
self.protocol = protocol
|
|
self.host = host
|
|
self.port = port
|
|
self.method = method
|
|
self.path = path
|
|
self.body = body
|
|
# make headers a frozenset so it will be hashable
|
|
self.headers = frozenset(headers.items())
|
|
|
|
def add_header(self, key, value):
|
|
tmp = dict(self.headers)
|
|
tmp[key] = value
|
|
self.headers = frozenset(tmp.iteritems())
|
|
|
|
@property
|
|
def url(self):
|
|
return "{0}://{1}{2}".format(self.protocol, self.host, self.path)
|
|
|
|
def __key(self):
|
|
return (
|
|
self.host,
|
|
self.port,
|
|
self.method,
|
|
self.path,
|
|
self.body,
|
|
self.headers
|
|
)
|
|
|
|
def __hash__(self):
|
|
return hash(self.__key())
|
|
|
|
def __eq__(self, other):
|
|
return hash(self) == hash(other)
|
|
|
|
def __str__(self):
|
|
return "<Request ({0}) {1}>".format(self.method, self.url)
|
|
|
|
def __repr__(self):
|
|
return self.__str__()
|
|
|
|
def _to_dict(self):
|
|
return {
|
|
'protocol': self.protocol,
|
|
'host': self.host,
|
|
'port': self.port,
|
|
'method': self.method,
|
|
'path': self.path,
|
|
'body': self.body,
|
|
'headers': self.headers,
|
|
}
|
|
|
|
@classmethod
|
|
def _from_dict(cls, dct):
|
|
return Request(**dct)
|