AnkiNoteTemplate/antp/updater.py
2022-01-30 09:31:09 +03:00

56 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():
stored_models = {
(model := read_model(dir_name)).name: model
for dir_name in os.listdir(NOTE_TYPES_DIR)
}
updatable_models = [
model_name for model_name in stored_models
if model_name in request_model_names()
]
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 = stored_models[model_name]
store_fonts(get_used_fonts(model.css))
send_note_type(model)
print("Done.")