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,17 +253,24 @@ 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 = Vertical( vertical = session.exec(
name="Goose", select(Vertical).where(Vertical.slug == "goose")
slug="goose", ).first()
description="Goose is a jam band from Connecticut"
) if not vertical:
session.add(vertical) vertical = Vertical(
session.commit() name="Goose",
session.refresh(vertical) slug="goose",
print(f"✓ Created vertical (ID: {vertical.id})") 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 # 2. Create users
users = create_users(session) users = create_users(session)