ProjectHentai/nonebot2
6
0
Fork 0
mirror of https://github.com/nonebot/nonebot2.git synced 2025-02-08 03:46:18 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

♻️ refactor template format_field to improve readability

Browse Source
This commit is contained in:
Mix 2022-01-02 13:13:43 +08:00
parent 983e5aefdb
commit 43938a004e
1 changed files with 9 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
nonebot/adapters/_template.py
View File

@ -171,27 +171,16 @@ class MessageTemplate(Formatter, Generic[TF]):
) )
def format_field(self, value: Any, format_spec: str) -> Any: def format_field(self, value: Any, format_spec: str) -> Any:
if issubclass(self.factory, str): formatter: Optional[FormatSpecFunc] = self.format_specs.get(format_spec)
return super().format_field(value, format_spec) if (formatter is None) and (not issubclass(self.factory, str)):
segment_class: Type["MessageSegment"] = self.factory.get_segment_class()
segment_class: Type[MessageSegment] = self.factory.get_segment_class()
method = getattr(segment_class, format_spec, None) method = getattr(segment_class, format_spec, None)
method_type = inspect.getattr_static(segment_class, format_spec, None) if inspect.ismethod(method):
formatter = getattr(segment_class, format_spec)
return ( return (
(
super().format_field(value, format_spec) super().format_field(value, format_spec)
if ( if formatter is None
method is None else formatter(value)
or not isinstance(method_type, (classmethod, staticmethod))
)
else (
self.format_specs[format_spec](value)
if format_spec in self.format_specs
else method(value)
)
)
if format_spec
else value
) )
def _add(self, a: Any, b: Any) -> Any: def _add(self, a: Any, b: Any) -> Any: