diff --git a/backend/db/users.py b/backend/db/users.py index a8eb4d4..ed6bad5 100644 --- a/backend/db/users.py +++ b/backend/db/users.py @@ -3,9 +3,21 @@ from db.db import get_db conn = get_db() -def get_users(): +def get_users(search): + # Get search param + search_param = f"%{search}%" + print(search_param) + + # Generate cursor cursor = conn.cursor() - cursor.execute("SELECT * FROM Employees") + + # Search if needed + if search: + cursor.execute("SELECT * FROM Employees WHERE username LIKE ?", (search_param,)) + else: + cursor.execute("SELECT * FROM Employees") + + # Fetch users from database users = cursor.fetchall() return users diff --git a/backend/main.py b/backend/main.py index bb0e077..f1aaa17 100644 --- a/backend/main.py +++ b/backend/main.py @@ -16,8 +16,8 @@ app.add_middleware( ) @app.get("/") -def list_users(): - return get_users() +def list_users(s: str = None): + return get_users(s) @app.get("/{eployee_id}") def list_attendance(eployee_id: int):