Files
RedditVideoMakerBot/utils/cleanup.py
T
Jason 804967e5e4 refactor: refactored using black formatter.
cmd: black . --line-length 101
2022-07-01 19:13:44 -04:00

28 lines
757 B
Python

import os
from os.path import exists
def cleanup() -> int:
"""Deletes all temporary assets in assets/temp
Returns:
int: How many files were deleted
"""
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