velvox
/
litecord
mirror of https://gitlab.com/litecord/litecord.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

images: hardcode jpeg/webp

This commit is contained in:
Luna Mendes 2018-11-20 17:02:15 -03:00
parent 3e28501bdb
commit 6a085c2190
1 changed files with 18 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
litecord/images.py
View File

@ -14,17 +14,30 @@ IMAGE_FOLDER = Path('./images')
log = Logger(__name__) log = Logger(__name__)
def _get_ext(mime: str): EXTENSIONS = {
if mime == 'image/webp': 'image/jpeg': 'jpeg',
return 'webp' 'image/webp': 'webp'
}
MIMES = {
'jpg': 'image/jpeg',
'jpe': 'image/jpeg',
'webp': 'image/webp',
}
def _get_ext(mime: str) -> str:
if mime in EXTENSIONS:
return EXTENSIONS[mime]
extensions = mimetypes.guess_all_extensions(mime) extensions = mimetypes.guess_all_extensions(mime)
return extensions[0].strip('.') return extensions[0].strip('.')
def _get_mime(ext: str): def _get_mime(ext: str):
if ext == 'webp': if ext in MIMES:
return 'image/webp' return MIMES[ext]
return mimetypes.types_map[f'.{ext}'] return mimetypes.types_map[f'.{ext}']