Add files via upload

Author: GitTimeraider

debug_users.py

@@ -0,0 +1,28 @@
+#!/usr/bin/env python3
+from app.main import create_app
+from app.models import db, User
+
+app = create_app()
+
+with app.app_context():
+    print("\n=== User Debug Info ===")
+    users = User.query.all()
+
+    for user in users:
+        print(f"\nUser: {user.username}")
+        print(f"  ID: {user.id}")
+        print(f"  Admin: {user.is_admin}")
+        print(f"  Has encryption key: {bool(user.encryption_key)}")
+        print(f"  DA Server: {user.da_server}")
+        print(f"  DA Username: {user.da_username}")
+        print(f"  Has DA Password: {bool(user.da_password_encrypted)}")
+        print(f"  DA Domain: {user.da_domain}")
+
+        # Fix missing encryption keys
+        if not user.encryption_key:
+            print("  [!] Generating missing encryption key...")
+            user.generate_encryption_key()
+            db.session.commit()
+            print("  [✓] Encryption key generated")
+
+print("\n=== Debug complete ===")

migrate_add_admin.py

@@ -0,0 +1,14 @@
+from app.main import create_app
+from app.models import db, User
+
+app = create_app()
+
+with app.app_context():
+    # Add is_admin column to existing users
+    users = User.query.all()
+    for user in users:
+        if not hasattr(user, 'is_admin'):
+            user.is_admin = user.username == 'admin'  # Make 'admin' user an admin
+
+    db.session.commit()
+    print("Migration completed. Admin flags added to users.")

migrate_da_settings.py

@@ -0,0 +1,29 @@
+from app.main import create_app
+from app.models import db, User
+from app.config import Config
+
+app = create_app()
+
+with app.app_context():
+    # Add new columns to existing users
+    users = User.query.all()
+
+    # If global settings exist, migrate them to admin user
+    admin = User.query.filter_by(username='admin').first()
+    if admin and hasattr(Config, 'DA_SERVER'):
+        print("Migrating global DirectAdmin settings to admin user...")
+        admin.da_server = Config.DA_SERVER
+        admin.da_username = Config.DA_USERNAME
+        admin.da_domain = Config.DA_DOMAIN
+
+        # Generate encryption key if not exists
+        if not admin.encryption_key:
+            from cryptography.fernet import Fernet
+            admin.encryption_key = Fernet.generate_key().decode()
+
+        # Encrypt and store password
+        if hasattr(Config, 'DA_PASSWORD'):
+            admin.set_da_password(Config.DA_PASSWORD)
+
+    db.session.commit()
+    print("Migration completed!")