mirror of
https://github.com/PaiGramTeam/PaiGram.git
synced 2024-12-01 11:41:59 +00:00
34 lines
1.0 KiB
Python
34 lines
1.0 KiB
Python
|
import inspect
|
||
|
|
from functools import wraps
|
||
|
|
|
||
|
|
from logger import Log
|
||
|
|
from model.types import Func
|
||
|
|
from utils.app.manager import ServiceDict
|
||
|
|
|
||
|
|
|
||
|
|
def inject(func: Func) -> Func:
|
||
|
|
"""依赖注入"""
|
||
|
|
|
||
|
|
@wraps(func)
|
||
|
|
async def decorator(*args, **kwargs):
|
||
|
|
try:
|
||
|
|
signature = inspect.signature(func)
|
||
|
|
except ValueError as exception:
|
||
|
|
if "no signature found" in str(exception):
|
||
|
|
Log.warning("no signature found", exception)
|
||
|
|
elif "not supported by signature" in str(exception):
|
||
|
|
Log.warning("not supported by signature", exception)
|
||
|
|
else:
|
||
|
|
raise exception
|
||
|
|
else:
|
||
|
|
for parameter_name, parameter in signature.parameters.items():
|
||
|
|
annotation = parameter.annotation
|
||
|
|
class_name = annotation.__name__
|
||
|
|
param = ServiceDict.get(class_name)
|
||
|
|
if param is not None:
|
||
|
|
kwargs.setdefault(parameter_name, param)
|
||
|
|
|
||
|
|
return await func(*args, **kwargs)
|
||
|
|
|
||
|
|
return decorator
|