fix and use cardmodule command

1 files changed, 14 insertions, 4 deletions

diff --git a/scripts/generate_latex.py b/scripts/generate_latex.py
index 282f0a7f..9e4f8d35 100755
--- a/scripts/generate_latex.py
+++ b/scripts/generate_latex.py
@@ -46,6 +46,8 @@ CARD_TEMPLATE = \
 \\end{{document}}
 """
 
+SET_SYMBOLS = {'base': 'base', 'magic': 'magic', 'equipments': 'equipments'}
+
 LATEX_ROOT = pathlib.Path(os.getcwd())
 ASSETS_DIR = LATEX_ROOT / 'assets'
 CARDS_DIR = LATEX_ROOT / 'cards'
@@ -109,19 +111,24 @@ def generate_card(card_name: str, card: MutableMapping):
         card_content += f'\cardplaycost{{{card["upkeep"]}}}'
     elif card['type'] == 'spell':
         card_content += f'\cardcontent{{{card["effect"]}}}\n'
-        card_content += f'\cardplaycost{{{card["play"]}}}'
+        card_content += f'\cardplaycost{{{card["play"]}}}\n'
 
     elif card['type'] == 'artifact':
         card_content += f'\cardcontent{{{card["effect"]}}}\n'
-        card_content += f'\cardplaycost{{{card["play"]}}}'
+        card_content += f'\cardplaycost{{{card["play"]}}}\n'
 
     elif card['type'] == 'equipment':
         # card_content += f'\cardsplitcontent{{\\faRecycle: {card["durability"]}}}{{{card["effect"]}}}'
         card_content += f'\cardsplitcontent{{Durability: {card["durability"]}}}{{{card["effect"]}}}\n'
-        card_content += f'\cardplaycost{{{card["play"]}}}'
+        card_content += f'\cardplaycost{{{card["play"]}}}\n'
     else:
         print('WARNING: unknown card type {card["type"]}!')
 
+    if card['set'] in SET_SYMBOLS:
+        card_content += f'\cardmodule{{{SET_SYMBOLS[card["set"]]}}}'
+    else:
+        print(f'WARNING unknown set: {card["set"]}')
+
     with open(CARDS_DIR / card_file, 'w') as tex_file:
         print(CARD_TEMPLATE.format(card_content), file=tex_file)
 
@@ -149,7 +156,10 @@ def main():
     collection_inputs = ""
     cards = [card for card in data if card != 'DEFAULT']
     for i, card in enumerate(cards):
-        generate_card(card, data[card])
+        card_content = data[card]
+        card_content = dict(card_content)
+        card_content['set'] = pathlib.Path(args.datafile).stem
+        generate_card(card, card_content)
         collection_inputs += f'\\input{{cards/{file_for_card(card)}}}\n'
         if i == 0 or i % 4 < 3:
             collection_inputs += f'\\hspace{{5mm}}\n'