批量创建文件

yjptx121

每次课后题都要新建文件然后改名，就试着写了个简单的脚本

import os
print('注意：请将脚本放于当前文件!')

def set_file_num(file_num, file_ext):
    count = int(file_num)
    sum = count
    os.chdir(os.getcwd())
    while count:
        file_name = '第' + str(count) + '题' + file_ext
        open_file = open(file_name, 'w')
        open_file.close()
        count -= 1
    print('已在路径：{} 新建总计 {} 个 {} 文件'.format(os.getcwd(), sum, file_ext))

ext_dict = {'a': '.txt', 'b': '.py', 'c': '.doc', 'd': '.xls', 'e': '.ppt'}
print('可创建文件为:{}'.format(ext_dict))
ext_choose = input('请选择文件后缀名： ')
file_ext = ext_dict[ext_choose]
file_num = input('请输入要创建的文件个数： ')
set_file_num(file_num, file_ext)