import sys
import os
from dotenv import load_dotenv

# Add parent directory to path to allow importing from root
sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))

# Load environment variables first
load_dotenv()

from database.connection import get_db_session
from database.models import User, UserRole

def promote_all_users():
    print("Promoting all users to ADMIN...")
    with get_db_session() as db:
        users = db.query(User).all()
        count = 0
        for user in users:
            if user.role != UserRole.ADMIN:
                user.role = UserRole.ADMIN
                count += 1
        
        db.commit()
        print(f"Promoted {count} users to ADMIN.")

if __name__ == "__main__":
    promote_all_users()