Update src/services/auth_service.py

src/services/auth_service.py

@@ -1,9 +1,6 @@
-import sqlite3
 from passlib.context import CryptContext
-
 from src.db.connection import get_conn
 
-# Password hashing context (bcrypt)
 PWD_CTX = CryptContext(
     schemes=["bcrypt"],
     deprecated="auto",
@@ -11,13 +8,6 @@ PWD_CTX = CryptContext(
 
 
 def authenticate_user(db_path: str, username: str, password: str):
-    """
-    Authenticate a user against the database.
-
-    Returns:
-        dict {username, role} on success
-        None on failure
-    """
     if not username or not password:
         return None
 
@@ -27,15 +17,12 @@ def authenticate_user(db_path: str, username: str, password: str):
             """
             SELECT username, role, password_hash, is_active
             FROM users
-            WHERE username = ?
+            WHERE username=?
             """,
             (username,),
         ).fetchone()
 
-        if not row:
-            return None
-
-        if not row["is_active"]:
+        if not row or not row["is_active"]:
             return None
 
         if not PWD_CTX.verify(password, row["password_hash"]):
@@ -45,15 +32,11 @@ def authenticate_user(db_path: str, username: str, password: str):
             "username": row["username"],
             "role": row["role"],
         }
-
     finally:
         conn.close()
+
+
 def is_admin(auth_state) -> bool:
-    """
-    Returns True if the current user is an admin.
-    Expects auth_state to be a dict like:
-    { "username": "...", "role": "admin" }
-    """
     if not auth_state:
         return False
     return auth_state.get("role") == "admin"