from pathlib import Path


def check_create_new_folder(dir_or_path):
    """
    检查是否存在某个文件夹，或者输入的文件所在的目录是否存在，如果不存在话 就生成新的文件夹
    """
    path = Path(dir_or_path)
    if not path.exists():
        path.mkdir(mode=0o777,parents=True,exist_ok=True)
    #如果传入的是一个文档，需要删掉新建的以文档命名的文件夹
        if not path.is_file() :
            path.rmdir()
    return dir_or_path