feat(webui): use get_class_field_docs for schema field descriptions

pull/1496/head
DrSmoothl 2026-02-17 16:58:59 +08:00
parent 1631774452
commit 19c9c5a39a
No known key found for this signature in database
1 changed files with 121 additions and 0 deletions

View File

@ -0,0 +1,121 @@
import inspect
from typing import Any, get_args, get_origin
from pydantic_core import PydanticUndefined
from src.config.config_base import ConfigBase
class ConfigSchemaGenerator:
@classmethod
def generate_schema(cls, config_class: type[ConfigBase], include_nested: bool = True) -> dict[str, Any]:
return cls.generate_config_schema(config_class, include_nested=include_nested)
@classmethod
def generate_config_schema(cls, config_class: type[ConfigBase], include_nested: bool = True) -> dict[str, Any]:
fields: list[dict[str, Any]] = []
nested: dict[str, dict[str, Any]] = {}
for field_name, field_info in config_class.model_fields.items():
if field_name in {"field_docs", "_validate_any", "suppress_any_warning"}:
continue
field_schema = cls._build_field_schema(config_class, field_name, field_info.annotation, field_info)
fields.append(field_schema)
if include_nested:
nested_schema = cls._build_nested_schema(field_info.annotation)
if nested_schema is not None:
nested[field_name] = nested_schema
return {
"className": config_class.__name__,
"classDoc": (config_class.__doc__ or "").strip(),
"fields": fields,
"nested": nested,
}
@classmethod
def _build_nested_schema(cls, annotation: Any) -> dict[str, Any] | None:
origin = get_origin(annotation)
args = get_args(annotation)
if inspect.isclass(annotation) and issubclass(annotation, ConfigBase):
return cls.generate_config_schema(annotation)
if origin in {list, tuple} and args:
first = args[0]
if inspect.isclass(first) and issubclass(first, ConfigBase):
return cls.generate_config_schema(first)
return None
@classmethod
def _build_field_schema(
cls, config_class: type[ConfigBase], field_name: str, annotation: Any, field_info: Any
) -> dict[str, Any]:
field_docs = config_class.get_class_field_docs()
field_type = cls._map_field_type(annotation)
schema: dict[str, Any] = {
"name": field_name,
"type": field_type,
"label": field_name,
"description": field_docs.get(field_name, field_info.description or ""),
"required": field_info.is_required(),
}
if field_info.default is not PydanticUndefined:
schema["default"] = field_info.default
origin = get_origin(annotation)
args = get_args(annotation)
if origin is list and args:
schema["items"] = {"type": cls._map_field_type(args[0])}
options = cls._extract_options(annotation)
if options:
schema["options"] = options
return schema
@staticmethod
def _extract_options(annotation: Any) -> list[str] | None:
origin = get_origin(annotation)
if origin is None:
return None
if str(origin) != "typing.Literal":
return None
args = get_args(annotation)
options = [str(item) for item in args]
return options or None
@classmethod
def _map_field_type(cls, annotation: Any) -> str:
origin = get_origin(annotation)
args = get_args(annotation)
if origin in {list, tuple}:
return "array"
if inspect.isclass(annotation) and issubclass(annotation, ConfigBase):
return "object"
if annotation is bool:
return "boolean"
if annotation is int:
return "integer"
if annotation is float:
return "number"
if annotation is str:
return "string"
if origin in {list, tuple} and args:
return "array"
if origin in {dict}:
return "object"
if origin is not None and str(origin) == "typing.Literal":
return "select"
return "string"