Skip to content

Commit b7070e2

Browse files
authored
Update __init__.py
Signed-off-by: FuNK3Y <[email protected]>
1 parent 65a1411 commit b7070e2

File tree

1 file changed

+13
-3
lines changed

1 file changed

+13
-3
lines changed

python-ecosys/aiohttp/aiohttp/__init__.py

Lines changed: 13 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -17,9 +17,15 @@
1717
class ClientResponse:
1818
def __init__(self, reader):
1919
self.content = reader
20+
21+
def get_header(self, keyname, value=None):
22+
try:
23+
return next((v for k, v in self.headers.items() if k.lower() == keyname.lower()))
24+
except StopIteration:
25+
return value
2026

2127
def _decode(self, data):
22-
c_encoding = self.headers.get("Content-Encoding")
28+
c_encoding = self.get_header("Content-Encoding")
2329
if c_encoding in ("gzip", "deflate", "gzip,deflate"):
2430
try:
2531
import deflate
@@ -39,10 +45,10 @@ async def read(self, sz=-1):
3945
return self._decode(await self.content.read(sz))
4046

4147
async def text(self, encoding="utf-8"):
42-
return (await self.read(int(self.headers.get("Content-Length", -1)))).decode(encoding)
48+
return (await self.read(int(self.get_header("Content-Length", -1)))).decode(encoding)
4349

4450
async def json(self):
45-
return _json.loads(await self.read(int(self.headers.get("Content-Length", -1))))
51+
return _json.loads(await self.read(int(self.get_header("Content-Length", -1))))
4652

4753
def __repr__(self):
4854
return "<ClientResponse %d %s>" % (self.status, self.headers)
@@ -267,3 +273,7 @@ async def _ws_connect(self, url, ssl=None):
267273
await ws_client.connect(url, ssl=ssl, handshake_request=self.request_raw)
268274
self._reader = ws_client.reader
269275
return ClientWebSocketResponse(ws_client)
276+
277+
278+
__version__ = '0.0.3'
279+

0 commit comments

Comments
 (0)