需求
在命令行运行一个 python 文件,同时传入自定义参数:
$ python main.py --node=Table --action=create --data="{'name': 'test2', 'is_sys_obj': False, 'encoding': 'UTF8','datconnlimit': -1, 'variables': []"
希望 main.py 接收命令行参数,并根据 --node 参数,导入指定的模块、模块中的文件、文件中的函数:
实现
main.py:import importlib
import argparse
import ast# 按间距中的绿色按钮以运行脚本。
if __name__ == '__main__':parser = argparse.ArgumentParser()parser.add_argument('--node', type=str)parser.add_argument('--action', type=str)parser.add_argument('--data', type=ast.literal_eval)parser.add_argument('--old_data', type=ast.literal_eval)args = parser.parse_args()# 将args.params转为字典data_dict = dict(args.data)old_data_dict = {}if args.action == 'update':old_data_dict = dict(args.old_data)# 动态导入模块module = importlib.import_module(args.node)# 从模块中导入模块同名文件file = importlib.import_module(module.__name__ + '.' + module.__name__)# 从文件中导入函数 create 和 updatecreate = getattr(file, 'create')update = getattr(file, 'update')if args.action == 'create':create(data_dict)if args.action == 'update':update(data_dict, old_data_dict)