import django
import os
import sys

def add_menu_items():
    from apps.menus.models import Menu

    # Check if "Master Data" parent exists
    master_data, created = Menu.objects.get_or_create(
        name="Master Data",
        defaults={
            "route": "",
            "icon": "fa-solid fa-database",
            "sort_order": 8,
            "permission_code": "",
            "is_active": True,
            "admin_only": False,
        },
    )

    # Check if "Master Skills" exists
    master_skills, created = Menu.objects.get_or_create(
        name="Master Skills",
        parent=master_data,
        defaults={
            "route": "/master-skills",
            "icon": "fa-solid fa-lightbulb",
            "sort_order": 1,
            "permission_code": "master_skills.view_masterskill",
            "is_active": True,
            "admin_only": False,
        },
    )

    print("Menu items checked/added successfully!")

if __name__ == "__main__":
    # Set up Django
    sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
    os.environ.setdefault("DJANGO_SETTINGS_MODULE", "config.settings.dev")
    django.setup()
    
    add_menu_items()