| """ |
| Schema 基类和 Mixin |
| 提供通用的字段解析方法，减少重复代码 |
| """ |
|
|
| from datetime import datetime |
| from typing import Any, cast |
|
|
| from django.utils import timezone |
|
|
|
|
| class TimestampMixin: |
| """ |
| 时间戳字段解析 Mixin |
| 为 Schema 提供统一的时间字段处理 |
| """ |
|
|
| @classmethod |
| def _resolve_datetime(cls, value: Any) -> datetime | None: |
| """ |
| 统一处理 datetime 字段，转换为本地时间 |
| |
| Args: |
| value: datetime 对象或 None |
| |
| Returns: |
| 本地化的 datetime 或 None |
| """ |
| if value is None: |
| return None |
| try: |
| return timezone.localtime(value) |
| except (TypeError, AttributeError, ValueError): |
| return cast(datetime | None, value) |
|
|
| @classmethod |
| def _resolve_datetime_iso(cls, value: Any) -> str | None: |
| """ |
| 统一处理 datetime 字段，转换为 ISO 格式字符串 |
| |
| Args: |
| value: datetime 对象或 None |
| |
| Returns: |
| ISO 格式字符串或 None |
| """ |
| if value is None: |
| return None |
| try: |
| local_time = timezone.localtime(value) |
| return local_time.isoformat() |
| except (TypeError, AttributeError, ValueError): |
| return value.isoformat() if hasattr(value, "isoformat") else str(value) |
|
|
|
|
| class DisplayLabelMixin: |
| """ |
| 显示标签解析 Mixin |
| 为 choices 字段提供统一的 label 获取方法 |
| """ |
|
|
| @classmethod |
| def _get_display(cls, obj: Any, field_name: str) -> str | None: |
| """ |
| 获取 choices 字段的显示值 |
| |
| Args: |
| obj: Model 实例 |
| field_name: 字段名 |
| |
| Returns: |
| 显示值或 None |
| """ |
| try: |
| getter = getattr(obj, f"get_{field_name}_display", None) |
| if getter: |
| return cast(str | None, getter()) |
| return getattr(obj, field_name, None) |
| except AttributeError: |
| return None |
|
|
|
|
| class FileFieldMixin: |
| """ |
| 文件字段解析 Mixin |
| 为 FileField 提供统一的 URL 和路径获取方法 |
| """ |
|
|
| @classmethod |
| def _get_file_url(cls, file_field: Any) -> str | None: |
| """ |
| 获取文件的 URL |
| |
| Args: |
| file_field: FileField 实例 |
| |
| Returns: |
| 文件 URL 或 None |
| """ |
| if not file_field: |
| return None |
| try: |
| return cast(str | None, file_field.url) |
| except (AttributeError, ValueError): |
| return None |
|
|
| @classmethod |
| def _get_file_path(cls, file_field: Any) -> str | None: |
| """ |
| 获取文件的路径 |
| |
| Args: |
| file_field: FileField 实例 |
| |
| Returns: |
| 文件路径或 None |
| """ |
| if not file_field: |
| return None |
| try: |
| return cast(str | None, file_field.path) |
| except (AttributeError, ValueError): |
| return None |
|
|
|
|
| |
| class SchemaMixin(TimestampMixin, DisplayLabelMixin, FileFieldMixin): |
| """ |
| Schema 通用 Mixin |
| 组合所有常用的字段解析方法 |
| """ |
|
|
| pass |
|
|
