fix: use get_or_create for vertical in import script
This commit is contained in:
parent
6e37c02715
commit
70b48ffcc7
1 changed files with 17 additions and 10 deletions
|
|
@ -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)
|
||||||
|
|
|
||||||
Loading…
Add table
Reference in a new issue