b58edfc6eb
- Update session-context.md with exact resume point for next session - Update settings.local.json with broader git permissions - feature/grundstruktur merged to develop - PAT authentication configured Version: 0.2.3
134 lines
4.1 KiB
Python
134 lines
4.1 KiB
Python
import enum
|
|
import uuid
|
|
|
|
from sqlalchemy import (
|
|
Boolean,
|
|
Enum,
|
|
ForeignKey,
|
|
Integer,
|
|
String,
|
|
Text,
|
|
UniqueConstraint,
|
|
)
|
|
from sqlalchemy.orm import Mapped, mapped_column, relationship
|
|
|
|
from app.db.base import Base
|
|
|
|
|
|
class NutrientDemand(str, enum.Enum):
|
|
SCHWACH = "schwach"
|
|
MITTEL = "mittel"
|
|
STARK = "stark"
|
|
|
|
|
|
class WaterDemand(str, enum.Enum):
|
|
WENIG = "wenig"
|
|
MITTEL = "mittel"
|
|
VIEL = "viel"
|
|
|
|
|
|
class CompatibilityRating(str, enum.Enum):
|
|
GUT = "gut"
|
|
NEUTRAL = "neutral"
|
|
SCHLECHT = "schlecht"
|
|
|
|
|
|
class PlantFamily(Base):
|
|
__tablename__ = "plant_families"
|
|
|
|
id: Mapped[uuid.UUID] = mapped_column(
|
|
primary_key=True, default=uuid.uuid4, index=True
|
|
)
|
|
name: Mapped[str] = mapped_column(String(255), nullable=False, unique=True)
|
|
latin_name: Mapped[str] = mapped_column(String(255), nullable=False, unique=True)
|
|
|
|
# Relationships
|
|
plants: Mapped[list["Plant"]] = relationship("Plant", back_populates="family")
|
|
|
|
|
|
class Plant(Base):
|
|
__tablename__ = "plants"
|
|
|
|
id: Mapped[uuid.UUID] = mapped_column(
|
|
primary_key=True, default=uuid.uuid4, index=True
|
|
)
|
|
tenant_id: Mapped[uuid.UUID | None] = mapped_column(
|
|
ForeignKey("tenants.id", ondelete="CASCADE"),
|
|
nullable=True,
|
|
index=True,
|
|
)
|
|
family_id: Mapped[uuid.UUID] = mapped_column(
|
|
ForeignKey("plant_families.id", ondelete="RESTRICT"),
|
|
nullable=False,
|
|
index=True,
|
|
)
|
|
name: Mapped[str] = mapped_column(String(255), nullable=False)
|
|
latin_name: Mapped[str | None] = mapped_column(String(255), nullable=True)
|
|
nutrient_demand: Mapped[NutrientDemand] = mapped_column(
|
|
Enum(NutrientDemand, name="nutrient_demand"),
|
|
nullable=False,
|
|
)
|
|
water_demand: Mapped[WaterDemand] = mapped_column(
|
|
Enum(WaterDemand, name="water_demand"),
|
|
nullable=False,
|
|
)
|
|
spacing_cm: Mapped[int] = mapped_column(Integer, nullable=False)
|
|
sowing_start_month: Mapped[int] = mapped_column(Integer, nullable=False)
|
|
sowing_end_month: Mapped[int] = mapped_column(Integer, nullable=False)
|
|
rest_years: Mapped[int] = mapped_column(Integer, nullable=False, default=0)
|
|
notes: Mapped[str | None] = mapped_column(Text, nullable=True)
|
|
is_active: Mapped[bool] = mapped_column(Boolean, default=True, nullable=False)
|
|
|
|
# Relationships
|
|
tenant: Mapped["Tenant | None"] = relationship( # noqa: F821
|
|
"Tenant", back_populates="plants"
|
|
)
|
|
family: Mapped["PlantFamily"] = relationship("PlantFamily", back_populates="plants")
|
|
plantings: Mapped[list["BedPlanting"]] = relationship( # noqa: F821
|
|
"BedPlanting", back_populates="plant"
|
|
)
|
|
compatibility_a: Mapped[list["PlantCompatibility"]] = relationship(
|
|
"PlantCompatibility",
|
|
foreign_keys="PlantCompatibility.plant_id_a",
|
|
back_populates="plant_a",
|
|
cascade="all, delete-orphan",
|
|
)
|
|
compatibility_b: Mapped[list["PlantCompatibility"]] = relationship(
|
|
"PlantCompatibility",
|
|
foreign_keys="PlantCompatibility.plant_id_b",
|
|
back_populates="plant_b",
|
|
cascade="all, delete-orphan",
|
|
)
|
|
|
|
|
|
class PlantCompatibility(Base):
|
|
__tablename__ = "plant_compatibilities"
|
|
__table_args__ = (
|
|
UniqueConstraint("plant_id_a", "plant_id_b", name="uq_plant_compatibility"),
|
|
)
|
|
|
|
id: Mapped[uuid.UUID] = mapped_column(
|
|
primary_key=True, default=uuid.uuid4, index=True
|
|
)
|
|
plant_id_a: Mapped[uuid.UUID] = mapped_column(
|
|
ForeignKey("plants.id", ondelete="CASCADE"),
|
|
nullable=False,
|
|
)
|
|
plant_id_b: Mapped[uuid.UUID] = mapped_column(
|
|
ForeignKey("plants.id", ondelete="CASCADE"),
|
|
nullable=False,
|
|
)
|
|
rating: Mapped[CompatibilityRating] = mapped_column(
|
|
Enum(CompatibilityRating, name="compatibility_rating"),
|
|
nullable=False,
|
|
)
|
|
reason: Mapped[str | None] = mapped_column(Text, nullable=True)
|
|
|
|
# Relationships
|
|
plant_a: Mapped["Plant"] = relationship(
|
|
"Plant", foreign_keys=[plant_id_a], back_populates="compatibility_a"
|
|
)
|
|
plant_b: Mapped["Plant"] = relationship(
|
|
"Plant", foreign_keys=[plant_id_b], back_populates="compatibility_b"
|
|
)
|