""" File contains util functions. """ from datetime import datetime from config import DakinyConfig def get_datetime(day: str) -> datetime: """ Takes in a string month and day and returns a datetime object of that Month/Day this year. Args: day (str): Month/Day in that format. Returns: datetime: Datetime representation of Month/Day. """ year = datetime.today().year month = int(day.split("/")[0]) day = int(day.split("/")[1]) # ensure that year rolls over if current month is December and the next month is less than 12. if datetime.today().month == 12 & month < 12: year += 1 return datetime(year, month, day) def get_election_cycle_name(config: DakinyConfig) -> str: """ Builds the unique name of the current election cycle Returns: str: the name of the election cycle """ cycle = get_current_cycle(config) current_cycle = config.election_cycles[cycle] return f"vote-{get_datetime(current_cycle.election_start).year}-{current_cycle.name}" def get_current_cycle(config: DakinyConfig) -> int: """ Returns index for active election. An election is considered active between the election start and the term start. Args: config (DakinyConfig): Config object to use Returns: int: cycle index for active election. -1 for no active cycle """ today = datetime.today() for cycle, election in enumerate(config.election_cycles): if ( today.date() >= get_datetime(election.election_start).date() and today.date() <= get_datetime(election.term_start).date() ): return cycle return -1