fonrey/apps/region/models/region.py

from django.db import models

from core.enums import SchoolLevel, SchoolNature, SchoolType
from core.models.base import TimeStampedModel


class District(TimeStampedModel):
    city = models.CharField(max_length=50)
    name = models.CharField(max_length=50)
    short_name = models.CharField(max_length=20, blank=True, default="")
    sort_order = models.IntegerField(default=0)
    is_active = models.BooleanField(default=True)

    class Meta:
        db_table = "districts"
        verbose_name = "城区"
        verbose_name_plural = "城区"
        constraints = [
            models.UniqueConstraint(
                fields=["city", "name"],
                condition=models.Q(is_active=True),
                name="uq_districts_city_name",
            ),
        ]
        ordering = ["city", "sort_order", "name"]

    def __str__(self) -> str:
        return f"{self.city} / {self.name}"


class BusinessArea(TimeStampedModel):
    district = models.ForeignKey(
        "region.District",
        on_delete=models.PROTECT,
        related_name="business_areas",
    )
    name = models.CharField(max_length=100)
    sort_order = models.IntegerField(default=0)
    latitude = models.DecimalField(max_digits=10, decimal_places=7, null=True, blank=True)
    longitude = models.DecimalField(max_digits=10, decimal_places=7, null=True, blank=True)
    is_active = models.BooleanField(default=True)

    class Meta:
        db_table = "business_areas"
        verbose_name = "商圈"
        verbose_name_plural = "商圈"
        constraints = [
            models.UniqueConstraint(
                fields=["district", "name"],
                name="uq_business_areas_name",
            ),
        ]
        indexes = [
            models.Index(
                fields=["district"],
                name="idx_business_areas_district",
                condition=models.Q(is_active=True),
            ),
        ]
        ordering = ["district_id", "sort_order", "name"]

    def __str__(self) -> str:
        return self.name


class MetroLine(TimeStampedModel):
    city = models.CharField(max_length=50)
    name = models.CharField(max_length=50)
    color = models.CharField(max_length=7, blank=True, default="")
    sort_order = models.IntegerField(default=0)
    is_active = models.BooleanField(default=True)

    class Meta:
        db_table = "metro_lines"
        verbose_name = "地铁线路"
        verbose_name_plural = "地铁线路"
        ordering = ["city", "sort_order", "name"]

    def __str__(self) -> str:
        return f"{self.city} {self.name}"


class MetroStation(TimeStampedModel):
    metro_line = models.ForeignKey(
        "region.MetroLine",
        on_delete=models.CASCADE,
        related_name="stations",
    )
    name = models.CharField(max_length=50)
    latitude = models.DecimalField(max_digits=10, decimal_places=7, null=True, blank=True)
    longitude = models.DecimalField(max_digits=10, decimal_places=7, null=True, blank=True)
    sort_order = models.IntegerField(default=0)
    is_active = models.BooleanField(default=True)

    class Meta:
        db_table = "metro_stations"
        verbose_name = "地铁站点"
        verbose_name_plural = "地铁站点"
        indexes = [
            models.Index(
                fields=["metro_line"],
                name="idx_metro_stations_line",
                condition=models.Q(is_active=True),
            ),
        ]
        ordering = ["metro_line_id", "sort_order"]

    def __str__(self) -> str:
        return self.name


class School(TimeStampedModel):
    district = models.ForeignKey(
        "region.District",
        null=True,
        blank=True,
        on_delete=models.SET_NULL,
        related_name="schools",
    )
    name = models.CharField(max_length=100)
    type = models.CharField(
        max_length=20,
        blank=True,
        default="",
        choices=SchoolType.choices,
    )
    nature = models.CharField(
        max_length=20,
        blank=True,
        default="",
        choices=SchoolNature.choices,
    )
    level = models.CharField(
        max_length=20,
        blank=True,
        default="",
        choices=SchoolLevel.choices,
    )
    is_active = models.BooleanField(default=True)

    class Meta:
        db_table = "schools"
        verbose_name = "学校"
        verbose_name_plural = "学校"
        indexes = [
            models.Index(
                fields=["district"],
                name="idx_schools_district",
                condition=models.Q(is_active=True),
            ),
        ]
        ordering = ["name"]

    def __str__(self) -> str:
        return self.name