python-os创建文件夹-create_dir_if_not_exist.py

#!/bin/usr/env python3

__author__ = 'nxz'

import os
import argparse

MESSAGE = '%s 文件夹已经存在'

def create_dir(work_dir, createdir):
    try:
        for dir in createdir:
            if not os.path.exists(os.path.join(work_dir, dir)):
                os.makedirs(os.path.join(work_dir, dir))
                print("%s 文件夹创建成功" % dir)
            else:
                print(MESSAGE % dir)
    except Exception as e:
        print(e)

def get_parser():
    parser = argparse.ArgumentParser(description='如果指定路径下文件夹不存在,则创建')
    parser.add_argument('work_dir', metavar='WORK_DIR', nargs=1, type=str, help='指定文件夹')
    parser.add_argument('create_dir', metavar='CREATE_DIR', nargs='+', type=str, help='要创建的文件夹')
    return parser

def main():
    args = vars(get_parser().parse_args())
    work_dir = args['work_dir'][0]
    dir = args['create_dir']
    create_dir(work_dir, dir)

if __name__ == '__main__':
    main()

执行:python create_dir_if_not_exist.py C:\Users\nxz\Desktop\123 1111 2222

posted @ 2019-04-03 16:52  xj-record  阅读(2909)  评论(0编辑  收藏  举报