自动生成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:
#-----------------------------------------------------