diff --git a/core/base/schema.py b/core/base/schema.py index 86eda86..eaaca40 100644 --- a/core/base/schema.py +++ b/core/base/schema.py @@ -5,6 +5,7 @@ # @Author : xingc # @Desc : import enum +from datetime import date, datetime from typing import Optional, Generic, TypeVar from pydantic import BaseModel @@ -12,6 +13,14 @@ from pydantic import BaseModel T = TypeVar('T') +class BaseMixin(BaseModel): + class Config: + json_encoders = { + datetime: lambda dt: dt.strftime('%Y-%m-%d %H:%M:%S'), + date: lambda dt: dt.strftime('%Y-%m-%d') + } + + class BaseResponse(BaseModel, Generic[T]): status: int = 200 message: str = 'success'