from pydantic import BaseModel
from typing import Optional

class SetPublicKey(BaseModel):
    public_key: str
    
class RefreshRequest(BaseModel):
    refresh_token: str

class SetupAccount(BaseModel):
    first_name: str
    last_name: str
    public_key: str
    encrypted_private_key: str
    
class UserPublic(BaseModel):
    id: int
    username: str
    first_name: Optional[str] = None
    last_name: Optional[str] = None
    public_key: Optional[str] = None

    class Config:
        from_attributes = True