# Generated by Django 5.2.4 on 2026-04-24 15:42

from django.db import migrations
import json

def add_tt_booking(apps, schema_editor):
    TtSetting = apps.get_model("api", "TtSetting")

    setting = TtSetting.objects.filter(param="import_tables").first()
    if not setting:
        return

    try:
        tables = json.loads(setting.value)
    except Exception:
        tables = []

    if "tt_booking" not in tables:
        tables.append("tt_booking")
        setting.value = json.dumps(tables)
        setting.save()


def reverse_add_tt_booking(apps, schema_editor):
    TtSetting = apps.get_model("api", "TtSetting")

    setting = TtSetting.objects.filter(param="import_tables").first()
    if not setting:
        return

    try:
        tables = json.loads(setting.value)
    except Exception:
        tables = []

    if "tt_booking" in tables:
        tables.remove("tt_booking")
        setting.value = json.dumps(tables)
        setting.save()


class Migration(migrations.Migration):

    dependencies = [
        ('api', '0096_cron'),
    ]

    operations = [
        migrations.RunPython(add_tt_booking, reverse_add_tt_booking),
    ]