15 lines
520 B
Python
15 lines
520 B
Python
from app.factories.subscription_factory import SubscriptionFactory
|
|
|
|
|
|
def seed_subscriptions(db, count: int = 5, tenant_ids: list[str] | None = None, plan_ids: list[str] | None = None):
|
|
result = []
|
|
for i in range(count):
|
|
kwargs = {}
|
|
if tenant_ids:
|
|
kwargs["tenant_id"] = tenant_ids[i % len(tenant_ids)]
|
|
if plan_ids:
|
|
kwargs["plan_id"] = plan_ids[i % len(plan_ids)]
|
|
sub = SubscriptionFactory().create(db, **kwargs)
|
|
result.append(sub)
|
|
return result
|