aboutsummaryrefslogtreecommitdiff
path: root/scripts/generate_deck.py
diff options
context:
space:
mode:
Diffstat (limited to 'scripts/generate_deck.py')
-rwxr-xr-xscripts/generate_deck.py8
1 files changed, 4 insertions, 4 deletions
diff --git a/scripts/generate_deck.py b/scripts/generate_deck.py
index 7003e750..8ccd12d0 100755
--- a/scripts/generate_deck.py
+++ b/scripts/generate_deck.py
@@ -1,7 +1,7 @@
#!/usr/bin/env python3
import argparse
from pathlib import Path
-from data import CARDS_TO_SETS
+from data import name2set
DECK_TEMPLATE = \
"""\\documentclass[a4paper,landscape]{{article}}
@@ -18,13 +18,13 @@ DECK_TEMPLATE = \
\\end{{document}}"""
-def generate_markdown(deck: list[str], language='en', indentation=0):
+def generate_markdown(deck: list[str], language='en', indentation=0): # pylint: disable=W0613
"""Output a markdown enumeration"""
m = []
for line in deck:
line = line.strip()
amount, card = line.split(' ', maxsplit=1)
- m += [f'{indentation}* {card}'] * amount
+ m += [f'{indentation}* {card}'] * int(amount)
return ''.join(m)
@@ -36,7 +36,7 @@ def generate_latex(deck: list[str], language='en'):
line = line.strip()
amount, card = line.split(' ', maxsplit=1)
card = card.replace(' ', '_').lower()
- containing_set = CARDS_TO_SETS[card]
+ containing_set = name2set(card)
for _ in range(int(amount)):
cards.append(
f'\\includestandalone{{cards/{language}/{containing_set}/{card}}}'