From 7e5a6c35e0d3f98cf1cfc1130648b826b4b1fe6c Mon Sep 17 00:00:00 2001 From: Gabriella Date: Fri, 17 Oct 2025 11:35:24 -0600 Subject: [PATCH] Enhancement: Update `db_helper` to take in days parameter --- src/py/helper/db_helper.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/py/helper/db_helper.py b/src/py/helper/db_helper.py index cf1e105..90efcc6 100644 --- a/src/py/helper/db_helper.py +++ b/src/py/helper/db_helper.py @@ -100,11 +100,11 @@ class DBHelper(): return cursor.fetchall() - def query_breaks(self) -> list: + def query_breaks(self, days) -> list: self.__CONN = connect_config(self._config) self.__CONN.autocommit = True cursor = self.__CONN.cursor() - cursor.execute(f"SELECT * FROM breaks") + cursor.execute(f"SELECT * FROM breaks WHERE date >= NOW() - INTERVAL '1 day' and DATE <= NOW() + interval '{days} days' ORDER BY date ASC;") self.__CONN.commit() return cursor.fetchall()