MVP done

1 files changed, 61 insertions, 2 deletions

diff --git a/solver.py b/solver.py
index e322c8c..4c302b5 100644
--- a/solver.py
+++ b/solver.py
@@ -1,3 +1,62 @@
+import random
 
-def solve(case, article, number, adjective, noun):
-    return ""
+import tables
+
+class Query:
+    def __init__(self, case, article, cardinality, adjective, noun):
+        self.case = case
+        self.article = article
+        self.cardinality = cardinality
+        self.adjective = adjective
+        self.noun = noun
+
+class Solution:
+    def __init__(self, full, noun):
+        self.full = full
+        self.noun = noun
+
+    def evaluate(self, candidate):
+        return self.full.lower().strip() == candidate.lower().strip()
+
+def _get():
+    case = random.choice(tables.CASES)
+    article = random.choice(tables.ARTICLE_TYPES)
+    cardinality = random.choice(tables.CARDINALITIES)
+    adjective = random.choice(tables.ADJECTIVES)
+    noun = random.choice(tables.NOUNS)
+
+    return Query(case, article, cardinality, adjective, noun)
+
+def get():
+    attempts = 0
+    while attempts < 10:
+        attempts += 1
+
+        query = _get()
+        try:
+            solution = solve(query)
+            return (query, solution)
+        except Exception as e:
+            print(e)
+            continue
+
+    raise Exception("Could not find query after 10 attempts!")
+
+def solve(query):
+    gender = "p" if query.cardinality == "Plural" else query.noun["gender"]
+    nounKey = f"{query.case.lower()[0:3]}-{query.cardinality.lower()[0:3]}"
+
+    # full solution
+    decArt = tables.ARTICLES[query.article][query.case][gender]
+    adjEnding = tables.ADJ_ENDINGS[query.article][query.case][gender]
+    decNoun = query.noun[nounKey]
+    
+    # noun details
+    basicArt = tables.ARTICLES['bestimmter Artikel']['Nominativ'][gender]
+    basicNoun = query.noun[f"nom-{'plu' if query.cardinality == 'Plural' else 'sin'}"]
+
+    if decNoun == "_":
+        raise Exception("Invalid query")
+    
+    return Solution(f"{decArt} {query.adjective}{adjEnding} {decNoun}".strip(),
+                    f"{basicArt} {basicNoun} ({query.cardinality})")