folder_creation.py

import os

def remove_files(dirs):
    for files in os.listdir(dirs):
        os.remove(os.path.join(dirs, files))


def make_directory_if_not_exists(dir_name):
    if not os.path.exists(dir_name):
        os.mkdir(dir_name)
    else:
        remove_files(dir_name)