From 3a29bcfa511a884849815921ca27ce6babf20b85 Mon Sep 17 00:00:00 2001 From: Florian Fischer Date: Sun, 10 Oct 2021 14:48:08 +0200 Subject: [scripts/generate_card] fix type hints --- scripts/generate_card.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) (limited to 'scripts') diff --git a/scripts/generate_card.py b/scripts/generate_card.py index 509cca25..d403eac8 100755 --- a/scripts/generate_card.py +++ b/scripts/generate_card.py @@ -2,7 +2,7 @@ import argparse import pathlib from pathlib import Path -from typing import MutableMapping +from typing import MutableMapping, Tuple, Union import yaml CARD_TEMPLATE = \ @@ -79,7 +79,7 @@ def get_formatted_key(key: str, language='en') -> str: def get_latex_field(card: MutableMapping, field: str, - language='en') -> (str, bool): + language='en') -> Tuple[str, bool]: """Return the latex_ prefixed field from card or the field itself""" latex_field = f'latex_{field}' value = card.get(latex_field, card[field]) @@ -191,9 +191,9 @@ def generate_latex(card: MutableMapping, language='en'): elif card['type'] == 'equipment': formatted_durability = get_formatted_key('durability', language) durability = get_field(card, 'durability', language) - stats = f'{formatted_durability}: {durability}' + equipment_stats = f'{formatted_durability}: {durability}' effect, _ = get_latex_field(card, 'effect', language) - card_content += f'\\cardsplitcontent{{{stats}}}{{{effect}}}\n' + card_content += f'\\cardsplitcontent{{{equipment_stats}}}{{{effect}}}\n' elif card['type'] == 'potion': formatted_use = get_formatted_key('use', language) @@ -244,7 +244,7 @@ def main(): card_set = data_path.parent.stem for data_file in data_files: - with open(data_file, 'r') as yaml_file: + with open(data_file, 'r', encoding="utf8") as yaml_file: card = yaml.full_load(yaml_file) card['set'] = card_set -- cgit v1.2.3