Enhancement: Update db_helper to take in days parameter

This commit is contained in:
2025-10-17 11:35:24 -06:00
parent c1854cc893
commit 7e5a6c35e0

View File

@@ -100,11 +100,11 @@ class DBHelper():
return cursor.fetchall() return cursor.fetchall()
def query_breaks(self) -> list: def query_breaks(self, days) -> list:
self.__CONN = connect_config(self._config) self.__CONN = connect_config(self._config)
self.__CONN.autocommit = True self.__CONN.autocommit = True
cursor = self.__CONN.cursor() 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() self.__CONN.commit()
return cursor.fetchall() return cursor.fetchall()