Files
Reviflow/fix_db_roles.py

27 lines
761 B
Python

import sqlite3
import os
db_path = 'reviflow.db'
if not os.path.exists(db_path):
print(f"Database {db_path} not found")
exit(1)
conn = sqlite3.connect(db_path)
cursor = conn.cursor()
# Check current values
cursor.execute("SELECT DISTINCT role FROM users")
roles = cursor.fetchall()
print(f"Current roles in DB: {roles}")
# Update to uppercase to match Enum member names if that's what's causing the lookup error
cursor.execute("UPDATE users SET role = 'PARENT' WHERE role = 'parent'")
print(f"Updated {cursor.rowcount} 'parent' rows to 'PARENT'")
cursor.execute("UPDATE users SET role = 'LEARNER' WHERE role = 'learner'")
print(f"Updated {cursor.rowcount} 'learner' rows to 'LEARNER'")
conn.commit()
conn.close()
print("Database update complete")