37 lines
1.2 KiB
Python
37 lines
1.2 KiB
Python
from sqlalchemy.orm import Session
|
|
from app import models
|
|
|
|
INITIAL_CARDS = [
|
|
{"name": "Goblin Grunt", "rarity": "COMMON"},
|
|
{"name": "Elven Archer", "rarity": "COMMON"},
|
|
{"name": "Orc Warrior", "rarity": "COMMON"},
|
|
{"name": "Dwarven Shield", "rarity": "COMMON"},
|
|
{"name": "Healing Potion", "rarity": "COMMON"},
|
|
{"name": "Fireball Scroll", "rarity": "RARE"},
|
|
{"name": "Silver Sword", "rarity": "RARE"},
|
|
{"name": "Mystic Amulet", "rarity": "RARE"},
|
|
{"name": "Dragon Scale", "rarity": "LEGENDARY"},
|
|
{"name": "King's Crown", "rarity": "LEGENDARY"},
|
|
]
|
|
|
|
INITIAL_CHESTS = [
|
|
{"name": "Wooden Chest", "cost_gold": 100, "cards_per_open": 3}
|
|
]
|
|
|
|
def seed_data(db: Session):
|
|
# Seed Cards
|
|
for card_data in INITIAL_CARDS:
|
|
exists = db.query(models.Card).filter_by(name=card_data["name"]).first()
|
|
if not exists:
|
|
card = models.Card(**card_data)
|
|
db.add(card)
|
|
|
|
# Seed Chests
|
|
for chest_data in INITIAL_CHESTS:
|
|
exists = db.query(models.Chest).filter_by(name=chest_data["name"]).first()
|
|
if not exists:
|
|
chest = models.Chest(**chest_data)
|
|
db.add(chest)
|
|
|
|
db.commit()
|