FreakU
/
takahe
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Don't download files with no content-length

This commit is contained in:
Andrew Godwin 2022-12-17 12:16:37 -07:00
parent 5bbcc0f6c3
commit 256ebe56d5
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
core/files.py
View File

@ -58,7 +58,7 @@ async def get_remote_file(
try: try:
content_length = int(stream.headers["content-length"]) content_length = int(stream.headers["content-length"])
allow_download = content_length <= max_size allow_download = content_length <= max_size
except TypeError: except (KeyError, TypeError):
pass pass
if allow_download: if allow_download:
file = ContentFile(await stream.aread(), name=url) file = ContentFile(await stream.aread(), name=url)