feat: update hand-made orm to sqlmodel (UserDatabase)

This commit is contained in:
estrella
2023-07-30 21:33:29 +08:00
parent 9ae187a05d
commit 3ca9a9737f
2 changed files with 48 additions and 45 deletions

View File

@@ -1,14 +1,24 @@
from pydantic import BaseModel, Field
from typing import Optional
from sqlmodel import SQLModel, Field
class User(BaseModel):
class User(SQLModel, table=True):
id: int = Field(default=None, primary_key=True)
username: str = Field(
"admin", min_length=4, max_length=20, regex=r"^[a-zA-Z0-9_]+$"
)
password: str = Field("adminadmin", min_length=8)
class UserLogin(BaseModel):
class UserUpdate(SQLModel):
username: Optional[str] = Field(
None, min_length=4, max_length=20, regex=r"^[a-zA-Z0-9_]+$"
)
password: Optional[str] = Field(None, min_length=8)
class UserLogin(SQLModel):
username: str
password: str = Field(..., min_length=8)