函数如何限制入参为指定值之一?

如题,在Python中如果函数的入参可选值是固定的几个值之一,可以通过如下两种方式来校验:

方式1:通过枚举类型

示例如下:

from enum import Enum

# 1. 定义一个自定义类
class LogLevelEnum(Enum):
    level_debug = "DEBUG"
    level1_info = "INFO"
    level1_warning = "WARNING"
    level1_error = "ERROR"
    level1_critical = "CRITICAL"

# 2. 形参level的可选值只能为如上几种之一
def mylogger(level=LogLevelEnum.level_debug):
    # 创建一个 logger 对象
    if not isinstance(level, LogLevelEnum):
        raise TypeError("参数必须是 MyEnum 类型")
        exit(500)
    if level not in LogLevelEnum:
        raise ValueError("参数值必须是 MyEnum 中的选项之一")
        exit(500)
    global logger
    logger = logging.getLogger(__name__)
    # 设置日志级别 DEBUG\INFO\WARNING\ERROR\CRITICAL
    logger.setLevel(level.value)
    # 创建一个文件处理器,将日志信息写入文件
    file_handler = logging.FileHandler('../DebugScripts/logs/my_app.log')
    # 设置日志格式
    formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
    # 将格式器添加到处理器
    file_handler.setFormatter(formatter)
    # 将处理器添加到 logger 对象
    logger.addHandler(file_handler)
    # 创建一个控制台处理器,将日志信息打印到控制台
    console_handler = logging.StreamHandler()
    console_handler.setFormatter(formatter)
    logger.addHandler(console_handler)
    return logger

# 3. 使用:实例化一个枚举类型
log_level = LogLevelEnum.level_debug
print(type(log_level))
# 4. 使用:调用函数,传入枚举类型参数
logger = mylogger(log_level)
logger.debug("hahaha")

说明,通过枚举类来实现这个需求,分4步:
1)通过enum.Enum来创建枚举类
2)待限制入参的函数
3)使用时,实例化一个枚举类型
4)使用时,调用函数是传入枚举类型参数

方式二: 使用typing.Literal

示例如下:

from typing import Literal

# 使用typing.Literal来限制允许的入参值
def mylogger(level: Literal["DEBUG", "INFO", "WARNING", "ERROR", "CRITICAL"]):
    # 创建一个 logger 对象
    # global logger
    logger = logging.getLogger(__name__)
    # 设置日志级别 DEBUG\INFO\WARNING\ERROR\CRITICAL
    logger.setLevel(level)
    # 创建一个文件处理器,将日志信息写入文件
    log_dir = "../logs/"
    os.mkdir(log_dir) if not os.path.exists(log_dir) else None
    file_handler = logging.FileHandler(log_dir+'my_app.log')
    # 设置日志格式
    formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
    # 将格式器添加到处理器
    file_handler.setFormatter(formatter)
    # 将处理器添加到 logger 对象
    logger.addHandler(file_handler)
    # 创建一个控制台处理器,将日志信息打印到控制台
    # console_handler = logging.StreamHandler()
    # console_handler.setFormatter(formatter)
    # logger.addHandler(console_handler)
    return logger

  • 3
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值