fix: use get_or_create for vertical in import script

This commit is contained in:
fullsizemalt 2025-12-19 22:14:10 -08:00
parent 6e37c02715
commit 70b48ffcc7

View file

@ -253,8 +253,13 @@ def main():
print("="*60) print("="*60)
with Session(engine) as session: with Session(engine) as session:
# 1. Create vertical # 1. Get or create vertical
print("\n🦆 Creating Goose vertical...") print("\n🦆 Creating Goose vertical...")
vertical = session.exec(
select(Vertical).where(Vertical.slug == "goose")
).first()
if not vertical:
vertical = Vertical( vertical = Vertical(
name="Goose", name="Goose",
slug="goose", slug="goose",
@ -264,6 +269,8 @@ def main():
session.commit() session.commit()
session.refresh(vertical) session.refresh(vertical)
print(f"✓ Created vertical (ID: {vertical.id})") print(f"✓ Created vertical (ID: {vertical.id})")
else:
print(f"✓ Using existing vertical (ID: {vertical.id})")
# 2. Create users # 2. Create users
users = create_users(session) users = create_users(session)