分享一个写小demo用的模板

timespace

有时想写一个小程序，接收几个命令行参数，输出一些信息。如果是‘hello，world’这种复杂度，sys模块和print函数就搞定了，但现实常常复杂那么一点点，查argparse和logging的文档是个体力活，容易被文档淹没，通常又不需要太复杂的功能，所以就抽空写了一个简单模板。

1.完整源码。

1. #!/usr/bin/env python
   
2. # -*- coding: utf-8 -*-
   
3. """Python Demo Template.
   
4. 
   
5. notes:
   
6.   - log file <app.log> will be created in CWD
   
7. 
   
8. tips:
   
9.   - itertools and operator may simplify sort and iterate
   
10.   - sqlite3 may be helpful for your housekeeping file format
    
11. 
    
12. """
    
13. 
    
14. import argparse
    
15. import logging
    
16. import logging.config
    
17. 
    
18. def main():
    
19.     # all of command line arguments should be added here
    
20.     parser = argparse.ArgumentParser(description='Python Demo Template')
    
21.     parser.add_argument('infile', nargs='?', type=argparse.FileType('r'),
    
22.                         default='-', help='text source [default: sys.stdin]')
    
23.     parser.add_argument("-v", "--verbose", action="count", default=0,
    
24.                        help="increase output verbosity")
    
25.     args = parser.parse_args()
    
26. 
    
27.     # runtime services: logging
    
28.     env_init(args.verbose)
    
29.    
    
30.     # main logic begins
    
31.     for line in args.infile:
    
32.         logging.info(line.rstrip())
    
33. 
    
34. def env_init(verbose):
    
35.     # logging facility, only loglevel can be changed
    
36.     if verbose >= 2:
    
37.         loglevel = logging.DEBUG
    
38.     elif verbose >= 1:
    
39.         loglevel = logging.INFO
    
40.     else:
    
41.         loglevel = logging.WARNING
    
42. 
    
43.     LOGGING = {
    
44.         'version': 1,
    
45.         'formatters': {
    
46.             'simple': {
    
47.                 'format': '%(levelname)s %(message)s'
    
48.             },
    
49.             'verbose': {
    
50.                 'format': ('%(levelname)s %(asctime)s %(module)s '
    
51.                            '%(process)d %(thread)d %(message)s')
    
52.             },
    
53.         },
    
54.         'handlers': {
    
55.             'console': {
    
56.                'level': 'DEBUG',
    
57.                'class': 'logging.StreamHandler',
    
58.                'formatter': 'simple'
    
59.             },
    
60.             'file': {
    
61.                 'level': 'INFO',
    
62.                 'class': 'logging.handlers.RotatingFileHandler',
    
63.                 'formatter': 'verbose',
    
64.                 'filename': 'app.log',
    
65.                 'maxBytes': 10*1024*1024,
    
66.                 'backupCount': 5
    
67.             },
    
68.         },
    
69.         'root': {
    
70.             'level': loglevel,
    
71.             'handlers': ['console', 'file']
    
72.         },
    
73.     }
    
74.     logging.config.dictConfig(LOGGING)
    
75. 
    
76. if __name__ == '__main__':
    
77.     main()

复制代码

2. 如果用Vim编辑代码，可以加一个autocmd，假设上述代码保存为py_main。

1. "Python main template
   
2. autocmd! BufNewFile *main.py 0r ~/.vim/templates/py_main

复制代码

3. 用Vim新建以main.py为后缀的文件时，就会自动载入这个模板。