from pydantic import BaseModel, Field
from typing import Generic, TypeVar, Optional

T = TypeVar('T')


class RespBean(BaseModel, Generic[T]):
    code: str = Field(default="success", description="code")
    message: str = Field(default="success", description='message')
    data: Optional[T] = Field(description="数据", default=None)


def error(message):
    return RespBean(code="error", message=message, data=None)


def success(data=None):
    return RespBean(data=data)