edh-elo/app/sql/__init__.py
2024-02-14 22:43:25 -08:00

16 lines
493 B
Python

from . import models
from .database import SessionLocal
def prime_database():
db = SessionLocal()
win_types = db.query(models.WinType).all()
if not win_types:
db.add(models.WinType(name="Combat Damage"))
db.add(models.WinType(name="Commander Damage"))
db.add(models.WinType(name="Direct Damage"))
db.add(models.WinType(name="Poison"))
db.add(models.WinType(name="Decking"))
db.add(models.WinType(name="other"))
db.commit()