from app.extensions import db


class City(db.Model):
    """City managed via admin panel."""

    __tablename__ = "cities"

    id = db.Column(db.Integer, primary_key=True, autoincrement=True)
    name = db.Column(db.String(100), nullable=False)
    slug = db.Column(db.String(120), unique=True, nullable=False, index=True)
    state = db.Column(db.String(2), nullable=False)  # UF: SP, RJ, MG, ...

    neighborhoods = db.relationship(
        "Neighborhood",
        backref="city",
        order_by="Neighborhood.name",
        cascade="all, delete-orphan",
        lazy="select",
    )

    def __repr__(self) -> str:
        return f"<City {self.name!r}/{self.state!r}>"


class Neighborhood(db.Model):
    """Neighborhood (bairro) managed via admin panel."""

    __tablename__ = "neighborhoods"

    id = db.Column(db.Integer, primary_key=True, autoincrement=True)
    name = db.Column(db.String(100), nullable=False)
    slug = db.Column(db.String(120), nullable=False, index=True)
    city_id = db.Column(
        db.Integer,
        db.ForeignKey("cities.id", ondelete="CASCADE"),
        nullable=False,
        index=True,
    )

    __table_args__ = (
        db.UniqueConstraint("slug", "city_id", name="uq_neighborhood_slug_city"),
    )

    def __repr__(self) -> str:
        return f"<Neighborhood {self.name!r} city_id={self.city_id}>"