SnapFeast / users /schemas.py
Testys's picture
Adding migrations from alembic
d6866b9
from pydantic import BaseModel, EmailStr, Field
from typing import Optional, List
from datetime import datetime
class UserBase(BaseModel):
first_name: str = Field(..., min_length=1, max_length=50)
last_name: str = Field(..., min_length=1, max_length=50)
email: EmailStr
age: Optional[int] = Field(None, ge=0, le=120)
preferences: Optional[List[str]] = None
class UserCreate(UserBase):
password: str = Field(..., min_length=8)
class UserUpdate(BaseModel):
first_name: Optional[str] = Field(None, min_length=1, max_length=50)
last_name: Optional[str] = Field(None, min_length=1, max_length=50)
email: Optional[EmailStr] = None
age: Optional[int] = Field(None, ge=0, le=120)
preferences: Optional[List[str]] = None
password: Optional[str] = Field(None, min_length=8)
class User(UserBase):
id: int
is_active: bool
is_admin: bool
created_at: datetime
updated_at: datetime
class Config:
orm_mode = True
class UserEmbeddingsBase(BaseModel):
embeddings: List[float]
class UserEmbeddingsCreate(UserEmbeddingsBase):
pass
class UserEmbeddings(UserEmbeddingsBase):
id: int
user_id: int
class Config:
orm_mode = True