Devine/core/dir.py

import os

from ..logging import LOGGER


def dirr():
    for file in os.listdir():
        if file.endswith(".jpg"):
            os.remove(file)
        elif file.endswith(".jpeg"):
            os.remove(file)
        elif file.endswith(".png"):
            os.remove(file)

    if "downloads" not in os.listdir():
        os.mkdir("/tmp/downloads")
    if "cache" not in os.listdir():
        os.mkdir("/tmp/cache")

    LOGGER(__name__).info("Directories Updated.")