def test_set(self): headers = Headers() headers.add('Content-Type', 'application/json') headers.set('Content-Type', 'application/xml') self.assertEqual(headers.get('Content-Type'), 'application/xml') headers = Headers() headers.set('Content-Type', 'application/xml') self.assertEqual(headers.get('Content-Type'), 'application/xml')
def test_normalization(self): headers = Headers([self.ct_headers]) self.assertTrue(headers.get('Content-Type')) self.assertTrue(headers.get('content-type')) self.assertEqual(headers.get('content-type'), 'application/json') headers.add('User-Agent', 'fluffy') self.assertTrue(headers.get('User-Agent')) self.assertTrue(headers.get('user-agent')) self.assertEqual(headers.get('user-agent'), 'fluffy')
def test_multi(self): headers = Headers() self.assertTrue(headers) headers.add('X-Foo', 'bar') headers.add('X-Foo', 'baz') self.assertEqual(headers.get('X-Foo'), 'bar') self.assertEqual(headers.get_all('X-Foo'), ['bar', 'baz']) headers = Headers() headers.add('X-Foo', 'bar', 'baz', 'foo') self.assertEqual(headers.get_all('X-Foo'), ['bar', 'baz', 'foo']) self.assertEqual(headers.get_all('x-fOo'), ['bar', 'baz', 'foo'])
class HTTPClient(object): def __init__(self, agent=None, timeout=60, keep_alive=1, default_headers={}, max_redirect=7, with_exceptions=False): self.timeout = timeout self.max_redirect = max_redirect self._handlers = Handlers() self._with_exceptions = with_exceptions if agent is None: self.agent = "python-httpclient/{version}".format( version=__version__) else: self.agent = agent if len(default_headers) == 0: default_headers = [( 'Connection', 'keep-alive', )] self._default_headers = Headers(default_headers) if 'User-Agent' not in default_headers: self._default_headers.add('User-Agent', self.agent) self._default_headers.add('User-Agent', 'foooo') self._poolmanager = PoolManager( maxsize=keep_alive ) def HTTPException(fn): def wrapper(*args): res = fn(*args) if args[0]._with_exceptions: if res.is_error: raise http_exception(res) return res return wrapper def add_handler(self, position, cb): self._handlers.add_handler(position, cb) def remove_handler(self, position): self._handler.remove_handler(position) def default_header(self, key): return self.default_headers.get('key') @property def default_headers(self): return self._default_headers @HTTPException def request(self, request): return self._request(request) @HTTPException def head(self, url, headers={}): request = Request('HEAD', url, headers=headers) return self._request(request) @HTTPException def get(self, url, headers={}): request = Request('GET', url, headers=headers) return self._request(request) @HTTPException def put(self, url, headers={}, content=None): request = Request('PUT', url, headers=headers, content=content) return self._request(request) @HTTPException def post(self, url, headers={}, content=None): request = Request('POST', url, headers=headers, content=content) return self._request(request) @HTTPException def delete(self, url, headers={}, content=None): request = Request('DELETE', url, headers=headers) return self._request(request) @HTTPException def mirror(self, url, filename): req = Request('GET', url) res = self.request(req) if res.is_success: f = open(filename, 'w') f.write(res.content) f.close() last_modified = Date.time2epoch(res.last_modified) if last_modified: os.utime(file, (last_modified, last_modified)) def _request(self, request): try: resp = self._make_request(request) except Exception, e: print e raise e if resp.is_redirect and len(resp.redirects) < self.max_redirect: try: return self._follow_redirect(resp, request) except Exception, e: raise e