From 2eeeade1e51cc4e24106a95b1e849d2e3a806ffb Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Maximilian=20Ke=C3=9Fler?= <git@maximilian-kessler.de>
Date: Wed, 9 Oct 2024 17:47:53 +0200
Subject: [PATCH] Seed generation: Generate custom random seeds independent of
 hanab.live

---
 src/hanabi/database/games_seeds_schema.sql |  1 +
 src/hanabi/live/generate_seeds.py          | 54 ++++++++++++++++++++++
 2 files changed, 55 insertions(+)

diff --git a/src/hanabi/database/games_seeds_schema.sql b/src/hanabi/database/games_seeds_schema.sql
index 8cdac2b..b853bf0 100644
--- a/src/hanabi/database/games_seeds_schema.sql
+++ b/src/hanabi/database/games_seeds_schema.sql
@@ -12,6 +12,7 @@ CREATE TABLE seeds (
     num_players             SMALLINT    NOT NULL,
     variant_id              SMALLINT    NOT NULL,
     starting_player         SMALLINT    NOT NULL DEFAULT 0,
+    custom                  BOOLEAN     NOT NULL DEFAULT FAlSE,
     feasible                BOOLEAN     DEFAULT NULL,
     max_score_theoretical   SMALLINT
 );
diff --git a/src/hanabi/live/generate_seeds.py b/src/hanabi/live/generate_seeds.py
index e69de29..a7e515b 100644
--- a/src/hanabi/live/generate_seeds.py
+++ b/src/hanabi/live/generate_seeds.py
@@ -0,0 +1,54 @@
+from hanabi.hanab_game import DeckCard
+from hanabi import database
+from hanabi.live.variants import Variant
+from hanabi.database import games_db_interface
+import random
+
+def get_deck(variant: Variant):
+    deck = []
+    for suit_index, suit in enumerate(variant.suits):
+        if suit.dark:
+            for rank in range(1, 6):
+                deck.append(DeckCard(suit_index, rank))
+        else:
+            deck.append(DeckCard(suit_index, 1))
+            if not suit.reversed:
+                deck.append(DeckCard(suit_index, 1))
+                deck.append(DeckCard(suit_index, 1))
+
+            for rank in range(2,5):
+                deck.append(DeckCard(suit_index, rank))
+                deck.append(DeckCard(suit_index, rank))
+
+            deck.append(DeckCard(suit_index, 5))
+            if suit.reversed:
+                deck.append(DeckCard(suit_index, 5))
+                deck.append(DeckCard(suit_index, 5))
+    return deck
+
+def generate_deck(variant: Variant, num_players: int, seed: int):
+    deck = get_deck(variant)
+    seed = "p{}cs{}".format(num_players, seed)
+    random.seed(seed)
+    random.shuffle(deck)
+    return seed, deck
+
+def generate_decks_for_variant(variant_id: int, num_players: int, num_seeds: int):
+    variant = Variant.from_db(variant_id)
+    for seed in range(num_seeds):
+        seed, deck = generate_deck(variant, num_players, seed)
+        database.cur.execute(
+            "INSERT INTO seeds (seed, num_players, starting_player, variant_id, custom)"
+            "VALUES (%s, %s, %s, %s, %s)"
+            "ON CONFLICT (seed) DO NOTHING",
+            (seed, num_players, 0, variant_id, True)
+        )
+        games_db_interface.store_deck_for_seed(seed, deck)
+
+def main():
+    database.global_db_connection_manager.read_config()
+    database.global_db_connection_manager.connect()
+    generate_decks_for_variant(0, 4, 20000)
+
+if __name__ == '__main__':
+    main()