Update check_table_name function to pass tests

util/drawing.py

@@ -3,17 +3,35 @@
 
     Utility functions for drawing data processing
 """
+import os
+
 from datetime import datetime, timedelta
+from dotenv import load_dotenv
 
 from data.database import PowerballDraw, MegaMillionsDraw
 
+load_dotenv()
 
-def check_table_name(table_name: str) -> bool:
+
+def check_table_name(table_name: str) -> dict | None:
     """
     :param table_name: table name
-    :return: True if table name is valid
+    :return: dict of game info or None
     """
-    return table_name in ['PowerballDraw', 'MegaMillionsDraw']
+    game_dict: dict = {
+        'PowerballDraw': {
+            'game_name': 'powerball',
+            'db_table_name': 'powerball_draws',
+            'inception_year': os.getenv('LP_POWERBALL_INCEPTION'),
+        },
+        'MegaMillionsDraw': {
+            'game_name': 'mega-millions',
+            'db_table_name': 'mega_millions_draws',
+            'inception_year': os.getenv('LP_MEGA_MILLIONS_INCEPTION'),
+        },
+    }
+
+    return game_dict[table_name] if table_name in game_dict.keys() else None
 
 
 def get_next_draw_date(game: str | None, from_date: str | None = None) -> str | None: