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()