From fa002d679118c03c21bc64772cf51595f31a70fb Mon Sep 17 00:00:00 2001 From: parberge Date: Tue, 30 Jun 2020 22:01:50 +0200 Subject: [PATCH] Match part of event_date for users lock Add the possibility to list locks for a user for a specific year --- app.py | 2 +- chalicelib/lib/db_v2.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/app.py b/app.py index 5d31c12..d7640a8 100644 --- a/app.py +++ b/app.py @@ -258,7 +258,7 @@ def get_lock(user_id, event_date): :param event_date: :return: """ - return db_v2.get_lock_by_user_id_and_date(user_id=user_id, event_date=event_date) + return db_v2.get_locks_by_user_id_and_date(user_id=user_id, event_date=event_date) @app.route("/lock", methods=["POST"], cors=True) diff --git a/chalicelib/lib/db_v2.py b/chalicelib/lib/db_v2.py index ef7f449..829bbb7 100644 --- a/chalicelib/lib/db_v2.py +++ b/chalicelib/lib/db_v2.py @@ -138,14 +138,14 @@ def delete_all_locks_by_user_id(user_id): ) -def get_lock_by_user_id_and_date(user_id, event_date): +def get_locks_by_user_id_and_date(user_id, event_date): """ :param user_id: :param event_date: :return: lock for user """ scan = LockTable.scan( - (LockTable.user_id == user_id) & (LockTable.event_date == event_date) + (LockTable.user_id == user_id) & (LockTable.event_date.startswith(event_date)) ) locks = [] for lock in scan: