Firepup650/bitbot-3.11-fork
2
0
Fork 0
Code Issues Pull requests Activity Actions

parse out content_type in Response ctor

Browse source
This commit is contained in:
jesopo 2019-10-05 22:56:56 +01:00
parent cac8cb7b7e
commit 9ab817ca58
1 changed files with 3 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
src/utils/http.py
View file

@ -121,11 +121,11 @@ class Request(object):
class Response(object): class Response(object):
def __init__(self, code: int, data: typing.Any, def __init__(self, code: int, data: typing.Any,
content_type: str, headers: typing.Dict[str, str], encoding: str): headers: typing.Dict[str, str], encoding: str):
self.code = code self.code = code
self.data = data self.data = data
self.content_type = content_type
self.headers = headers self.headers = headers
self.content_type = headers.get("Content-Type", "").split(";", 1)[0]
self.encoding = encoding self.encoding = encoding
def _meta_content(s: str) -> typing.Dict[str, str]: def _meta_content(s: str) -> typing.Dict[str, str]:
@ -177,10 +177,8 @@ def _request(request_obj: Request) -> Response:
raise ValueError("Response too large") raise ValueError("Response too large")
headers = utils.CaseInsensitiveDict(dict(response.headers)) headers = utils.CaseInsensitiveDict(dict(response.headers))
content_type = headers.get("Content-Type", "").split(";", 1)[0]
our_response = Response(response.status_code, response_content, our_response = Response(response.status_code, response_content,
content_type=content_type, headers=headers, headers=headers, encoding=response.encoding)
encoding=response.encoding)
return our_response return our_response
try: try: