You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
RedditVideoMakerBot/utils/cleanup.py

23 lines
643 B

import os
from os.path import exists
def cleanup() -> int:
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/mp3/' + file)
return count
return 0