自动生成Python初始模板——makecode.py增强版

#!/usr/bin/python

# -*- coding:utf-8 -*-

# file: MakeCode.py

# Modified from 孙广雷 《征服Python——语言基础和典型应用》

# By Wubin Qu [quwubin AT gmail.com]

import os

import re

import sys

import string

import datetime

# python脚本模板

py = ”’#!/usr/bin/env python

# -*- coding:utf-8 -*-

__version__ = ‘1.0′

__author__ = ‘Wubin Qu @ZCGLAB @BMI @CHINA’

__blog__ = ‘http://quwubin.cnblogs.com/’

__license__ = ‘GPL’

#—————————————————–

”’

# c模板

c = ”’

*—————————————————–

* TO:

*—————————————————–

* BY:

*—————————————————–

”’

if len(sys.argv) 2:

print ‘One file one time’

sys.exit()

else:

file_name = sys.argv[1]

if os.path.isfile(file_name): # 判断要创建的文件是否存在，如果存在则退出脚本

print ‘%s already exist!’ % file_name

sys.exit()

file = open(file_name, ‘w’) # 创建文件

today = datetime.date.today() # 获得当前日期，并格式化为xxxx-xx-xx的形式

date = today.strftime(’%Y’)+’-'+today.strftime(’%m’)+’-'+today.strftime(’%d’)

if re.search(’\.’, file_name):

filetypes = string.split(file_name,’.') # 判断将创建的文件是什么类型以便对其分别处理

length = len(filetypes)

filetype = filetypes[length - 1]

else:

filetype = ‘Others’

if filetype == ‘py’:

print ‘use python mode’

file.write(py)

file.writelines(’# File: ‘ + file_name)

file.write(’\n’)

file.write(’# Date: ‘ + date)

file.write(’\n’)

file.write(’# Description: ‘)

file.write(’\n’)

file.write(’#—————————————————–’)

file.write(’\n’)

file.write(’\n’)

file.write(’\n’)

elif filetype == ‘c’ or filetype == ‘cpp’:

print ‘use c mode’

file.writelines(’/*’)

file.write(’\n’)

file.writelines(’ *—————————————————–’)

file.write(’\n’)

file.writelines(’ * File: ‘ + file_name)

file.write(’\n’)

file.write(c)

file.write(’ * Date: ‘ + date)

file.write(’\n’)

file.write(’ *—————————————————–’)

file.write(’\n’)

file.write(’ */ \n’)

else:

print ‘just create %s’ % file_name

file.close() # 关闭文件

生成的文件样例：

1#!/usr/bin/env python

2# -*- coding:utf-8 -*-

3

4__version__ = '1.0'

5__author__ = 'Wubin Qu @ZCGLAB @BMI @CHINA'

6__blog__ = 'http://quwubin.cnblogs.com/'

7__license__ = 'GPL'

8

9#-----------------------------------------------------

# File: example.py

# Date: 2008-06-21

# Description:

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