from sqlalchemy.orm import Mapped, mapped_column

from dmp.adapters.db import ReportingModel


class Country(ReportingModel):
    __tablename__ = "location_country"

    code2: Mapped[str] = mapped_column(primary_key=True)
    code3: Mapped[str]
    name: Mapped[str]


class Region(ReportingModel):
    __tablename__ = "location_region"

    code: Mapped[str] = mapped_column(primary_key=True)
    country_code2: Mapped[str] = mapped_column(primary_key=True)
    name: Mapped[str]


class City(ReportingModel):
    __tablename__ = "location_city"

    id: Mapped[str] = mapped_column(primary_key=True)
    country_code2: Mapped[str] = mapped_column(primary_key=True)
    name: Mapped[str]
    name_native: Mapped[str]
    region_code: Mapped[str]
    latitude: Mapped[float]
    longitude: Mapped[float]