Change to new util function for table name test, add get_recent_draw_date function

This commit is contained in:
chris committed 2026-03-05 11:04:19 -05:00
1 parent 52944a53ab
commit 72a1ce6b81
1 file changed
+13 -2
+13 -2
View File
@@ -20,10 +20,21 @@ project_variables = envutil.load_environment_variables()
def get_record_count(table_name: str) -> str | None:
"""get the most recent draw record from the specified table"""
if table_name in ['PowerballDraw', 'MegaMillionsDraw']:
if drawutil.check_table_name(table_name):
dal.connect()
session = dal.Session()
target_table = Base.registry._class_registry.get(table_name)
return session.query(func.count(target_table.draw_date)).first()[0]
else:
raise ValueError('Valid values are PowerballDraw, MegaMillionsDraw')
raise ValueError('An invalid table name was specified.')
def get_recent_draw_date(table_name: str) -> datetime | None:
"""get the most recent draw date from the specified table"""
if drawutil.check_table_name(table_name):
dal.connect()
session = dal.Session()
target_table = Base.registry._class_registry.get(table_name)
return session.query(func.max(target_table.draw_date)).first()[0]
else:
raise ValueError('An invalid table name was specified.')