evanferrao
/
dump-mirrorbot
mirror of https://github.com/Box-boi/dump-mirrorbot.git
1
0
Fork 0
Code Issues Actions Packages Projects Releases Wiki Activity

fs_utils.py: Use python-magic for guessing mimetype (#31) 

1. mimetypes is not a reliable way to know the mime type of a file.

2. python-magic uses libmagic which identifies file types by checking their headers according to a predefined list of file types.

	https://github.com/ahupp/python-magic
This commit is contained in:
Aki 2020-01-16 22:36:55 +05:30 committed by Shivam Jha
parent 00a8e57467
commit 351d4030a8
2 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
bot/helper/ext_utils/fs_utils.py
View File

@ -3,7 +3,7 @@ from bot import aria2, LOGGER, DOWNLOAD_DIR
import shutil import shutil
import os import os
import pathlib import pathlib
import mimetypes import magic
def clean_download(path: str): def clean_download(path: str):
@ -39,6 +39,7 @@ def tar(orig_path: str):
def get_mime_type(file_path): def get_mime_type(file_path):
mime_type = mimetypes.guess_type(file_path)[0] mime = magic.Magic(mime=True)
mime_type = mime.from_file(file_path)
mime_type = mime_type if mime_type else "text/plain" mime_type = mime_type if mime_type else "text/plain"
return mime_type return mime_type

1
requirements.txt
View File

@ -5,3 +5,4 @@ google-auth-oauthlib>=0.4.1,<0.10.0
aria2p>=0.3.0,<0.10.0 aria2p>=0.3.0,<0.10.0
python-dotenv>=0.10 python-dotenv>=0.10
tenacity>=6.0.0 tenacity>=6.0.0
python-magic