Marty/papis-extract
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

test: Fix formatting and annotation tests

Browse source
This commit is contained in:
Marty Oehme 2023-09-22 20:04:39 +02:00
parent ee4690f52b
commit 1e29642cba
Signed by: Marty
GPG key ID: EDBF2ED917B2EF6A
3 changed files with 77 additions and 52 deletions
Download patch file Download diff file Expand all files Collapse all files

12
papis_extract/formatter.py
View file

@ -10,8 +10,8 @@ def format_markdown(
) -> str:
template = (
"{{#tag}}#{{tag}}\n{{/tag}}"
"{{#quote}}> {{quote}}{{/quote}} {{#page}}[p. {{page}}]{{/page}}"
"\n{{#note}} NOTE: {{note}}{{/note}}"
"{{#quote}}> {{quote}}{{/quote}}{{#page}} [p. {{page}}]{{/page}}"
"{{#note}}\n NOTE: {{note}}{{/note}}"
)
output = ""
for entry in docs:
@ -30,11 +30,11 @@ def format_markdown(
for a in entry.annotations:
output += a.format(template)
output += "\n"
output += "\n\n"
output += "\n\n\n"
return output
return output.rstrip()
def format_markdown_atx(docs: list[AnnotatedDocument] = []) -> str:
@ -63,7 +63,7 @@ def format_count(docs: list[AnnotatedDocument] = []) -> str:
f"{count}\n"
)
return output
return output.rstrip()
def format_csv(docs: list[AnnotatedDocument] = []) -> str:
@ -82,7 +82,7 @@ def format_csv(docs: list[AnnotatedDocument] = []) -> str:
output += a.format(template, doc=d)
output += "\n"
return output
return output.rstrip()
formatters: dict[str, Formatter] = {