Build Dockerized MVP scaffold and records importer

2026-09-02 20:29:02 +07:00
parent a6f91a1329
commit d3a45248ef
39 changed files with 6638 additions and 13 deletions
@@ -0,0 +1,111 @@
+from __future__ import annotations
+
+import enum
+import uuid
+from datetime import datetime, time
+
+from sqlalchemy import JSON, DateTime, Enum, ForeignKey, Integer, String, Text, Time, UniqueConstraint
+from sqlalchemy.orm import Mapped, mapped_column, relationship
+
+from .database import Base
+
+
+class SourceType(str, enum.Enum):
+    official_record = "official_record"
+    user = "user"
+    manual_import = "manual_import"
+
+
+class ModerationStatus(str, enum.Enum):
+    pending = "pending"
+    approved = "approved"
+    rejected = "rejected"
+
+
+class BaitKind(str, enum.Enum):
+    bait = "bait"
+    lure = "lure"
+    unknown = "unknown"
+
+
+class ImportStatus(str, enum.Enum):
+    running = "running"
+    success = "success"
+    partial = "partial"
+    failed = "failed"
+
+
+class Fish(Base):
+    __tablename__ = "fish"
+    id: Mapped[uuid.UUID] = mapped_column(primary_key=True, default=uuid.uuid4)
+    slug: Mapped[str] = mapped_column(String(100), unique=True)
+    name_ru: Mapped[str] = mapped_column(String(200), unique=True)
+    trophy_weight_g: Mapped[int | None]
+
+
+class Waterbody(Base):
+    __tablename__ = "waterbody"
+    id: Mapped[uuid.UUID] = mapped_column(primary_key=True, default=uuid.uuid4)
+    slug: Mapped[str] = mapped_column(String(100), unique=True)
+    name_ru: Mapped[str] = mapped_column(String(200), unique=True)
+    unlock_level: Mapped[int | None]
+
+
+class Bait(Base):
+    __tablename__ = "bait"
+    id: Mapped[uuid.UUID] = mapped_column(primary_key=True, default=uuid.uuid4)
+    name: Mapped[str] = mapped_column(String(200))
+    normalized_name: Mapped[str] = mapped_column(String(200), unique=True)
+    kind: Mapped[BaitKind] = mapped_column(Enum(BaitKind))
+
+
+class Spot(Base):
+    __tablename__ = "spot"
+    __table_args__ = (UniqueConstraint("waterbody_id", "x", "y"),)
+    id: Mapped[uuid.UUID] = mapped_column(primary_key=True, default=uuid.uuid4)
+    waterbody_id: Mapped[uuid.UUID] = mapped_column(ForeignKey("waterbody.id"))
+    x: Mapped[int]
+    y: Mapped[int]
+    description: Mapped[str | None] = mapped_column(Text)
+    waterbody: Mapped[Waterbody] = relationship()
+
+
+class CatchReport(Base):
+    __tablename__ = "catch_report"
+    id: Mapped[uuid.UUID] = mapped_column(primary_key=True, default=uuid.uuid4)
+    fish_id: Mapped[uuid.UUID] = mapped_column(ForeignKey("fish.id"))
+    spot_id: Mapped[uuid.UUID | None] = mapped_column(ForeignKey("spot.id"))
+    waterbody_id: Mapped[uuid.UUID] = mapped_column(ForeignKey("waterbody.id"))
+    bait_id: Mapped[uuid.UUID | None] = mapped_column(ForeignKey("bait.id"))
+    weight_g: Mapped[int]
+    fishing_method: Mapped[str | None] = mapped_column(String(50))
+    rig_type: Mapped[str | None] = mapped_column(String(100))
+    retrieve_method: Mapped[str | None] = mapped_column(String(100))
+    retrieve_speed: Mapped[int | None]
+    game_time: Mapped[time | None] = mapped_column(Time)
+    caught_at: Mapped[datetime | None] = mapped_column(DateTime(timezone=True))
+    reported_at: Mapped[datetime] = mapped_column(DateTime(timezone=True))
+    player_name: Mapped[str | None] = mapped_column(String(100))
+    source_type: Mapped[SourceType] = mapped_column(Enum(SourceType))
+    source_url: Mapped[str | None] = mapped_column(Text)
+    source_external_id: Mapped[str | None] = mapped_column(String(64), unique=True)
+    source_confidence: Mapped[int]
+    moderation_status: Mapped[ModerationStatus] = mapped_column(Enum(ModerationStatus))
+    raw_payload: Mapped[dict | None] = mapped_column(JSON)
+    fish: Mapped[Fish] = relationship()
+    spot: Mapped[Spot | None] = relationship()
+    waterbody: Mapped[Waterbody] = relationship()
+    bait: Mapped[Bait | None] = relationship()
+
+
+class OfficialRecordImport(Base):
+    __tablename__ = "official_record_import"
+    id: Mapped[uuid.UUID] = mapped_column(primary_key=True, default=uuid.uuid4)
+    started_at: Mapped[datetime] = mapped_column(DateTime(timezone=True))
+    finished_at: Mapped[datetime | None] = mapped_column(DateTime(timezone=True))
+    status: Mapped[ImportStatus] = mapped_column(Enum(ImportStatus))
+    source_url: Mapped[str] = mapped_column(Text)
+    rows_seen: Mapped[int] = mapped_column(default=0)
+    rows_created: Mapped[int] = mapped_column(default=0)
+    rows_updated: Mapped[int] = mapped_column(default=0)
+    error_summary: Mapped[str | None] = mapped_column(Text)