如题,在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