import enum from database import cur def variant_id(name): cur.execute( "SELECT id FROM variants WHERE name = %s", (name,) ) return cur.fetchone()[0] def variant_name(var_id): cur.execute( "SELECT name FROM variants WHERE id = %s", (var_id,) ) return cur.fetchone()[0] def num_suits(var_id): cur.execute( "SELECT num_suits FROM variants WHERE id = %s", (var_id,) ) return cur.fetchone()[0] class ClueBehaviour(enum.Enum): none = 0 default = 1 all = 2 class Suit: def __init__(self, name, display_name, abbreviation, rank_clues, color_clues, prism, dark, rev, colors): self.name = name self.display_name = display_name self.abbreviation = abbreviation self.rank_clues = ClueBehaviour(rank_clues) self.color_clues = ClueBehaviour(color_clues) self.prism = prism self.dark = dark self.reversed = rev self.colors = colors @staticmethod def from_db(suit_id): cur.execute( "SELECT name, display_name, abbreviation, rank_clues, color_clues, prism, dark, reversed " "FROM suits " "WHERE id = %s", (suit_id,) ) suit_properties = cur.fetchone() cur.execute( "SELECT color_id FROM suit_colors WHERE suit_id = %s", (suit_id,) ) colors = list(map(lambda t: t[0], cur.fetchall())) return Suit(*suit_properties, colors) class Variant: def __init__( self, name, clue_starved, throw_it_in_a_hole, alternating_clues, synesthesia, chimneys, funnels, no_color_clues, no_rank_clues, odds_and_evens, up_or_down, critical_fours, num_suits, special_rank, special_rank_ranks, special_rank_colors, suits ): self.name = name self.clue_starved = clue_starved self.throw_it_in_a_hole = throw_it_in_a_hole self.alternating_clues = alternating_clues self.synesthesia = synesthesia self.chimneys = chimneys self.funnels = funnels self.no_color_clues = no_color_clues self.no_rank_clues = no_rank_clues self.odds_and_evens = odds_and_evens self.up_or_down = up_or_down self.critical_fours = critical_fours self.num_suits = num_suits self.special_rank = special_rank self.special_rank_ranks = special_rank_ranks self.special_rank_colors = special_rank_colors self.suits = suits @staticmethod def from_db(var_id): cur.execute( "SELECT " "name, clue_starved, throw_it_in_a_hole, alternating_clues, synesthesia, chimneys, funnels, " "no_color_clues, no_rank_clues, odds_and_evens, up_or_down, critical_fours, num_suits, special_rank, " "special_rank_ranks, special_rank_colors " "FROM variants WHERE id = %s", (var_id,) ) var_properties = cur.fetchone() cur.execute( "SELECT suit_id FROM variant_suits " "WHERE variant_id = %s " "ORDER BY index", (var_id,) ) var_suits = list(map(lambda x: x[0], cur.fetchall())) return Variant(*var_properties, var_suits)