fix: Avoid TypeError when guess_extension returns None

This commit is contained in:
Gavin D'souza 2022-01-31 11:43:24 +05:30
parent 8d9b46527f
commit adc745a72c

View file

@ -660,7 +660,9 @@ def get_extension(filename, extn, content: bytes = None, response: "Response" =
content_type = response.headers.get("Content-Type")
if content_type:
return mimetypes.guess_extension(content_type)[1:]
_extn = mimetypes.guess_extension(content_type)
if _extn:
return _extn[1:]
if extn:
# remove '?' char and parameters from extn if present