from django.db import migrations


def add_courses_menu(apps, schema_editor):
    Menu = apps.get_model('menu', 'Menu')
    Menu.objects.get_or_create(
        route='/courses',
        defaults={
            'title': 'دوره‌ها',
            'icon': 'fitness_center',
            'order': 30,
            'is_active': True,
        },
    )


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


class Migration(migrations.Migration):

    dependencies = [
        ('menu', '0002_rename_menuitem_to_menu'),
    ]

    operations = [
        migrations.RunPython(add_courses_menu, remove_courses_menu),
    ]