import os
import django
from django.contrib.auth import get_user_model

# Set up Django
os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'manufacture_app.settings')
django.setup()

# Get the User model
User = get_user_model()

# Check if admin user exists and update their user_type if needed
try:
    admin_user = User.objects.get(username='admin')
    print(f"Admin user found: {admin_user.username}")
    print(f"Current user_type: {admin_user.user_type}")
    
    # Set user_type to 'admin' if it's not already
    if admin_user.user_type != 'admin':
        admin_user.user_type = 'admin'
        admin_user.save()
        print("Updated admin user to 'admin' user_type")
    else:
        print("Admin user already has 'admin' user_type")
        
    # Print permissions
    print(f"Has purchase permission: {admin_user.has_purchase_permission()}")
    print(f"Has user management permission: {admin_user.has_user_management_permission()}")
    
except User.DoesNotExist:
    print("Admin user not found")
except Exception as e:
    print(f"Error: {e}")