jonas/typstar
1
0
Fork 0
mirror of https://github.com/Ascyii/typstar.git synced 2026-01-01 05:24:24 -05:00
Code Issues Packages Projects Releases Wiki Activity

feat(anki): custom preamble per directory

Browse Source
This commit is contained in:
arne314
2024-12-24 13:22:09 +01:00
parent b0d8031a8e
commit 1f2bc8e486
6 changed files with 61 additions and 132 deletions
Download Patch File Download Diff File Expand all files Collapse all files

44
src/anki/parser.py
View File

@@ -1,3 +1,7 @@
import glob
import os.path
from functools import cache
from typing import List
import tree_sitter
@@ -32,12 +36,15 @@ class FlashcardParser:
typst_parser: tree_sitter.Parser
flashcard_query: tree_sitter.Query
file_handlers: List[tuple[FileHandler, List[Flashcard]]]
def __init__(self):
self.typst_language = get_language("typst")
self.typst_parser = get_parser("typst")
self.flashcard_query = self.typst_language.query(ts_flashcard_query)
self.file_handlers = []
def parse_file(self, file: FileHandler) -> List[Flashcard]:
def parse_file(self, file: FileHandler, preamble: str) -> List[Flashcard]:
cards = []
tree = self.typst_parser.parse(file.get_bytes(), encoding="utf8")
captures = self.flashcard_query.captures(tree.root_node)
@@ -57,8 +64,43 @@ class FlashcardParser:
file.get_node_content(back, True),
None,
int(file.get_node_content(note_id)),
preamble,
file,
)
card.set_ts_nodes(front, back, note_id)
cards.append(card)
return cards
def parse_directory(self, root_dir):
print(f"Parsing flashcards in {root_dir}...")
preambles = {}
flashcards = []
@cache
def get_preamble(path) -> str | None:
while len(path) > len(root_dir):
if preamble := preambles.get(path):
return preamble
path = os.path.dirname(path)
for file in sorted(glob.glob(f"{root_dir}/**/**.typ", include_hidden=True, recursive=True)):
if os.path.basename(file) == ".anki.typ":
with open(file, encoding="utf-8") as f:
preambles[os.path.dirname(file)] = f.read()
continue
fh = FileHandler(file)
cards = self.parse_file(fh, get_preamble(os.path.dirname(file)))
self.file_handlers.append((fh, cards))
flashcards.extend(cards)
return flashcards
def update_ids_in_source(self):
print("Updating ids in source...")
for fh, cards in self.file_handlers:
file_updated = False
for c in cards:
if c.id_updated:
fh.update_node_content(c.note_id_node, c.note_id)
file_updated = True
if file_updated:
fh.write()