From 0e710f09c592d88c5c314acfd2589db5e7e312ca Mon Sep 17 00:00:00 2001 From: Chris Smith Date: Thu, 5 Mar 2026 10:56:43 -0500 Subject: [PATCH] Add check_table_name function --- util/drawing.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/util/drawing.py b/util/drawing.py index e1df284..b99952a 100644 --- a/util/drawing.py +++ b/util/drawing.py @@ -8,6 +8,14 @@ from datetime import datetime, timedelta from data.database import PowerballDraw, MegaMillionsDraw +def check_table_name(table_name: str) -> bool: + """ + :param table_name: table name + :return: True if table name is valid + """ + return table_name in ['PowerballDraw', 'MegaMillionsDraw'] + + def get_next_draw_date(game: str | None, from_date: str | None = None) -> str | None: """ Gets the next drawing date from the current or specified date for the given game