From 70b48ffcc78d10d6cac0d6edf581b9b127a89a40 Mon Sep 17 00:00:00 2001 From: fullsizemalt <106900403+fullsizemalt@users.noreply.github.com> Date: Fri, 19 Dec 2025 22:14:10 -0800 Subject: [PATCH] fix: use get_or_create for vertical in import script --- backend/import_elgoose.py | 27 +++++++++++++++++---------- 1 file changed, 17 insertions(+), 10 deletions(-) diff --git a/backend/import_elgoose.py b/backend/import_elgoose.py index 362dae5..9b284d7 100644 --- a/backend/import_elgoose.py +++ b/backend/import_elgoose.py @@ -253,17 +253,24 @@ def main(): print("="*60) with Session(engine) as session: - # 1. Create vertical + # 1. Get or create vertical print("\nšŸ¦† Creating Goose vertical...") - vertical = Vertical( - name="Goose", - slug="goose", - description="Goose is a jam band from Connecticut" - ) - session.add(vertical) - session.commit() - session.refresh(vertical) - print(f"āœ“ Created vertical (ID: {vertical.id})") + vertical = session.exec( + select(Vertical).where(Vertical.slug == "goose") + ).first() + + if not vertical: + vertical = Vertical( + name="Goose", + slug="goose", + description="Goose is a jam band from Connecticut" + ) + session.add(vertical) + session.commit() + session.refresh(vertical) + print(f"āœ“ Created vertical (ID: {vertical.id})") + else: + print(f"āœ“ Using existing vertical (ID: {vertical.id})") # 2. Create users users = create_users(session)