blankie
/
redditbot
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Use strict=False

This commit is contained in:
blank X 2021-04-29 16:52:08 +07:00
parent ad17d5a4a5
commit fb1b228d5b
Signed by: blankie
GPG Key ID: CC15FC822C7F61F5
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
redditbot.py
View File

@ -219,7 +219,7 @@ async def main():
file.write(chunk) file.write(chunk)
async def _get_file_mimetype(filename): async def _get_file_mimetype(filename):
mimetype = mimetypes.guess_type(filename)[0] mimetype = mimetypes.guess_type(filename, strict=False)[0]
if not mimetype: if not mimetype:
proc = await asyncio.create_subprocess_exec('file', '--brief', '--mime-type', filename, stdout=asyncio.subprocess.PIPE) proc = await asyncio.create_subprocess_exec('file', '--brief', '--mime-type', filename, stdout=asyncio.subprocess.PIPE)
stdout, _ = await proc.communicate() stdout, _ = await proc.communicate()
@ -232,7 +232,7 @@ async def main():
ext = stdout.decode().strip().split('/', maxsplit=1)[0] ext = stdout.decode().strip().split('/', maxsplit=1)[0]
if not ext or ext == '???': if not ext or ext == '???':
mimetype = await _get_file_mimetype(filename) mimetype = await _get_file_mimetype(filename)
ext = mimetypes.guess_extension(mimetype) or '.bin' ext = mimetypes.guess_extension(mimetype, strict=False) or '.bin'
if not ext.startswith('.'): if not ext.startswith('.'):
ext = '.' + ext ext = '.' + ext
return ext return ext