mirror of
https://github.com/nonebot/nonebot2.git
synced 2024-11-24 17:15:05 +08:00
59 lines
1.6 KiB
Python
59 lines
1.6 KiB
Python
"""
|
|
FrontMatter:
|
|
sidebar_position: 1
|
|
description: nonebot.dependencies.utils 模块
|
|
"""
|
|
|
|
import inspect
|
|
from typing import Any, Dict, TypeVar, Callable, ForwardRef
|
|
|
|
from loguru import logger
|
|
from pydantic.fields import ModelField
|
|
from pydantic.typing import evaluate_forwardref
|
|
|
|
from nonebot.exception import TypeMisMatch
|
|
|
|
V = TypeVar("V")
|
|
|
|
|
|
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: V) -> V:
|
|
"""检查字段类型是否匹配"""
|
|
|
|
_, errs_ = field.validate(value, {}, loc=())
|
|
if errs_:
|
|
raise TypeMisMatch(field, value)
|
|
return value
|