代码框架生成器

一 代码

1. import os
2. import sys
3. import datetime
5. py ='''
6. #-----------------------------------------------
7. # TO:
8. #-----------------------------------------------
9. # BY:
10. #-----------------------------------------------
11. '''
13. if os.path.isfile(sys.argv[1]):
14. print('%s already exist!'% sys.argv[1])
15. sys.exit()
16. file = open(sys.argv[1],'w')
17. today =datetime.date.today()
18. date = today.strftime('%Y')+'-'+ today.strftime('%m')+'-'+today.strftime('%d')
19. filetypes = str.split(sys.argv[1],'.')
20. length = len(filetypes)
21. filetype = filetypes[length -1]
22. if filetype =='py':
23. print('use python mode')
24. file.writelines('# File: '+ sys.argv[1])
25. file.write('\n')
26. file.write(py)
27. file.write('# Date: '+date)
28. file.write('\n')
29. file.write('#-----------------------------------------------')
30. else:
31. print('just create %s '% sys.argv[1])
32. file.close()

 

二 运行说明

E:\python\work>MakeCode.py test1.py

use python mode

在当前目录下生成test1.py，内容如下：

# File: test1.py

 

#-----------------------------------------------

# TO:

#-----------------------------------------------

# BY:

#-----------------------------------------------

# Date: 2017-05-07

#-----------------------------------------------