generate_deck: format and move data related code into data.py

1 files changed, 4 insertions, 17 deletions

diff --git a/scripts/generate_deck.py b/scripts/generate_deck.py
index 18c73cc5..6a4519c5 100755
--- a/scripts/generate_deck.py
+++ b/scripts/generate_deck.py
@@ -3,6 +3,7 @@ import argparse
 import pathlib
 from pathlib import Path
 from typing import MutableMapping, Tuple
+from data import CARDS_TO_SETS, SETS_TO_CARDS
 
 DECK_TEMPLATE = \
 """\\documentclass[a4paper,landscape]{{article}}
@@ -18,28 +19,12 @@ DECK_TEMPLATE = \
 \\end{{center}}
 \\end{{document}}"""
 
-GAME_ROOT = Path(__file__).parent.parent
-CARDS_DATA_DIR = GAME_ROOT / 'data' / 'cards'
-
-CARDS_TO_SETS = {}
-SETS_TO_CARDS = {}
-
-
-# build up the card lookup dictionaries
-for set_dir in CARDS_DATA_DIR.iterdir(): 
-    set_name = set_dir.name
-    SETS_TO_CARDS[set_name] = []
-    for card in set_dir.iterdir():
-        SETS_TO_CARDS[set_name].append(card.stem)
-        CARDS_TO_SETS[card.stem] = set_name
-
 
 def generate_markdown(deck: list[str], language='en', indentation=3):
     """Output a markdown enumeration"""
     return ''
 
 
-
 def generate_latex(deck: list[str], language='en'):
     """Generate a document containing all included cards"""
 
@@ -50,7 +35,9 @@ def generate_latex(deck: list[str], language='en'):
         card = card.replace(' ', '_').lower()
         containing_set = CARDS_TO_SETS[card]
         for _ in range(int(amount)):
-            cards.append(f'\\includestandalone{{cards/{language}/{containing_set}/{card}}}')
+            cards.append(
+                f'\\includestandalone{{cards/{language}/{containing_set}/{card}}}'
+            )
     print(DECK_TEMPLATE.format(CARDS='\n'.join(cards)))