From 71d892a8a951858110c89bab9cd2c66da25df01b Mon Sep 17 00:00:00 2001 From: Aevann1 Date: Wed, 22 Jun 2022 19:44:43 +0200 Subject: [PATCH] get file extension --- files/helpers/media.py | 14 ++++++++++++-- 1 file changed, 12 insertions(+), 2 deletions(-) diff --git a/files/helpers/media.py b/files/helpers/media.py index c1aa58206..abf828156 100644 --- a/files/helpers/media.py +++ b/files/helpers/media.py @@ -35,7 +35,13 @@ def process_other(file): def process_audio(file): - name = f'/audio/{time.time()}'.replace('.','') + '.mp3' + name = f'/audio/{time.time()}'.replace('.','') + + extension = file.filename.split('.')[-1].lower() + if extension not in ['aac', 'amr', 'flac', 'm4a', 'm4b', 'mp3', 'ogg', 'wav']: + extension = 'mp3' + name = name + '.' + extension + file.save(name) if os.stat(name).st_size > 8 * 1024 * 1024: @@ -56,7 +62,11 @@ def webm_to_mp4(old, new): def process_video(file): old = f'/videos/{time.time()}'.replace('.','') file.save(old) - new = old + '.mp4' + + extension = file.filename.split('.')[-1].lower() + if extension not in ['avi', 'mp4', 'webm', 'm4v', 'mov', 'mkv']: + extension = 'mp4' + new = old + '.' + extension if file.filename.split('.')[-1].lower() == 'webm': file.save(new)