AnkiNoteTemplate/antp/updater.py
2022-12-02 16:33:02 +03:00

57 lines
1.6 KiB
Python

# Updater updates previously imported note types.
# Copyright: Ren Tatsumoto <tatsu at autistici.org>
# License: GNU GPL, version 3 or later; http://www.gnu.org/licenses/agpl.html
from typing import Any
from .ankiconnect import invoke, request_model_names
from .common import select, get_used_fonts, NoteType
from .consts import *
from .importer import read_model, store_fonts
def format_templates(model: NoteType) -> dict[str, Any]:
return {
"model": {
"name": model.name,
"templates": {
template.name: {"Front": template.front, "Back": template.back} for template in model.templates
}
}
}
def format_styling(model: NoteType) -> dict[str, Any]:
return {
"model": {
"name": model.name,
"css": model.css
}
}
def send_note_type(model: NoteType):
invoke("updateModelTemplates", **format_templates(model))
invoke("updateModelStyling", **format_styling(model))
def update_note_type():
anki_models = request_model_names()
models_on_disk = {
(model := read_model(dir_name)).name: model
for dir_name in os.listdir(NOTE_TYPES_DIR)
}
updatable_models = [
model_name for model_name in models_on_disk
if model_name in anki_models
]
if not updatable_models:
print("No note types can be updated.")
return
if model_name := select(updatable_models):
print(f"Selected note type: {model_name}")
model = models_on_disk[model_name]
store_fonts(get_used_fonts(model.css))
send_note_type(model)
print("Done.")