#!/usr/bin/env python3

import pymysql
import json

try:
    # Connect to the database
    conn = pymysql.connect(
        host='localhost',
        database='trasabilitate',
        user='trasabilitate',
        password='Initial01!',
        cursorclass=pymysql.cursors.DictCursor
    )
    
    with conn.cursor() as cursor:
        # Update Ciprian's role from quality_manager to manager
        print("Updating Ciprian's role from 'quality_manager' to 'manager'...")
        cursor.execute("UPDATE users SET role = 'manager' WHERE username = 'Ciprian'")
        
        # Assign quality module to Ciprian since he was a quality manager
        quality_modules = json.dumps(['quality'])
        print(f"Assigning quality module to Ciprian: {quality_modules}")
        cursor.execute("UPDATE users SET modules = %s WHERE username = 'Ciprian'", (quality_modules,))
        
        # Commit the changes
        conn.commit()
        print("Database updated successfully!")
        
        # Show updated users
        print("\nUpdated users:")
        print("-" * 50)
        cursor.execute("SELECT id, username, role, modules FROM users")
        users = cursor.fetchall()
        
        for user in users:
            print(f"ID: {user['id']}")
            print(f"Username: {user['username']}")
            print(f"Role: {user['role']}")
            print(f"Modules: {user['modules']}")
            print("-" * 30)
            
finally:
    conn.close()