ProjectHentai/nonebot2
6
0
Fork 0
mirror of https://github.com/nonebot/nonebot2.git synced 2024-11-28 03:15:24 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

✨ support use MessageSegment method as template format spec

Browse Source
This commit is contained in:
Mix 2021-10-03 16:26:55 +08:00
parent 69f8821acb
commit 6b79106cd1
1 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
nonebot/adapters/_template.py
View File

@ -1,3 +1,4 @@
import inspect
import functools import functools
from string import Formatter from string import Formatter
from typing import (TYPE_CHECKING, Any, Set, List, Type, Tuple, Union, Generic, from typing import (TYPE_CHECKING, Any, Set, List, Type, Tuple, Union, Generic,
@ -114,8 +115,14 @@ class MessageTemplate(Formatter, Generic[TM]):
[""])), auto_arg_index [""])), auto_arg_index
def format_field(self, value: Any, format_spec: str) -> Any: def format_field(self, value: Any, format_spec: str) -> Any:
return super().format_field(value, segment_class: Type[MessageSegment] = self.factory.get_segment_class()
format_spec) if format_spec else value method = getattr(segment_class, format_spec, None)
method_type = inspect.getattr_static(segment_class, format_spec, None)
return (super().format_field(value, format_spec) if
((method is None) or
(not isinstance(method_type, (classmethod, staticmethod))
) # Only Call staticmethod or classmethod
) else method(value)) if format_spec else value
def _add(self, a: Any, b: Any) -> Any: def _add(self, a: Any, b: Any) -> Any:
try: try: