python.snippet

python 的代码片段， 拿出来分享。

#!/usr/bin/env python# -*- coding: utf-8 -*-"""desc"""# ---------# Change Logs:## ---------__author__ = 'Pidong'__email__ = '***'__version__ = '1.0.1'__status__ = 'Production'import sys# reload(sys)# sys.setdefaultencoding('utf-8')import argparseimport loggingdef log(file_name, logger_name='lipidong', verbose=False):    logger = logging.getLogger(logger_name)    formatter = logging.Formatter("%(asctime)s - %(levelname)s - %(message)s")    handler = logging.FileHandler(file_name)    handler.setFormatter(formatter)    logger.addHandler(handler)    logger.setLevel(logging.INFO)    if verbose:        console = logging.StreamHandler()        console.setLevel(logging.INFO)        logger.addHandler(console)    return loggerdef get_args():    parser = argparse.ArgumentParser(prog='desc')    parser.add_argument('--input_file', help='')    parser.add_argument('--log', help='log file, default=log.log', default='log.log')    parser.add_argument("--verbose", help="increase output verbosity",                         action="store_true")    if len(sys.argv) == 1:        parser.print_help()        exit()    return parser.parse_args()def test(input_file):    """TODO: Docstring for test.    :input_file:    :returns: None    """    logger.info(input_file)def main():    args = get_args()    input_file = args.input_file    log_file = args.log    verbose = args.verbose    global logger    logger = log(log_file, verbose=verbose)    test(input_file)if __name__ == '__main__':    main()