62 lines
3.1 KiB
Python
62 lines
3.1 KiB
Python
from __future__ import annotations
|
|
|
|
from datetime import datetime, timedelta, timezone
|
|
from uuid import UUID
|
|
|
|
from sqlalchemy import select
|
|
|
|
from .database import SessionLocal
|
|
from .models import Bait, BaitKind, CatchReport, Fish, ModerationStatus, SourceType, Spot, Waterbody
|
|
|
|
|
|
IDS = {
|
|
"vyunok": UUID("10000000-0000-0000-0000-000000000001"),
|
|
"kuori": UUID("10000000-0000-0000-0000-000000000002"),
|
|
"pike": UUID("20000000-0000-0000-0000-000000000001"),
|
|
"trout": UUID("20000000-0000-0000-0000-000000000002"),
|
|
"spiker": UUID("30000000-0000-0000-0000-000000000001"),
|
|
"shad": UUID("30000000-0000-0000-0000-000000000002"),
|
|
"spot1": UUID("40000000-0000-0000-0000-000000000001"),
|
|
"spot2": UUID("40000000-0000-0000-0000-000000000002"),
|
|
}
|
|
|
|
|
|
def seed() -> None:
|
|
with SessionLocal.begin() as db:
|
|
if db.scalar(select(Fish.id).limit(1)) is not None:
|
|
return
|
|
vyunok = Waterbody(id=IDS["vyunok"], slug="vyunok", name_ru="Вьюнок", unlock_level=1)
|
|
kuori = Waterbody(id=IDS["kuori"], slug="kuori", name_ru="Куори", unlock_level=16)
|
|
pike = Fish(id=IDS["pike"], slug="pike", name_ru="Щука", trophy_weight_g=10_000)
|
|
trout = Fish(id=IDS["trout"], slug="lake-trout", name_ru="Озёрная форель", trophy_weight_g=10_000)
|
|
spiker = Bait(id=IDS["spiker"], name="Spiker #2 01-015", normalized_name="spiker #2 01-015", kind=BaitKind.lure)
|
|
shad = Bait(id=IDS["shad"], name="Salmon T1 Shad 12 005", normalized_name="salmon t1 shad 12 005", kind=BaitKind.lure)
|
|
spot1 = Spot(id=IDS["spot1"], waterbody=vyunok, x=110, y=103, description="Кромка травы у северного берега")
|
|
spot2 = Spot(id=IDS["spot2"], waterbody=kuori, x=85, y=92, description="Свальчик в глубину")
|
|
db.add_all([vyunok, kuori, pike, trout, spiker, shad, spot1, spot2])
|
|
now = datetime.now(timezone.utc)
|
|
for index in range(12):
|
|
db.add(CatchReport(
|
|
fish=pike, spot=spot1, waterbody=vyunok, bait=spiker,
|
|
weight_g=2_600 + index * 480, fishing_method="spinning",
|
|
retrieve_method="равномерная", retrieve_speed=22,
|
|
caught_at=now - timedelta(minutes=25 + index * 47),
|
|
reported_at=now - timedelta(minutes=20 + index * 47),
|
|
player_name=f"DemoPlayer{index % 7 + 1}", source_type=SourceType.manual_import,
|
|
source_confidence=80 + index % 3 * 5, moderation_status=ModerationStatus.approved,
|
|
))
|
|
for index in range(5):
|
|
db.add(CatchReport(
|
|
fish=trout, spot=spot2, waterbody=kuori, bait=shad,
|
|
weight_g=4_200 + index * 900, fishing_method="spinning",
|
|
retrieve_method="ступенчатая", retrieve_speed=18,
|
|
caught_at=now - timedelta(hours=2 + index * 4),
|
|
reported_at=now - timedelta(hours=2 + index * 4),
|
|
player_name=f"DemoAngler{index + 1}", source_type=SourceType.manual_import,
|
|
source_confidence=85, moderation_status=ModerationStatus.approved,
|
|
))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
seed()
|