import argparse
import os


def parse_args():
    parser = argparse.ArgumentParser(description="Utility for folders's clean up.")
    parser.add_argument('-f', '--folders', type=str, nargs='*', help='Names of the folders to be cleaned.')

    args = parser.parse_args()
    return args


def check_dir(dir_path):
    if not os.path.exists(dir_path):
        try:
            os.mkdir(dir_path)
        except OSError as exc:
            raise exc


if __name__ == '__main__':
    folders = parse_args().folders
    if not folders:
        folders = ['docx', 'html', 'json', 'logs', 'config']

    folder_path = os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
    folders = [os.path.join(folder_path, folder) for folder in folders]

    try:
        for folder in folders:
            check_dir(folder)

    except OSError as exc:
        print(exc)
        raise