mirror of
https://github.com/TeamPGM/PagerMaid-Pyro.git
synced 2024-11-16 13:01:21 +00:00
23 lines
787 B
Python
23 lines
787 B
Python
import inspect
|
|
import pagermaid.enums as enums
|
|
import pagermaid.services as services
|
|
from typing import Dict, Optional
|
|
|
|
|
|
def inject(message: enums.Message, function, **data) -> Optional[Dict]:
|
|
try:
|
|
signature = inspect.signature(function)
|
|
except Exception:
|
|
return None
|
|
for parameter_name, parameter in signature.parameters.items():
|
|
an = parameter.annotation
|
|
class_name = an.__name__ if hasattr(an, "__name__") else an
|
|
param = message if class_name == "Message" else services.get(class_name)
|
|
if not param:
|
|
if parameter_name == "message":
|
|
param = message
|
|
else:
|
|
param = services.get(parameter_name.capitalize())
|
|
data.setdefault(parameter_name, param)
|
|
return data
|