From cc7f8283bb735c5b6934f52f3fa33f208f6c7220 Mon Sep 17 00:00:00 2001
From: redxef <redxef@redxef.at>
Date: Mon, 17 Mar 2025 16:38:02 +0100
Subject: [PATCH] feat: allow creating a package with multiple decks.

---
 generate.py | 105 ++++++++++++++++++++++++++++------------------------
 1 file changed, 57 insertions(+), 48 deletions(-)

diff --git a/generate.py b/generate.py
index 94ddd96..77b3db4 100755
--- a/generate.py
+++ b/generate.py
@@ -55,7 +55,8 @@ def load_notes(
 @click.option('--destructive/--no-destructive', 'destructive', type=bool, default=False, help='Perform write operations, which may end in data loss')
 @click.option('--autogenerate/--no-autogenerate', 'autogenerate', type=bool, default=False, help='Generate missing ids, should probably be used with --destructive, to always use the same ones.')
 @click.option('--debug/--no-debug', 'debug', type=bool, default=False, help="Debug mode, don't write files")
-@click.argument('path', type=click.Path(path_type=pathlib.Path, dir_okay=True))
+@click.option('--multi', 'multi', type=click.Path(path_type=pathlib.Path))
+@click.argument('paths', type=click.Path(path_type=pathlib.Path, dir_okay=True), nargs=-1)
 def main(
         *,
         name: str | None,
@@ -69,54 +70,62 @@ def main(
         destructive: bool,
         autogenerate: bool,
         debug: bool,
-        path: pathlib.Path,
+        multi: pathlib.Path | None,
+        paths: list[pathlib.Path],
 ):
-    match = RE_DIRNAME.match(path.name)
-    if match:
-        model_name = model_name  or match.group(1)
-        name = name or match.group(1)
-        model_id = model_id or int(match.group(2))
-        deck_id = deck_id or int(match.group(3))
-    else:
-        model_name = model_name or path.name
-        name = name or path.name
-        if model_id is None:
-            if autogenerate:
-                model_id = random.randrange(1 << 30, 1 << 31)
-            else:
-                raise click.ClickException('model_id could not be derived from directory name and is not explicitly set, or autogenerate enabled')
-        if deck_id is None:
-            if autogenerate:
-                deck_id = random.randrange(1 << 30, 1 << 31)
-            else:
-                raise click.ClickException('deck_id could not be derived from directory name and is not explicitly set, or autogenerate enabled')
-        if destructive:
-            new_path = path.parent / f'{path.name}.{model_id}.{deck_id}'
-            if not debug:
-                path.rename(new_path)
-                path = new_path
-    model = genanki.Model(
-        model_id,
-        model_name,
-        fields=[dict(name=x) for x in fields],
-        templates=[
-            {
-                'name': template_name,
-                'qfmt': question_format,
-                'afmt': answer_format,
-            },
-        ],
-    )
-    notes: list[genanki.Note] = load_notes(path=path, model=model)
-    deck = genanki.Deck(
-        deck_id,
-        name,
-    )
-    for n in notes:
-        deck.add_note(n)
-    click.echo(f'added {len(notes)} notes')
-    if not debug:
-        genanki.Package(deck).write_to_file(path.parent / (path.name + '.apkg'))
+    decks = []
+    for path in paths:
+        match = RE_DIRNAME.match(path.name)
+        if match:
+            model_name = model_name  or match.group(1)
+            name = name or match.group(1)
+            model_id = model_id or int(match.group(2))
+            deck_id = deck_id or int(match.group(3))
+        else:
+            model_name = model_name or path.name
+            name = name or path.name
+            if model_id is None:
+                if autogenerate:
+                    model_id = random.randrange(1 << 30, 1 << 31)
+                else:
+                    raise click.ClickException('model_id could not be derived from directory name and is not explicitly set, or autogenerate enabled')
+            if deck_id is None:
+                if autogenerate:
+                    deck_id = random.randrange(1 << 30, 1 << 31)
+                else:
+                    raise click.ClickException('deck_id could not be derived from directory name and is not explicitly set, or autogenerate enabled')
+            if destructive:
+                new_path = path.parent / f'{path.name}.{model_id}.{deck_id}'
+                if not debug:
+                    path.rename(new_path)
+                    path = new_path
+        model = genanki.Model(
+            model_id,
+            model_name,
+            fields=[dict(name=x) for x in fields],
+            templates=[
+                {
+                    'name': template_name,
+                    'qfmt': question_format,
+                    'afmt': answer_format,
+                },
+            ],
+        )
+        notes: list[genanki.Note] = load_notes(path=path, model=model)
+        deck = genanki.Deck(
+            deck_id,
+            name,
+        )
+        for n in notes:
+            deck.add_note(n)
+        click.echo(f'added {len(notes)} notes')
+        decks += [deck]
+        if not debug:
+            genanki.Package(deck).write_to_file(path.parent / (path.name + '.apkg'))
+    if not debug and multi:
+        if multi.suffix != '.apkg':
+            click.echo('WARNING: destination file does not have extension ".apkg"')
+        genanki.Package(decks).write_to_file(multi)