mirror of
https://github.com/thecyberlearn/quantumtaskai-caprover.git
synced 2026-08-18 15:32:56 +00:00
b0e8c917ef
Complete Django AI agent marketplace with security optimizations and clean deployment documentation without any API keys or secrets. 🤖 Generated with [Claude Code](https://claude.ai/code) Co-Authored-By: Claude <noreply@anthropic.com>
111 lines
4.2 KiB
Python
111 lines
4.2 KiB
Python
from django.core.management.base import BaseCommand
|
|
from django.contrib.auth import get_user_model
|
|
import secrets
|
|
import getpass
|
|
|
|
User = get_user_model()
|
|
|
|
|
|
class Command(BaseCommand):
|
|
help = 'Check and fix admin user status'
|
|
|
|
def add_arguments(self, parser):
|
|
parser.add_argument(
|
|
'--password',
|
|
type=str,
|
|
help='Admin password (if not provided, will generate secure random password)'
|
|
)
|
|
parser.add_argument(
|
|
'--prompt-password',
|
|
action='store_true',
|
|
help='Prompt for password input (secure)'
|
|
)
|
|
parser.add_argument(
|
|
'--check-only',
|
|
action='store_true',
|
|
help='Only check user status, do not reset password'
|
|
)
|
|
|
|
def handle(self, *args, **options):
|
|
# Check both possible admin emails (preferred email first)
|
|
possible_emails = ['admin@quantumtaskai.com', 'admin@netcop.ai']
|
|
username = 'admin'
|
|
|
|
# Secure password handling
|
|
if options['check_only']:
|
|
password = None
|
|
elif options['prompt_password']:
|
|
password = getpass.getpass("Enter admin password: ")
|
|
if not password:
|
|
self.stdout.write(self.style.ERROR("Password cannot be empty"))
|
|
return
|
|
elif options['password']:
|
|
password = options['password']
|
|
else:
|
|
# Generate secure random password
|
|
password = secrets.token_urlsafe(16)
|
|
self.stdout.write(f"🔐 Generated secure password: {password}")
|
|
self.stdout.write("⚠️ SAVE THIS PASSWORD SECURELY - it will not be shown again!")
|
|
|
|
user = None
|
|
found_email = None
|
|
|
|
# Try to find existing admin user
|
|
for email in possible_emails:
|
|
try:
|
|
user = User.objects.get(email=email)
|
|
found_email = email
|
|
break
|
|
except User.DoesNotExist:
|
|
continue
|
|
|
|
if user:
|
|
# User found with one of the emails
|
|
self.stdout.write(f"✅ User found: {user.email}")
|
|
self.stdout.write(f"Username: {user.username}")
|
|
self.stdout.write(f"Is superuser: {user.is_superuser}")
|
|
self.stdout.write(f"Is staff: {user.is_staff}")
|
|
self.stdout.write(f"Is active: {user.is_active}")
|
|
self.stdout.write(f"Email verified: {user.email_verified}")
|
|
|
|
# Fix user permissions if needed
|
|
if not user.is_superuser or not user.is_staff:
|
|
user.is_superuser = True
|
|
user.is_staff = True
|
|
user.is_active = True
|
|
user.save()
|
|
self.stdout.write("🔧 Fixed user permissions")
|
|
|
|
# Reset password to ensure it's correct (only if password provided)
|
|
if password:
|
|
user.set_password(password)
|
|
user.save()
|
|
self.stdout.write("🔑 Password reset successfully")
|
|
|
|
# Show login instructions
|
|
self.stdout.write("\n📝 Login Instructions:")
|
|
self.stdout.write(f"URL: https://www.quantumtaskai.com/admin/")
|
|
self.stdout.write(f"Email: {found_email}")
|
|
self.stdout.write(f"Username: {username}")
|
|
if password:
|
|
self.stdout.write(f"Password: {password}")
|
|
else:
|
|
self.stdout.write("Password: (not changed - use existing password)")
|
|
|
|
else:
|
|
self.stdout.write("❌ Admin user not found! Creating new admin user...")
|
|
|
|
# Create new admin user with preferred email
|
|
preferred_email = 'admin@quantumtaskai.com'
|
|
user = User.objects.create_superuser(
|
|
username=username,
|
|
email=preferred_email,
|
|
password=password,
|
|
)
|
|
user.add_balance(100, "Initial admin balance")
|
|
|
|
self.stdout.write("✅ New admin user created successfully!")
|
|
self.stdout.write(f"Email: {preferred_email}")
|
|
self.stdout.write(f"Username: {username}")
|
|
self.stdout.write(f"Password: {password}")
|
|
self.stdout.write(f"Balance: {user.wallet_balance} AED") |