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