from data.database import MegaMillionsDraw, PowerballDraw
from util.drawing import get_powerball_object, get_mega_millions_object
from util.environment import load_environment_variables
from util.scrape import project_variables

PROJECT_VARIABLES = load_environment_variables()

RANDOM_SEED = 42
GAME_INFO = {
    'MegaMillions': {
        'game_name': 'mega-millions',
        'table': MegaMillionsDraw,
        'db_table_name': 'mega_millions_draws',
        'inception_year': int(project_variables["LP_MEGA_MILLIONS_INCEPTION"]),
        'rule_change': project_variables["LP_MEGA_MILLIONS_RULE_CHANGE_DATE"],
        'max_game_ball': int(project_variables["LP_MEGA_MILLIONS_MAX_MEGA_BALL"]),
        'max_main_ball': int(project_variables["LP_MEGA_MILLIONS_MAX_MAIN_BALL"]),
        'scrape_name': 'mega-millions',
        'object_function': get_mega_millions_object
    },
    'Powerball': {
        'game_name': 'powerball',
        'table': PowerballDraw,
        'db_table_name': 'powerball_draws',
        'inception_year': int(project_variables["LP_POWERBALL_INCEPTION"]),
        'rule_change': project_variables["LP_POWERBALL_RULE_CHANGE_DATE"],
        'max_game_ball': int(project_variables["LP_POWERBALL_MAX_MAIN_BALL"]),
        'max_main_ball': int(project_variables["LP_POWERBALL_MAX_POWERBALL"]),
        'scrape_name': 'powerball',
        'object_function': get_powerball_object
    }
}