Upload 12 files

database.py

@@ -125,6 +125,13 @@ 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):
+    with get_conn() as conn:
+        conn.execute(
+            "UPDATE users SET pw_hash = ? WHERE username = ?",
+            (pw_hash, username)
+        )
+
 # Videos
 def create_video(filename: str, original_name: str, uploader: str, status: str) -> int:
     with get_conn() as conn:

manage_users.py

@@ -0,0 +1,59 @@
+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
+
+DB_PATH = "/tmp/data/app.db"
+
+def hash_password(password: str) -> str:
+    return bcrypt.hash(password)
+
+def main():
+    set_db_path(DB_PATH)
+    init_schema()
+
+    parser = argparse.ArgumentParser(description="Gestionar usuarios de la aplicación Veureu.")
+    subparsers = parser.add_subparsers(dest="command", required=True)
+
+    # Comando para listar usuarios
+    parser_list = subparsers.add_parser("list", help="Listar todos los usuarios.")
+
+    # Comando para añadir un usuario
+    parser_add = subparsers.add_parser("add", help="Añadir un nuevo usuario.")
+    parser_add.add_argument("username", help="Nombre de usuario.")
+    parser_add.add_argument("password", help="Contraseña del usuario.")
+    parser_add.add_argument("role", help="Rol del usuario (ej: admin, editor, viewer).")
+
+    # Comando para resetear la contraseña
+    parser_reset = subparsers.add_parser("reset-password", help="Restablecer la contraseña de un usuario.")
+    parser_reset.add_argument("username", help="Nombre del usuario a modificar.")
+    parser_reset.add_argument("new_password", help="Nueva contraseña.")
+
+    args = parser.parse_args()
+
+    if args.command == "list":
+        users = get_all_users()
+        if not users:
+            print("No hay usuarios en la base de datos.")
+            return
+        print(f"Usuarios encontrados ({len(users)}):")
+        for user in users:
+            print(f" - ID: {user['id']}, Usuario: {user['username']}, Rol: {user['role']}")
+
+    elif args.command == "add":
+        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)
+        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)
+        print(f"Contraseña del usuario '{args.username}' actualizada con éxito.")
+
+if __name__ == "__main__":
+    main()