from pydantic import BaseModel
from typing import Optional


class CityEntity(BaseModel):
    id: int
    name: str
    federal_district_id: Optional[int] = None

    class Config:
        from_attributes = True