Migrate to motor for DB interaction

api/schemas/user.py

@@ -0,0 +1,99 @@
+from enum import Enum
+from typing import Optional
+
+from pydantic import BaseModel, Field, validator, field_validator
+
+
+def validate_username(value: str):
+    length = len(value)
+    if length < 4 or length > 32:
+        raise ValueError("Username must be between 4 and 32 characters long")
+    if any(not (x.isalnum() or x == "_" or x == " ") for x in value):
+        raise ValueError("Username must only contain letters, numbers, underscores, and dashes")
+    return value
+
+
+def validate_password(value: str):
+    length = len(value)
+    if length < 8 or length > 16:
+        raise ValueError("Password must be between 8 and 16 characters long")
+    return value
+
+
+class AuthLevel(Enum):
+    GUEST = 0
+    USER = 1
+    ADMIN = 2
+
+    def __lt__(self, other):
+        if self.__class__ is other.__class__:
+            return self.value < other.value
+        return NotImplemented
+
+    def __gt__(self, other):
+        if self.__class__ is other.__class__:
+            return self.value > other.value
+        return NotImplemented
+
+    def __eq__(self, other):
+        if self.__class__ is other.__class__:
+            return self.value == other.value
+        return NotImplemented
+
+
+class UserBaseSchema(BaseModel):
+    username: str
+
+
+class UserLoginSchema(UserBaseSchema):
+    password: str
+
+
+class UserCreateSchema(UserBaseSchema):
+    password: str
+    level: AuthLevel = Field(AuthLevel.USER)
+
+    @field_validator("username")
+    @classmethod
+    def _valid_username(cls, value):
+        validate_username(value)
+
+    @field_validator("password")
+    @classmethod
+    def _valid_password(cls, value):
+        validate_password(value)
+
+
+class UserUpdateSchema(BaseModel):
+    username: Optional[str] = None
+    password: Optional[str] = None
+    level: Optional[AuthLevel] = AuthLevel.USER
+
+    @field_validator("username")
+    @classmethod
+    def _valid_username(cls, value):
+        validate_username(value)
+
+    @field_validator("password")
+    @classmethod
+    def _valid_password(cls, value):
+        validate_password(value)
+
+
+class UserDisplaySchema(UserBaseSchema):
+    id: str
+    level: AuthLevel
+
+
+class UserSystemSchema(UserDisplaySchema):
+    password: str
+
+
+class TokenSchema(BaseModel):
+    access_token: str
+    refresh_token: str
+
+
+class TokenPayload(BaseModel):
+    sub: Optional[str]
+    exp: Optional[int]