from django.db import migrations


def add_attendance_report_menu(apps, schema_editor):
    Menu = apps.get_model('menu', 'Menu')
    Menu.objects.get_or_create(
        route='/reports/attendance',
        defaults={
            'title': 'گزارش حضور',
            'icon': 'how_to_reg',
            'order': 41,
            'is_active': True,
        },
    )


def remove_attendance_report_menu(apps, schema_editor):
    Menu = apps.get_model('menu', 'Menu')
    Menu.objects.filter(route='/reports/attendance').delete()


class Migration(migrations.Migration):

    dependencies = [
        ('menu', '0004_add_reports_menu'),
    ]

    operations = [
        migrations.RunPython(add_attendance_report_menu, remove_attendance_report_menu),
    ]