Compare commits

...

1 Commits

Author SHA1 Message Date
Sergio Martínez Portela
3be4475ddf Support get_raw_contents of list groups. 2022-11-03 23:01:21 +01:00

View File

@ -7,7 +7,6 @@ class DrawerNode:
class PropertyDrawerNode(DrawerNode): class PropertyDrawerNode(DrawerNode):
def __repr__(self): def __repr__(self):
return "<Properties: {}>".format(len(self.children)) return "<Properties: {}>".format(len(self.children))
@ -38,6 +37,9 @@ class ListGroupNode:
def append(self, child): def append(self, child):
self.children.append(child) self.children.append(child)
def get_raw(self):
return '\n'.join([c.get_raw() for c in self.children])
def __repr__(self): def __repr__(self):
return "<List: {}>".format(len(self.children)) return "<List: {}>".format(len(self.children))
@ -64,6 +66,9 @@ class Text:
def __init__(self, content): def __init__(self, content):
self.content = content self.content = content
def get_raw(self):
return ''.join(self.content.get_raw())
class ListItem: class ListItem:
def __init__(self, tag, content, orig=None): def __init__(self, tag, content, orig=None):
@ -71,6 +76,9 @@ class ListItem:
self.content = content self.content = content
self.orig = orig self.orig = orig
def get_raw(self):
return get_raw_contents(self.orig)
class BlockNode: class BlockNode:
def __init__(self): def __init__(self):
@ -92,3 +100,5 @@ class CodeBlock(BlockNode):
def __repr__(self): def __repr__(self):
return "<Code: {}>".format(len(self.lines)) return "<Code: {}>".format(len(self.lines))
from .utils import get_raw_contents