llcotp.com/tools/pf-filter.py
2024-06-05 22:49:59 +02:00

79 lines
2.5 KiB
Python

import panflute as pf
import subprocess
import os
import re
def prepare(doc):
pass
def action(elem, doc):
# .md to .html
# if isinstance(elem, pf.Table):
# return
# if isinstance(elem, pf.Link) and elem.url.endswith('.md'):
# elem.url = '../' + elem.url[:-3] + '.html'
# return elem
# reference
if isinstance(elem, pf.Code):
# todo sketchy if/else, replace with regex
if '/' in elem.text:
book, text = elem.text.split('/')
else:
book = '1'
text = elem.text
if '.' in text:
# equation ref
eq = text
res = subprocess.run(["./tools/tex_to_html.sh", "inline", f"{book}/equations/{eq}.tex"], capture_output=True)
tooltiptext = res.stdout.decode('utf-8')
html = f'''
<span class="tooltip" class="math inline">
({eq})
<span class="tooltiptext">{tooltiptext}</span>
</span>
'''
# <span class="tooltiptext">{tooltiptext}</span>
elem = pf.RawInline(html, format='html')
return elem
else:
# section ref
pnum = text
try:
slug = [f for f in os.listdir(book) if f.startswith(pnum)][0]
except:
return elem
title = [l for l in open(f"{book}/{slug}", 'r').readlines() if l.startswith('title:')][0]
title = title.strip("title: ")
slug = slug.replace(".md", ".html")
html = f'<a href="{slug}"{title.strip('\n')}</a>'
elem = pf.RawInline(html, format='html')
return elem
# load equation
if isinstance(elem, pf.CodeBlock):
if '/' in elem.text:
book, text = elem.text.split('/')
else:
book = '1'
text = elem.text
if 'load' in elem.classes:
eq = text
res = subprocess.run(["./tools/tex_to_html.sh", "block", f"{book}/equations/{eq}.tex"], capture_output=True)
html = res.stdout.decode('utf-8')
elem = pf.RawBlock(html, format='html')
return elem
# problem solution hidden
# if isinstance(elem, pf.Div):
# if 'solution' in elem.classes:
# details_start = pf.RawBlock(f'''<details><summary><b>Solution</b></summary>''')
# details_end = pf.RawBlock('''</details>''')
# return [details_start, elem, details_end]
# return elem
if __name__ == '__main__':
pf.run_filter(action, prepare=prepare)