import os
import argparse


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: str):
    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 = ["books/epub", "books/docx", "books/html", "books/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:
        [check_dir(folder) for folder in folders]
    except OSError as exc:
        print(exc)
        raise