sass-imobiliaria/backend/app/models/location.py

46 lines
1.3 KiB
Python

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}>"