from typing import Union
from pydantic import BaseModel
from uuid import UUID


class User(BaseModel):
    username: str
    name: Union[str, None] = None


class UserInDB(User):
    id: UUID

    class Config:
        from_attributes = True


class UserRegister(User):
    password: str
    password2: str


class Token(BaseModel):
    access_token: str
    token_type: str


class TokenData(BaseModel):
    username: Union[str, None] = None