# Generated by Django 5.2.4 on 2026-04-23 16:15

from django.db import migrations
import json


def add_tt_student_set_module(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_student_set_module" not in tables:
        tables.append("tt_student_set_module")
        setting.value = json.dumps(tables)
        setting.save()


def reverse_add_tt_student_set_module(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_student_set_module" in tables:
        tables.remove("tt_student_set_module")
        setting.value = json.dumps(tables)
        setting.save()

class Migration(migrations.Migration):

    dependencies = [
        ('api', '0057_remove_tt_student_from_import_tables'),
    ]

    operations = [
        migrations.RunPython(add_tt_student_set_module, reverse_add_tt_student_set_module),
    ]