third-party-mirrors/EDDN
1
0
Fork 0
mirror of https://github.com/EDCD/EDDN.git synced 2025-04-27 13:42:14 +03:00
Code Issues Actions Packages Projects Releases Wiki Activity

Gateway: Pass headers and body into get_decompressed_message()

Browse Source 
This makes it easier to write unit tests for this function, not having to
somehow mock up the bottle request/body stuff.
This commit is contained in:
Athanasius 2022-08-19 16:27:22 +01:00
parent 580160b95b
commit e28ab45be4
No known key found for this signature in database
GPG Key ID: 8C392035DD80FD62
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
src/eddn/Gateway.py
View File

@ -169,7 +169,7 @@ def get_remote_address() -> str:
return request.headers.get("X-Forwarded-For", request.remote_addr)
def get_decompressed_message() -> bytes:
def get_decompressed_message(headers: dict, input: bytes) -> bytes:
"""
Detect gzip Content-Encoding headers and de-compress on the fly.
@ -177,7 +177,7 @@ def get_decompressed_message() -> bytes:
:rtype: str
:returns: The de-compressed request body.
"""
content_encoding = request.headers.get("Content-Encoding", "")
content_encoding = headers.get("Content-Encoding", "")
logger.debug("Content-Encoding: %s", content_encoding)
if content_encoding in ["gzip", "deflate"]:
@ -187,18 +187,18 @@ def get_decompressed_message() -> bytes:
try:
# Auto header checking.
logger.debug("Trying zlib.decompress (15 + 32)...")
message_body = zlib.decompress(request.body.read(), 15 + 32)
message_body = zlib.decompress(input, 15 + 32)
except zlib.error:
logger.error("zlib.error, trying zlib.decompress (-15)")
# Negative wbits suppresses adler32 checksumming.
message_body = zlib.decompress(request.body.read(), -15)
message_body = zlib.decompress(input, -15)
logger.debug("Resulting message_body:\n%s\n", message_body)
else:
logger.debug("Content-Encoding indicates *not* compressed...")
message_body = request.body.read()
message_body = input
return message_body
@ -314,7 +314,7 @@ def upload() -> str:
"""
try:
# Body may or may not be compressed.
message_body = get_decompressed_message()
message_body = get_decompressed_message(request.headers, request.body.read())
except zlib.error as exc:
# Some languages and libs do a crap job zlib compressing stuff. Provide