mirror of
https://github.com/nonebot/nonebot2.git
synced 2024-11-24 09:05:04 +08:00
57 lines
1.7 KiB
Python
57 lines
1.7 KiB
Python
"""
|
|
FrontMatter:
|
|
sidebar_position: 1
|
|
description: nonebot.dependencies.utils 模块
|
|
"""
|
|
|
|
import inspect
|
|
from typing import Any, Callable, ForwardRef
|
|
|
|
from loguru import logger
|
|
|
|
from nonebot.exception import TypeMisMatch
|
|
from nonebot.typing import evaluate_forwardref
|
|
from nonebot.compat import DEFAULT_CONFIG, ModelField, model_field_validate
|
|
|
|
|
|
def get_typed_signature(call: Callable[..., Any]) -> inspect.Signature:
|
|
"""获取可调用对象签名"""
|
|
|
|
signature = inspect.signature(call)
|
|
globalns = getattr(call, "__globals__", {})
|
|
typed_params = [
|
|
inspect.Parameter(
|
|
name=param.name,
|
|
kind=param.kind,
|
|
default=param.default,
|
|
annotation=get_typed_annotation(param, globalns),
|
|
)
|
|
for param in signature.parameters.values()
|
|
]
|
|
return inspect.Signature(typed_params)
|
|
|
|
|
|
def get_typed_annotation(param: inspect.Parameter, globalns: dict[str, Any]) -> Any:
|
|
"""获取参数的类型注解"""
|
|
|
|
annotation = param.annotation
|
|
if isinstance(annotation, str):
|
|
annotation = ForwardRef(annotation)
|
|
try:
|
|
annotation = evaluate_forwardref(annotation, globalns, globalns)
|
|
except Exception as e:
|
|
logger.opt(colors=True, exception=e).warning(
|
|
f'Unknown ForwardRef["{param.annotation}"] for parameter {param.name}'
|
|
)
|
|
return inspect.Parameter.empty
|
|
return annotation
|
|
|
|
|
|
def check_field_type(field: ModelField, value: Any) -> Any:
|
|
"""检查字段类型是否匹配"""
|
|
|
|
try:
|
|
return model_field_validate(field, value, DEFAULT_CONFIG)
|
|
except ValueError:
|
|
raise TypeMisMatch(field, value)
|