Added MJ's show function.
This commit is contained in:
parent
4ca04d8202
commit
51ec48bf05
2 changed files with 50 additions and 3 deletions
|
@ -5,10 +5,10 @@ from whitespace import *
|
||||||
|
|
||||||
from panflute import *
|
from panflute import *
|
||||||
|
|
||||||
|
from mj_show import show
|
||||||
|
|
||||||
ifs = ["dog"]
|
ifs = ["dog"]
|
||||||
|
|
||||||
|
|
||||||
def transform(e: Element):
|
def transform(e: Element):
|
||||||
"""Transform the AST, making format-agnostic changes."""
|
"""Transform the AST, making format-agnostic changes."""
|
||||||
if isinstance(e, Whitespace) and bavlna(e):
|
if isinstance(e, Whitespace) and bavlna(e):
|
||||||
|
@ -26,10 +26,10 @@ def transform(e: Element):
|
||||||
|
|
||||||
doc = load()
|
doc = load()
|
||||||
|
|
||||||
print(stringify(doc))
|
print(show(doc))
|
||||||
transform(doc)
|
transform(doc)
|
||||||
print("---------------------")
|
print("---------------------")
|
||||||
print(stringify(doc))
|
print(show(doc))
|
||||||
print(vars(doc))
|
print(vars(doc))
|
||||||
|
|
||||||
|
|
||||||
|
|
47
mj_show.py
Normal file
47
mj_show.py
Normal file
|
@ -0,0 +1,47 @@
|
||||||
|
###
|
||||||
|
# For debugging, borrowed from here: git://git.ucw.cz/labsconf2022.git
|
||||||
|
###
|
||||||
|
|
||||||
|
import sys, re
|
||||||
|
from panflute import *
|
||||||
|
|
||||||
|
avoid_keys = {
|
||||||
|
'dict',
|
||||||
|
'list',
|
||||||
|
'location',
|
||||||
|
'oktypes',
|
||||||
|
'parent',
|
||||||
|
}
|
||||||
|
|
||||||
|
def show(e, file=sys.stdout, indent=""):
|
||||||
|
t = re.sub(r".*\.(.*)'>", r'\1', str(type(e)))
|
||||||
|
if isinstance(e, Str):
|
||||||
|
file.write(f'{indent}{t} "{e.text}"\n')
|
||||||
|
else:
|
||||||
|
file.write(f'{indent}{t}\n')
|
||||||
|
|
||||||
|
if not isinstance(e, Str):
|
||||||
|
for k in e.__slots__:
|
||||||
|
if hasattr(e, k) and not k.startswith('_') and not k in avoid_keys:
|
||||||
|
file.write(f'{indent} {k}={getattr(e, k)}\n')
|
||||||
|
|
||||||
|
if isinstance(e, Element):
|
||||||
|
children = sorted((child_name, getattr(e, child_name)) for child_name in e._children)
|
||||||
|
for name, c in children:
|
||||||
|
if name == 'content':
|
||||||
|
show(c, file, indent + ' ')
|
||||||
|
elif c is not None:
|
||||||
|
file.write(f'{indent} {name}=\n')
|
||||||
|
show(c, file, indent + ' ')
|
||||||
|
else:
|
||||||
|
file.write(f'{indent} {name}=None\n')
|
||||||
|
elif isinstance(e, ListContainer):
|
||||||
|
for c in e:
|
||||||
|
show(c, file, indent + ' ')
|
||||||
|
elif isinstance(e, DictContainer):
|
||||||
|
for name, c in e.items():
|
||||||
|
file.write(f'{indent} {name}:\n')
|
||||||
|
show(c, file, indent + ' ')
|
||||||
|
else:
|
||||||
|
raise TypeError(type(e))
|
||||||
|
|
Loading…
Reference in a new issue