create_admin.py

import os
import sys
from getpass import getpass
from pymongo import MongoClient
from dotenv import load_dotenv

# Add the project root to the Python path to allow importing project modules
sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))

# Now we can import from our project
from security import get_password_hash

# Load environment variables from .env file
load_dotenv()
MONGO_URL = os.getenv("MONGO_URL")

# Connect to the database
client = MongoClient(MONGO_URL)
db = client["memora_db"]
users_collection = db["users"]

def main():
    print("--- Create Admin User ---")
    
    # Get admin username and password from user input
    username = input("Enter admin username: ").strip()
    password = getpass("Enter admin password: ")
    
    # Check if the user already exists
    if users_collection.find_one({"username": username}):
        print(f"Error: User '{username}' already exists.")
        return
        
    # Hash the password
    hashed_password = get_password_hash(password)
    
    # Create the new admin user document
    admin_user = {
        "username": username,
        "hashed_password": hashed_password,
        "is_admin": True
    }
    
    # Insert the new user into the database
    result = users_collection.insert_one(admin_user)
    print(f"Successfully created admin user '{username}' with ID: {result.inserted_id}")

if __name__ == "__main__":
    main()