RedditVideoMakerBot/utils/cleanup.py

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