23d36e7903
JSON definitions and idempotent seed function for the 3 universal evolution tracks (Batter, Starting Pitcher, Relief Pitcher) with locked threshold values. Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
63 lines
2.0 KiB
Python
63 lines
2.0 KiB
Python
"""Seed script for EvolutionTrack records.
|
|
|
|
Loads track definitions from evolution_tracks.json and upserts them into the
|
|
database using get_or_create keyed on name. Existing tracks have their
|
|
thresholds and formula updated to match the JSON in case values have changed.
|
|
|
|
Can be run standalone:
|
|
python -m app.seed.evolution_tracks
|
|
"""
|
|
|
|
import json
|
|
from pathlib import Path
|
|
|
|
from app.db_engine import EvolutionTrack
|
|
|
|
_JSON_PATH = Path(__file__).parent / "evolution_tracks.json"
|
|
|
|
|
|
def seed_evolution_tracks() -> list[EvolutionTrack]:
|
|
"""Upsert evolution tracks from JSON seed data.
|
|
|
|
Returns a list of EvolutionTrack instances that were created or updated.
|
|
"""
|
|
raw = _JSON_PATH.read_text(encoding="utf-8")
|
|
track_defs = json.loads(raw)
|
|
|
|
results: list[EvolutionTrack] = []
|
|
|
|
for defn in track_defs:
|
|
track, created = EvolutionTrack.get_or_create(
|
|
name=defn["name"],
|
|
defaults={
|
|
"card_type": defn["card_type"],
|
|
"formula": defn["formula"],
|
|
"t1_threshold": defn["t1_threshold"],
|
|
"t2_threshold": defn["t2_threshold"],
|
|
"t3_threshold": defn["t3_threshold"],
|
|
"t4_threshold": defn["t4_threshold"],
|
|
},
|
|
)
|
|
|
|
if not created:
|
|
# Update mutable fields in case the JSON values changed.
|
|
track.card_type = defn["card_type"]
|
|
track.formula = defn["formula"]
|
|
track.t1_threshold = defn["t1_threshold"]
|
|
track.t2_threshold = defn["t2_threshold"]
|
|
track.t3_threshold = defn["t3_threshold"]
|
|
track.t4_threshold = defn["t4_threshold"]
|
|
track.save()
|
|
|
|
action = "created" if created else "updated"
|
|
print(f" [{action}] {track.name} (card_type={track.card_type})")
|
|
results.append(track)
|
|
|
|
return results
|
|
|
|
|
|
if __name__ == "__main__":
|
|
print("Seeding evolution tracks...")
|
|
tracks = seed_evolution_tracks()
|
|
print(f"Done. {len(tracks)} track(s) processed.")
|