批量创建文件

yjptx121

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

1. import os
   
2. print('注意：请将脚本放于当前文件!')
   
3. 
   
4. def set_file_num(file_num, file_ext):
   
5.     count = int(file_num)
   
6.     sum = count
   
7.     os.chdir(os.getcwd())
   
8.     while count:
   
9.         file_name = '第' + str(count) + '题' + file_ext
   
10.         open_file = open(file_name, 'w')
    
11.         open_file.close()
    
12.         count -= 1
    
13.     print('已在路径：{} 新建总计 {} 个 {} 文件'.format(os.getcwd(), sum, file_ext))
    
14. 
    
15. ext_dict = {'a': '.txt', 'b': '.py', 'c': '.doc', 'd': '.xls', 'e': '.ppt'}
    
16. print('可创建文件为:{}'.format(ext_dict))
    
17. ext_choose = input('请选择文件后缀名： ')
    
18. file_ext = ext_dict[ext_choose]
    
19. file_num = input('请输入要创建的文件个数： ')
    
20. set_file_num(file_num, file_ext)

复制代码