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