Upload 11 files

app.py

@@ -68,6 +68,34 @@ os.makedirs(DATA_DIR, exist_ok=True)
 ensure_dirs(DATA_DIR)
 DB_PATH = os.path.join(DATA_DIR, "app.db")
 set_db_path(DB_PATH)
+
+# Comprobación de sanidad: si la BD tiene el esquema viejo, se borra.
+def force_db_recreation_if_needed():
+    db_file = Path(DB_PATH)
+    if not db_file.exists():
+        return # No hay BD, no hay nada que hacer
+
+    try:
+        conn = sqlite3.connect(DB_PATH)
+        cursor = conn.cursor()
+        # Comprobar si la columna 'pw_hash' existe
+        cursor.execute("PRAGMA table_info(users)")
+        columns = [row[1] for row in cursor.fetchall()]
+        conn.close()
+
+        if 'pw_hash' in columns:
+            print("Detectada estructura de BD obsoleta. Eliminando para forzar la recreación...")
+            db_file.unlink()
+            st.rerun() # Forzar reinicio de la app
+    except Exception as e:
+        # La tabla puede no existir, o ser un fichero corrupto. En cualquier caso, lo borramos.
+        print(f"Error al comprobar la BD ({e}). Se intentará borrar y recrear.")
+        if db_file.exists():
+            db_file.unlink()
+            st.rerun()
+
+force_db_recreation_if_needed()
+
 init_schema()
 
 def create_default_users_if_needed():
@@ -82,8 +110,8 @@ def create_default_users_if_needed():
         ]
         for username, password, role in users_to_create:
             try:
-                pw_hash = bcrypt.hash(password)
-                create_user(username, pw_hash, role)
+                password_hash = bcrypt.hash(password)
+                create_user(username, password_hash, role)
                 print(f"Usuario '{username}' creado con rol '{role}'.")
             except Exception as e:
                 print(f"Error al crear el usuario {username}: {e}")
@@ -150,8 +178,7 @@ if not st.session_state.user:
             row = get_user(username)
             hash_to_check = None
             if row:
-                # Comprobar cualquiera de las dos columnas de hash de contraseña por compatibilidad
-                hash_to_check = row["password_hash"] or row["pw_hash"]
+                hash_to_check = row["password_hash"]
 
             if row and hash_to_check and verify_password(password, hash_to_check):
                 st.session_state.user = {"id": row["id"], "username": row["username"], "role": row["role"]}

database.py

@@ -35,12 +35,16 @@ def init_schema():
         CREATE TABLE IF NOT EXISTS users (
             id INTEGER PRIMARY KEY AUTOINCREMENT,
             username TEXT UNIQUE NOT NULL,
-            pw_hash TEXT,               -- o password_hash, deja las dos si convives con ambos esquemas
             password_hash TEXT,
             role TEXT NOT NULL,
             created_at TEXT NOT NULL
         );
         """)
+        # Migración: eliminar la columna antigua si existe
+        try:
+            c.execute("ALTER TABLE users DROP COLUMN pw_hash;")
+        except sqlite3.OperationalError:
+            pass # La columna no existía, no hay nada que hacer.
         # (opcional: tus otras tablas)
 
         # >>> NUEVA TABLA PARA FEEDBACK DE AD (no depende de videos)
@@ -108,11 +112,11 @@ def now_str():
     return datetime.utcnow().isoformat(timespec="seconds") + "Z"
 
 # Users
-def create_user(username: str, pw_hash: str, role: str):
+def create_user(username: str, password_hash: str, role: str):
     with get_conn() as conn:
         conn.execute(
             "INSERT INTO users(username, password_hash, role, created_at) VALUES (?,?,?,?)",
-            (username, pw_hash, role, now_str()),
+            (username, password_hash, role, now_str()),
         )
 
 def get_user(username: str):
@@ -125,11 +129,11 @@ def get_all_users() -> List[Dict[str, Any]]:
         cur = conn.execute("SELECT id, username, role FROM users ORDER BY username")
         return cur.fetchall()
 
-def update_user_password(username: str, pw_hash: str):
+def update_user_password(username: str, password_hash: str):
     with get_conn() as conn:
         conn.execute(
-            "UPDATE users SET pw_hash = ? WHERE username = ?",
-            (pw_hash, username)
+            "UPDATE users SET password_hash = ? WHERE username = ?",
+            (password_hash, username)
         )
 
 # Videos

scripts/manage_users.py

@@ -2,8 +2,11 @@ import argparse
 from passlib.hash import bcrypt
 from database import set_db_path, init_schema, get_all_users, create_user, get_user, update_user_password
 
-# Apuntar a la base de datos de origen para que los cambios persistan en el build
-DB_PATH = "init_data/veureu.db"
+# Apuntar a la base de datos de ejecución en el entorno de HF Spaces
+import os
+# La ruta debe coincidir con la configurada en config.yaml y app.py
+DATA_DIR = os.getenv("DATA_DIR_OVERRIDE", "/tmp/data")
+DB_PATH = os.path.join(DATA_DIR, "app.db")
 
 def hash_password(password: str) -> str:
     return bcrypt.hash(password)
@@ -44,16 +47,16 @@ def main():
         if get_user(args.username):
             print(f"Error: El usuario '{args.username}' ya existe.")
             return
-        pw_hash = hash_password(args.password)
-        create_user(args.username, pw_hash, args.role)
+        password_hash = hash_password(args.password)
+        create_user(args.username, password_hash, args.role)
         print(f"Usuario '{args.username}' creado con éxito.")
 
     elif args.command == "reset-password":
         if not get_user(args.username):
             print(f"Error: El usuario '{args.username}' no existe.")
             return
-        pw_hash = hash_password(args.new_password)
-        update_user_password(args.username, pw_hash)
+        password_hash = hash_password(args.new_password)
+        update_user_password(args.username, password_hash)
         print(f"Contraseña del usuario '{args.username}' actualizada con éxito.")
 
 if __name__ == "__main__":