| """Business logic services.""" |
|
|
| from __future__ import annotations |
|
|
| from collections.abc import Iterable |
| from dataclasses import dataclass |
| from typing import Any, cast |
|
|
| from django.utils.translation import gettext_lazy as _ |
|
|
|
|
| @dataclass(frozen=True) |
| class TemplateBindingAssemblerConfig: |
| sub_type_display: dict[str, str] |
|
|
|
|
| class TemplateBindingAssembler: |
| def __init__(self, config: TemplateBindingAssemblerConfig | None = None) -> None: |
| self._config = config or TemplateBindingAssemblerConfig(sub_type_display={}) |
|
|
| def binding_to_dict(self, *, binding: Any, template: Any) -> dict[str, Any]: |
| return { |
| "binding_id": binding.id, |
| "template_id": binding.template_id, |
| "name": getattr(template, "name", "") if template else "", |
| "description": (getattr(template, "description", "") or "") if template else "", |
| "binding_source": binding.binding_source, |
| "binding_source_display": str(binding.get_binding_source_display()), |
| "created_at": binding.created_at.isoformat() if getattr(binding, "created_at", None) else None, |
| } |
|
|
| def general_to_dict(self, *, template: Any) -> dict[str, Any]: |
| return { |
| "binding_id": None, |
| "template_id": template.id, |
| "name": getattr(template, "name", "") or "", |
| "description": getattr(template, "description", "") or "", |
| "binding_source": "general", |
| "binding_source_display": str(_("通用")), |
| "created_at": None, |
| } |
|
|
| def available_to_dict(self, *, template: Any) -> dict[str, Any]: |
| sub_type = cast(str | None, getattr(template, "case_sub_type", None)) |
| sub_type_key = sub_type or "" |
| return { |
| "template_id": template.id, |
| "name": getattr(template, "name", "") or "", |
| "case_sub_type": sub_type, |
| "case_sub_type_display": self._config.sub_type_display.get(sub_type_key, sub_type_key), |
| } |
|
|
| def categories_response(self, *, grouped: dict[str, list[dict[str, Any]]]) -> dict[str, Any]: |
| categories: list[dict[str, Any]] = [] |
| for category, templates in grouped.items(): |
| categories.append( |
| { |
| "category": category, |
| "category_display": self._config.sub_type_display.get(category, category), |
| "templates": templates, |
| } |
| ) |
| return {"categories": categories, "total_count": sum(len(cat["templates"]) for cat in categories)} |
|
|
| def unified_templates(self, *, items: Iterable[dict[str, Any]]) -> list[dict[str, Any]]: |
| return list(items) |
|
|
