profitra/api
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Refactor database seeder to improve error handling and maintainability

Browse Source
This commit is contained in:
Yoga Pangestu 2026-07-17 23:22:09 +07:00
parent 172f035d77
commit 717ec1e061
1 changed files with 13 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
app/seeders/database_seeder.py
View File

@ -9,19 +9,23 @@ from app.seeders.user_seeder import seed_users
def seed_all(db: Session, count: int = 5):
business_types = seed_business_types(db, count)
plans = seed_plans(db, count)
try:
business_types = seed_business_types(db, count)
plans = seed_plans(db, count)
bt_ids = [bt.id for bt in business_types]
tenants = seed_tenants(db, count, business_type_ids=bt_ids)
bt_ids = [bt.id for bt in business_types]
tenants = seed_tenants(db, count, business_type_ids=bt_ids)
plan_ids = [p.id for p in plans]
tenant_ids = [t.id for t in tenants]
seed_subscriptions(db, count, tenant_ids=tenant_ids, plan_ids=plan_ids)
plan_ids = [p.id for p in plans]
tenant_ids = [t.id for t in tenants]
seed_subscriptions(db, count, tenant_ids=tenant_ids, plan_ids=plan_ids)
users = seed_users(db, count)
users = seed_users(db, count)
db.commit()
db.commit()
except:
db.rollback()
raise
return {
"business_types": len(business_types),