from django.db import migrations


def seed_company_max_team_members(apps, schema_editor):
    SystemSetting = apps.get_model("contract", "SystemSetting")
    SystemSetting.objects.get_or_create(
        key="COMPANY_MAX_TEAM_MEMBERS",
        defaults={
            "value": "10",
            "description": "Maximum active members plus pending invitations allowed per company workspace.",
            "is_active": True,
        },
    )


def unseed_company_max_team_members(apps, schema_editor):
    SystemSetting = apps.get_model("contract", "SystemSetting")
    SystemSetting.objects.filter(key="COMPANY_MAX_TEAM_MEMBERS", value="10").delete()


class Migration(migrations.Migration):

    dependencies = [
        ("contract", "0040_systemsetting"),
    ]

    operations = [
        migrations.RunPython(seed_company_max_team_members, unseed_company_max_team_members),
    ]