From 49792f530dae352ae0a4c8a6b9c3b534deddb8da Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Maximilian=20Ke=C3=9Fler?= <git@maximilian-kessler.de>
Date: Fri, 12 May 2023 21:36:18 +0200
Subject: [PATCH] add suit class

---
 variants.py | 43 ++++++++++++++++++++++++++++++++++++++-----
 1 file changed, 38 insertions(+), 5 deletions(-)

diff --git a/variants.py b/variants.py
index 4976781..d5eaaef 100644
--- a/variants.py
+++ b/variants.py
@@ -1,8 +1,4 @@
-import json
-import os
-import networkx as nx
-from collections import OrderedDict
-import matplotlib.pyplot as plt
+import enum
 
 from database import cur
 
@@ -31,6 +27,43 @@ def num_suits(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,