704a8d9a0b
- Add Scene model (Jam, Bluegrass, Dead Family, Funk) - Add VerticalScene join table (many-to-many) - Update Vertical with setlistfm_mbid, is_active, is_featured - Add scenes relationship to Vertical - Add seed_scenes.py script to populate initial data
98 lines
3.2 KiB
Python
98 lines
3.2 KiB
Python
"""Seed script to create initial scenes and assign bands to them"""
|
|
|
|
from sqlmodel import Session, select
|
|
from database import engine
|
|
from models import Scene, Vertical, VerticalScene
|
|
|
|
|
|
# Scene definitions
|
|
SCENES = [
|
|
{"name": "Jam", "slug": "jam", "description": "Improvisational rock bands with extended jams"},
|
|
{"name": "Bluegrass", "slug": "bluegrass", "description": "Progressive and traditional bluegrass"},
|
|
{"name": "Dead Family", "slug": "dead-family", "description": "Grateful Dead and related projects"},
|
|
{"name": "Funk", "slug": "funk", "description": "Funk, soul, and groove-oriented bands"},
|
|
]
|
|
|
|
# Band -> Scene assignments
|
|
BAND_SCENES = {
|
|
"goose": ["jam"],
|
|
"phish": ["jam"],
|
|
"grateful-dead": ["jam", "dead-family"],
|
|
"dead-and-company": ["jam", "dead-family"],
|
|
"billy-strings": ["bluegrass", "jam"],
|
|
# Expansion Wave 1
|
|
"pigeons-playing-ping-pong": ["jam", "funk"],
|
|
"eggy": ["jam"],
|
|
"dogs-in-a-pile": ["jam"],
|
|
"greensky-bluegrass": ["bluegrass", "jam"],
|
|
"daniel-donato": ["jam"],
|
|
# Expansion Wave 2
|
|
"umphreys-mcgee": ["jam"],
|
|
"moe": ["jam"],
|
|
"widespread-panic": ["jam"],
|
|
"sturgill-simpson": ["bluegrass"],
|
|
"slightly-stoopid": ["jam", "funk"],
|
|
}
|
|
|
|
|
|
def seed_scenes():
|
|
"""Create scenes if they don't exist"""
|
|
print("Seeding scenes...")
|
|
|
|
with Session(engine) as session:
|
|
for scene_data in SCENES:
|
|
existing = session.exec(
|
|
select(Scene).where(Scene.slug == scene_data["slug"])
|
|
).first()
|
|
|
|
if not existing:
|
|
scene = Scene(**scene_data)
|
|
session.add(scene)
|
|
print(f" Created scene: {scene_data['name']}")
|
|
else:
|
|
print(f" Scene exists: {scene_data['name']}")
|
|
|
|
session.commit()
|
|
print("Scenes seeded.")
|
|
|
|
|
|
def assign_bands_to_scenes():
|
|
"""Assign existing bands to their scenes"""
|
|
print("\nAssigning bands to scenes...")
|
|
|
|
with Session(engine) as session:
|
|
for band_slug, scene_slugs in BAND_SCENES.items():
|
|
vertical = session.exec(
|
|
select(Vertical).where(Vertical.slug == band_slug)
|
|
).first()
|
|
|
|
if not vertical:
|
|
continue
|
|
|
|
for scene_slug in scene_slugs:
|
|
scene = session.exec(
|
|
select(Scene).where(Scene.slug == scene_slug)
|
|
).first()
|
|
|
|
if not scene:
|
|
continue
|
|
|
|
# Check if assignment exists
|
|
existing = session.exec(
|
|
select(VerticalScene)
|
|
.where(VerticalScene.vertical_id == vertical.id)
|
|
.where(VerticalScene.scene_id == scene.id)
|
|
).first()
|
|
|
|
if not existing:
|
|
vs = VerticalScene(vertical_id=vertical.id, scene_id=scene.id)
|
|
session.add(vs)
|
|
print(f" Assigned {band_slug} -> {scene_slug}")
|
|
|
|
session.commit()
|
|
print("Band scene assignments complete.")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
seed_scenes()
|
|
assign_bands_to_scenes()
|