python根据命令行参数动态导入模块或文件

需求

在命令行运行一个 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 和 update
    create = getattr(file, 'create')
    update = getattr(file, 'update')

    if args.action == 'create':
        create(data_dict)

    if args.action == 'update':
        update(data_dict, old_data_dict)
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值