diff options
| -rwxr-xr-x | anki-card-template/bulk-cleanup.py | 111 |
1 files changed, 111 insertions, 0 deletions
diff --git a/anki-card-template/bulk-cleanup.py b/anki-card-template/bulk-cleanup.py new file mode 100755 index 0000000..5c2c437 --- /dev/null +++ b/anki-card-template/bulk-cleanup.py @@ -0,0 +1,111 @@ +#!/bin/python3 +import sys +import os +import re +import io +from math import floor, log10 +from time import sleep +from pathlib import Path +from bs4 import BeautifulSoup + +real_stdout = sys.stdout +class TrashFileIO(object): + def write(self, x): pass + def flush(self): pass +trash_out = TrashFileIO() + +# replace stderr file descriptor with /dev/null +os.dup2(os.open(os.devnull, os.O_WRONLY), 2) + +sys.stdout = trash_out +import aqt +sys.stdout = real_stdout + +def escape(plain): + plain = plain.replace("*", "\\*") + plain = plain.replace("(", "\\(") + plain = plain.replace(")", "\\)") + plain = plain.replace("[", "\\[") + plain = plain.replace("]", "\\]") + plain = plain.replace("{", "\\{") + plain = plain.replace("}", "\\}") + return plain + +def recurseplainify(soup): + output = "" + for el in soup.children: + if el.string: + output += escape(el.string) + continue + + if el.name == 'b': + output += f"*{recurseplainify(el)}*" + continue + + if el.name == 'ruby': + output += f'[{el.text}]({el.rt.text})' + continue + + output += recurseplainify(el) + + return output + +def html2cardtemplate(html): + soup = BeautifulSoup(html) + return recurseplainify(soup) + +def main(): + sys.stdout = trash_out + app = aqt._run(sys.argv, False) + sys.stdout = real_stdout + if aqt.mw == None: + print("Please close any open Anki windows before running this script!") + sys.stdout = trash_out + exit(1) + + # load last open profile if no profile was specified on command line (option parsed by Anki) + if not aqt.mw.pm.name: + aqt.mw.pm.load(aqt.mw.pm.last_loaded_profile_name()) + + col = aqt.Collection(aqt.mw.pm.collectionPath()) + media_dir = col.media.dir() + + model = col.models.by_name("Sentence mining") + note_ids = col.models.nids(model) + + edited_notes = 0 + + note_index_format = ("{:0" + str(floor(log10(len(note_ids))) + 1) + "d}/{:d}") + for note_index, note_id in enumerate(note_ids): + edited = False + note = col.get_note(note_id) + print(f"[nid:{note_id}] ({note_index_format.format(note_index + 1, len(note_ids))})", end="") + + if note['Complete sentence'].find('<') >= 0: + print(" -> sentence HTML to plain", end="") + note['Complete sentence'] = html2cardtemplate(note['Complete sentence']) + edited = True + + if note['Target word reading'].find('<') >= 0: + soup = BeautifulSoup(note['Target word reading']) + note['Target word reading'] = soup.get_text() + print(" -> stripped HTML from TW reading", end="") + edited = True + + if not edited: + print("unmodified", end="\r") + else: + print("", end="\n") # flush line + col.update_note(note) + edited_notes += 1 + + if edited_notes == 0: + print("-- Done: no edits --") + else: + print(f"-- Done: succesfully edited {edited_notes} note{'' if edited_notes == 1 else 's'} --") + # circumvent "Exception ignored in atexit callbackException ignored in sys.unraisablehook" + sys.stdout = trash_out + +if __name__ == "__main__": + main() + |