lyz/org-rw
1
0
Fork 0
forked from kenkeiras/org-rw
Code Pull Requests Actions Activity

Simplify parsing of CodeBlocks.

Browse Source
This commit is contained in:
Sergio Martínez Portela 2022-05-07 23:33:32 +02:00
parent afd02f55d4
commit 7a9cff57f9
2 changed files with 23 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
org_rw/dom.py
View File

@ -60,10 +60,10 @@ class BlockNode:
class CodeBlock(BlockNode):
def __init__(self, header):
self.header = header
self.lines = []
self.lines = None
def append(self, child):
self.lines.append(child)
def set_lines(self, lines):
self.lines = lines
def __repr__(self):
return "<Code: {}>".format(len(self.lines))

23
org_rw/org_rw.py
View File

@ -331,10 +331,22 @@ class Headline:
isinstance(line, DelimiterLine)
and line.delimiter_type == DelimiterLineType.END_SRC
):
start = current_node.header.linenum
end = line.linenum
lines = self.get_lines_between(start + 1, end)
contents = "\n".join(lines)
if contents.endswith("\n"):
# This is not ideal, but to avoid having to do this maybe
# the content parsing must be re-thinked
contents = contents[:-1]
current_node.set_lines(contents)
tree.append(current_node)
current_node = None
else:
current_node.append(line)
pass # Ignore
elif isinstance(line, Property):
if type(current_node) in NON_FINISHED_GROUPS:
@ -407,7 +419,6 @@ class Headline:
):
assert type(current_node) in NON_FINISHED_GROUPS
current_node = dom.CodeBlock(line)
current_node.append(current_node)
elif isinstance(line, Keyword):
logging.warning("Keywords not implemented on `as_dom()`")
@ -583,7 +594,13 @@ class Headline:
def get_lines_between(self, start, end):
for line in self.contents:
if start <= line.linenum < end:
yield "".join(line.contents)
text = []
for item in line.contents:
if isinstance(item, str):
text.append(item)
elif isinstance(item, MarkerType):
text.append(ModeToMarker[item])
yield "".join(text)
def get_contents(self, format):
if format == "raw":