Django_Basic_Manufacturing/update_admin_user.py

33 lines
1.0 KiB
Python

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}")