def batch_import(argv=None): template={ 'proc':do_import, '-r --root':{ 'metavar':'path', 'dest':'root', 'default':ROOT, 'help':'指定根目录'}, '-d --docs':{ 'metavar':'doc', 'nargs':'*', 'dest':'docs', 'help':'指定导入的类型'}} import stdlib stdlib.parse_args(template,allow_empty=True)
def main(argv=None): '''主程序''' def proc(skip=False,branch=False): # 初始化数据库 mongo_init() # 导入数据 if not skip: for model in (AccInfo,PiaoJu,ZhangHu): model.load_file() print('导入 %s 文件成功'%(model.filename)) export_data(branch) Templates={'-s --skip': {'help':'跳过导入文件', 'action':'store_true'}, '-b --branch': {'help':'导出分行数据', 'action':'store_true'}, 'proc':proc} parse_args(Templates,allow_empty=True)