create_admin.py

import os
import sys
from dotenv import load_dotenv
from flask import Flask
from app.database import db, init_app
from app.models.user import User

def create_admin(username, email, password):
    """Create an admin user"""
    # Load environment variables
    load_dotenv()
    
    # Create Flask app
    app = Flask(__name__, 
                template_folder='app/templates',
                static_folder='app/static')
    
    # Configure app
    app.config['SECRET_KEY'] = os.getenv('SECRET_KEY', 'dev-secret-key')
    app.config['JWT_SECRET_KEY'] = os.getenv('JWT_SECRET_KEY', 'dev-jwt-secret')
    
    # Initialize database
    init_app(app)
    
    # Check if user exists
    with app.app_context():
        user = User.query.filter_by(username=username).first()
        if user:
            print(f"User '{username}' already exists.")
            if user.role == 'admin':
                print("User already has admin role.")
                return True
            else:
                # Update user to admin
                user.role = 'admin'
                db.session.commit()
                print(f"User '{username}' updated to admin role.")
                return True
        else:
            # Create new admin user
            admin = User(
                username=username,
                email=email,
                password=password,
                role='admin'
            )
            db.session.add(admin)
            db.session.commit()
            print(f"Admin user '{username}' created successfully.")
            return True

def main():
    """Main function to create admin user"""
    if len(sys.argv) < 4:
        print("Usage: python create_admin.py <username> <email> <password>")
        return 1
    
    username = sys.argv[1]
    email = sys.argv[2]
    password = sys.argv[3]
    
    # Validate inputs (basic validation)
    if not username or not email or not password:
        print("Error: Username, email, and password are required.")
        return 1
    
    if len(password) < 8:
        print("Error: Password must be at least 8 characters.")
        return 1
    
    if '@' not in email:
        print("Error: Invalid email format.")
        return 1
    
    # Create admin user
    if create_admin(username, email, password):
        return 0
    else:
        return 1

if __name__ == "__main__":
    sys.exit(main())