redxef/anki-decks
1
0
Fork 0
Code Issues Pull requests Projects Releases 3 Packages Wiki Activity Actions

feat: allow creating a package with multiple decks.

Browse source
This commit is contained in:
redxef 2025-03-17 16:38:02 +01:00
parent 4a129b60a3
commit cc7f8283bb
Signed by: redxef
GPG key ID: 7DAC3AA211CBD921
1 changed files with 57 additions and 48 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

105
generate.py
View file

@ -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('--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('--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.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( def main(
*, *,
name: str | None, name: str | None,
@ -69,54 +70,62 @@ def main(
destructive: bool, destructive: bool,
autogenerate: bool, autogenerate: bool,
debug: bool, debug: bool,
path: pathlib.Path, multi: pathlib.Path | None,
paths: list[pathlib.Path],
): ):
match = RE_DIRNAME.match(path.name) decks = []
if match: for path in paths:
model_name = model_name or match.group(1) match = RE_DIRNAME.match(path.name)
name = name or match.group(1) if match:
model_id = model_id or int(match.group(2)) model_name = model_name or match.group(1)
deck_id = deck_id or int(match.group(3)) name = name or match.group(1)
else: model_id = model_id or int(match.group(2))
model_name = model_name or path.name deck_id = deck_id or int(match.group(3))
name = name or path.name else:
if model_id is None: model_name = model_name or path.name
if autogenerate: name = name or path.name
model_id = random.randrange(1 << 30, 1 << 31) if model_id is None:
else: if autogenerate:
raise click.ClickException('model_id could not be derived from directory name and is not explicitly set, or autogenerate enabled') model_id = random.randrange(1 << 30, 1 << 31)
if deck_id is None: else:
if autogenerate: raise click.ClickException('model_id could not be derived from directory name and is not explicitly set, or autogenerate enabled')
deck_id = random.randrange(1 << 30, 1 << 31) if deck_id is None:
else: if autogenerate:
raise click.ClickException('deck_id could not be derived from directory name and is not explicitly set, or autogenerate enabled') deck_id = random.randrange(1 << 30, 1 << 31)
if destructive: else:
new_path = path.parent / f'{path.name}.{model_id}.{deck_id}' raise click.ClickException('deck_id could not be derived from directory name and is not explicitly set, or autogenerate enabled')
if not debug: if destructive:
path.rename(new_path) new_path = path.parent / f'{path.name}.{model_id}.{deck_id}'
path = new_path if not debug:
model = genanki.Model( path.rename(new_path)
model_id, path = new_path
model_name, model = genanki.Model(
fields=[dict(name=x) for x in fields], model_id,
templates=[ model_name,
{ fields=[dict(name=x) for x in fields],
'name': template_name, templates=[
'qfmt': question_format, {
'afmt': answer_format, 'name': template_name,
}, 'qfmt': question_format,
], 'afmt': answer_format,
) },
notes: list[genanki.Note] = load_notes(path=path, model=model) ],
deck = genanki.Deck( )
deck_id, notes: list[genanki.Note] = load_notes(path=path, model=model)
name, deck = genanki.Deck(
) deck_id,
for n in notes: name,
deck.add_note(n) )
click.echo(f'added {len(notes)} notes') for n in notes:
if not debug: deck.add_note(n)
genanki.Package(deck).write_to_file(path.parent / (path.name + '.apkg')) 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)