工具--入参处理

最新推荐文章于 2024-09-25 10:44:51 发布

l402072123

最新推荐文章于 2024-09-25 10:44:51 发布

阅读量181

点赞数

CC 4.0 BY-SA版权

分类专栏：工具文章标签： python

本文链接：https://blog.csdn.net/l402072123/article/details/123181731

工具专栏收录该内容

4 篇文章

订阅专栏

这段代码定义了一个获取脚本输入参数并将其转化为字典的函数。首先，通过`argparse`模块解析命令行参数，然后遍历参数对象的属性，如果值是JSON格式，将其转换为字典。最终返回处理后的字典，方便后续操作使用。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

入参处理

接受入参，并将入参处理为字典形式

# -*- coding: utf-8 -*-
"""
author:drliu
date:2022/2/16
desc: 获取代码的入参，并处理成字典类型返回
"""

import os
import sys
import argparse
import json

from print_log import print_log

__all__ = ['get_args']


def args_set():
    """
    :return: 返回所有脚本入参，入参列表
    """
    arg_set = argparse.ArgumentParser()

    arg_set.add_argument('--release', dest='release', help='release useless resource', default=None, type=str)
    arg_set.add_argument('--show', dest='show', help='show resource will release', default=None, type=str)

    args = arg_set.parse_args()
    return args


def get_args():
    """
    获取脚本的入参，并且调用不同的主函数
    :return:
    """
    # 获取入参
    _args = args_set()
    
    _args_dict = {}
    for key, value in _args.__dict__.items():
        # 如果有json格式，则转换为字典类型
        if not value:
            continue
        if all([value.find(':'), value.find('{'), value.find('}')]):
            value = json.loads(value)
        _args_dict[key] = value

    return _args_dict


if __name__ == '__main__':
    get_args()