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):