from django.db import migrations


def add_reports_menu(apps, schema_editor):
    Menu = apps.get_model('menu', 'Menu')
    Menu.objects.get_or_create(
        route='/reports',
        defaults={
            'title': 'گزارش‌ها',
            'icon': 'assessment',
            'order': 40,
            'is_active': True,
        },
    )


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


class Migration(migrations.Migration):

    dependencies = [
        ('menu', '0003_add_courses_menu'),
    ]

    operations = [
        migrations.RunPython(add_reports_menu, remove_reports_menu),
    ]