From c41f6977e7569c5c473306318e72d7fb8f1c8a33 Mon Sep 17 00:00:00 2001 From: Aevann1 Date: Sun, 22 May 2022 20:03:54 +0200 Subject: [PATCH] sdf --- files/helpers/media.py | 12 ++++-------- 1 file changed, 4 insertions(+), 8 deletions(-) diff --git a/files/helpers/media.py b/files/helpers/media.py index b78401258..0c9932251 100644 --- a/files/helpers/media.py +++ b/files/helpers/media.py @@ -15,16 +15,12 @@ def process_video(file): os.remove(name) with open(f"{name}.mp4", 'rb') as f: + if SITE_NAME != 'rDrama' or os.stat(f'{name}.mp4').st_size > 8 * 1024 * 1024: + os.remove(f"{name}.mp4") try: req = requests.request("POST", "https://pomf2.lain.la/upload.php", files={'files[]': f}, timeout=20).json() except requests.Timeout: return {"error": "Video upload timed out, please try again!"} - - try: url = req['files'][0]['url'] - except: return {"error": req['description']} - - if SITE_NAME != 'rDrama' or os.stat(f'{name}.mp4').st_size > 8 * 1024 * 1024: - os.remove(f"{name}.mp4") - - return url + try: return req['files'][0]['url'] + except: return {"error": req['description']} def process_image(patron, filename=None, resize=0):