95 lines
2.8 KiB
Python
95 lines
2.8 KiB
Python
import dataclasses
|
|
import errno
|
|
import re
|
|
import stat
|
|
from collections.abc import Iterator
|
|
from logging import getLogger
|
|
|
|
import frontmatter
|
|
import marko
|
|
import marko.md_renderer
|
|
|
|
from .favro_data_model import Card, SeqId
|
|
|
|
@dataclasses.dataclass(frozen=True)
|
|
class CardContents:
|
|
name: str | None
|
|
description: str | None
|
|
tags: list[str]
|
|
assignments: list[str]
|
|
|
|
class CardFileFormatter:
|
|
"""Component for formatting and parsing card files """
|
|
|
|
def __init__(self, obsidian_mode = True):
|
|
self.obsidian_mode = obsidian_mode
|
|
self.markdown = marko.Markdown()
|
|
self.renderer = marko.md_renderer.MarkdownRenderer()
|
|
|
|
def format_card_contents(self, card: CardContents) -> str:
|
|
ls = []
|
|
|
|
# Choose frontmatter data
|
|
frontmatter_data = {}
|
|
if card.name and self.obsidian_mode:
|
|
frontmatter_data['aliases'] = [card.name]
|
|
if card.tags:
|
|
frontmatter_data['tags'] = card.tags
|
|
if self.obsidian_mode:
|
|
frontmatter_data['tags'] = ['#'+t for t in frontmatter_data['tags']]
|
|
if card.assignments:
|
|
frontmatter_data['assignments'] = card.assignments
|
|
if self.obsidian_mode:
|
|
frontmatter_data['assignments'] = [f'[[{name}]]' for name in frontmatter_data['assignments']]
|
|
|
|
# Frontmatter
|
|
if frontmatter_data:
|
|
ls.append('---\n')
|
|
for key, values in frontmatter_data.items():
|
|
ls.append(key)
|
|
ls.append(':\n')
|
|
for v in values:
|
|
ls.append(' - ')
|
|
ls.append(v)
|
|
ls.append('\n')
|
|
ls.append('---\n\n')
|
|
|
|
# Card name
|
|
if card.name:
|
|
ls.append('# ')
|
|
ls.append(card.name)
|
|
ls.append('\n\n')
|
|
|
|
# Card contents
|
|
if card.description:
|
|
ls.append(card.description)
|
|
return ''.join(ls)
|
|
|
|
def parse_card_contents(self, contents: str) -> CardContents:
|
|
"""
|
|
1. Strips frontmatter
|
|
2. Parses header
|
|
3. Finds content.
|
|
"""
|
|
fm = frontmatter.loads(contents)
|
|
del contents
|
|
|
|
document = self.markdown.parse(fm.content.strip())
|
|
name = None
|
|
for elem in document.children:
|
|
if isinstance(elem, marko.block.Heading):
|
|
name = self.renderer.render_children(elem)
|
|
document.children.remove(elem)
|
|
break
|
|
|
|
description = self.renderer.render_children(document).strip()
|
|
return CardContents(
|
|
name,
|
|
description,
|
|
tags = [],
|
|
assignments = [],
|
|
)
|
|
|
|
def format_card(self, card: Card) -> str:
|
|
return self.format_card_contents(CardContents(card.name, card.detailed_description))
|