diff --git a/utils/cleanup.py b/utils/cleanup.py index c2f104f..ee3453b 100644 --- a/utils/cleanup.py +++ b/utils/cleanup.py @@ -1,20 +1,23 @@ import os +from os.path import exists def cleanup() -> int: - count = 0 - files = [f for f in os.listdir('.') if f.endswith('.mp4') and 'temp' in f.lower()] - count += len(files) - for f in files: - os.remove(f) - try: - for file in os.listdir('./assets/temp/mp4'): + if exists('./assets/temp'): + count = 0 + files = [f for f in os.listdir('.') if f.endswith('.mp4') and 'temp' in f.lower()] + count += len(files) + for f in files: + os.remove(f) + try: + for file in os.listdir('./assets/temp/mp4'): + count += 1 + os.remove('./assets/temp/mp4/' + file) + except FileNotFoundError: + pass + for file in os.listdir('./assets/temp/mp3'): count += 1 - os.remove('./assets/temp/mp4/' + file) - except FileNotFoundError: - pass - for file in os.listdir('./assets/temp/mp3'): - count += 1 - os.remove('./assets/temp/mp3/' + file) - return count + os.remove('./assets/temp/mp3/' + file) + return count + return 0