update bugfixes

src/anki_hsk_creator/anki_generation.py

@@ -13,6 +13,7 @@ from pinyin_tone_converter.pinyin_tone_converter import PinyinToneConverter
 
 # Local
 from .utility import DictionaryResult, ProcessFile, ProcessFolder, TranslationResult
+from .constants import DICT_TYPE, DICTATION_TYPE, PHRASES_TYPE
 
 # Constants
 
@@ -241,31 +242,60 @@ def output_anki_phrase(
 def output_anki_package(
     process_folder: ProcessFolder, results: dict[ProcessFile, list[TranslationResult]]
 ):
-    final_file = process_file.output_name.with_suffix(".apkg")
+    final_file = process_folder.output_name.with_suffix(".apkg")
     decks = []
     audios = []
     for process_file, results in results.items():
         deck_name = "::".join(
-            ProcessFolder.input_folder.parts[:-1] + (ProcessFolder.output_name.stem,)
+            process_file.input_file.parts[:-1] + (process_file.output_name.stem,)
         )
         deck = Deck(
             random.randrange(1 << 30, 1 << 31),
             deck_name,
-            f"Deck for {final_file.name}, "
+            f"Deck for {process_file.input_file}, "
             "created in https://www.wolfang.info.ve/hskankicreator/",
         )
-        for result in results:
-            note = Note(
-                model=DICTATION_MODEL,
-                fields=[
-                    result.translated,
-                    result.line,
-                    f"[sound:{result.audio_path}]",
-                ],
-            )
-            deck.add_note(note)
-            audios.append(result.audio_path)
+        if process_file.file_type is DICTATION_TYPE:
+            for result in results:
+                note = Note(
+                    model=DICTATION_MODEL,
+                    fields=[
+                        result.translated,
+                        result.line,
+                        f"[sound:{result.audio_path}]",
+                    ],
+                )
+                deck.add_note(note)
+                audios.append(result.audio_path)
+        elif  process_file.file_type is DICT_TYPE:
+            for result in results:
+                note = Note(
+                    model=HSK_MODEL,
+                    fields=[
+                        # "\n ".join(f"{n+1}. {m}" for n, m in enumerate(result.meanings)),
+                        result.meaning,
+                        PinyinToneConverter().convert_text(result.pinyin),
+                        result.simplified,
+                        result.traditional,
+                        f"[sound:{result.audio_path}]",
+                    ],
+                )
+                deck.add_note(note)
+                audios.append(result.audio_path)
+        elif  process_file.file_type is PHRASES_TYPE:
+            for result in results:
+                note = Note(
+                    model=PHRASE_MODEL,
+                    fields=[
+                        result.translated,
+                        result.line,
+                        f"[sound:{result.audio_path}]",
+                    ],
+                )
+                deck.add_note(note)
+                audios.append(result.audio_path)
         decks.append(deck)
-        package = Package(decks)
-        package.media_files = audios
-        package.write_to_file(final_file)
+    package = Package(decks)
+    package.media_files = audios
+    package.write_to_file(final_file)
+    return final_file