api/app/seeders/tenant_seeder.py

13 lines
421 B
Python

from app.factories.tenant_factory import TenantFactory
def seed_tenants(db, count: int = 5, business_type_ids: list[str] | None = None):
result = []
for i in range(count):
kwargs = {}
if business_type_ids:
kwargs["business_type_id"] = business_type_ids[i % len(business_type_ids)]
tenant = TenantFactory().create(db, **kwargs)
result.append(tenant)
return result