mirror of
https://github.com/fastapi-practices/fastapi-best-architecture.git
synced 2026-09-21 13:12:24 +00:00
50 lines
1.5 KiB
Python
50 lines
1.5 KiB
Python
from datetime import datetime
|
|
from typing import Annotated, Any
|
|
|
|
from pydantic import BaseModel, ConfigDict, EmailStr, Field, validate_email
|
|
|
|
from backend.common.enums import PrimaryKeyType
|
|
from backend.core.conf import settings
|
|
from backend.utils.timezone import timezone
|
|
|
|
CustomPhoneNumber = Annotated[str, Field(pattern=r'^1[3-9]\d{9}$')]
|
|
|
|
|
|
class CustomEmailStr(EmailStr):
|
|
"""自定义邮箱类型"""
|
|
|
|
@classmethod
|
|
def _validate(cls, input_value: str, /) -> str | None:
|
|
return None if not input_value else validate_email(input_value)[1]
|
|
|
|
|
|
class SchemaBase(BaseModel):
|
|
"""基础模型配置"""
|
|
|
|
model_config = ConfigDict(
|
|
use_enum_values=True,
|
|
json_encoders={
|
|
datetime: lambda x: (
|
|
timezone.to_str(timezone.from_datetime(x))
|
|
if x.tzinfo is not None and x.tzinfo != timezone.tz_info
|
|
else timezone.to_str(x)
|
|
),
|
|
},
|
|
)
|
|
|
|
if PrimaryKeyType.snowflake == settings.DATABASE_PK_MODE:
|
|
from pydantic import field_serializer
|
|
|
|
# 详情：https://fastapi-practices.github.io/fastapi_best_architecture_docs/backend/reference/pk.html#%E6%B3%A8%E6%84%8F%E4%BA%8B%E9%A1%B9
|
|
@field_serializer('id', check_fields=False)
|
|
def serialize_id(self, value: int) -> str | int:
|
|
if self.model_config.get('from_attributes'):
|
|
return str(value)
|
|
return value
|
|
|
|
|
|
def ser_string(value: Any) -> str | None:
|
|
if value:
|
|
return str(value)
|
|
return value
|