org-rw/org_rw/dom.py
2022-11-03 23:01:21 +01:00

105 lines
2.2 KiB
Python

class DrawerNode:
def __init__(self):
self.children = []
def append(self, child):
self.children.append(child)
class PropertyDrawerNode(DrawerNode):
def __repr__(self):
return "<Properties: {}>".format(len(self.children))
class LogbookDrawerNode(DrawerNode):
def __repr__(self):
return "<LogBook: {}>".format(len(self.children))
class ResultsDrawerNode(DrawerNode):
def __repr__(self):
return "<Results: {}>".format(len(self.children))
class PropertyNode:
def __init__(self, key, value):
self.key = key
self.value = value
def __repr__(self):
return "{{{}: {}}".format(self.key, self.value)
class ListGroupNode:
def __init__(self):
self.children = []
def append(self, child):
self.children.append(child)
def get_raw(self):
return '\n'.join([c.get_raw() for c in self.children])
def __repr__(self):
return "<List: {}>".format(len(self.children))
class TableNode:
def __init__(self):
self.children = []
def append(self, child):
self.children.append(child)
def __repr__(self):
return "<Table: {}>".format(len(self.children))
class TableSeparatorRow:
def __init__(self, orig=None):
self.orig = orig
class TableRow:
def __init__(self, cells, orig=None):
self.cells = cells
self.orig = orig
class Text:
def __init__(self, content):
self.content = content
def get_raw(self):
return ''.join(self.content.get_raw())
class ListItem:
def __init__(self, tag, content, orig=None):
self.tag = tag
self.content = content
self.orig = orig
def get_raw(self):
return get_raw_contents(self.orig)
class BlockNode:
def __init__(self):
self.children = []
def append(self, child):
self.children.append(child)
class CodeBlock(BlockNode):
def __init__(self, header, subtype):
super().__init__()
self.header = header
self.lines = None
self.subtype = subtype
def set_lines(self, lines):
self.lines = lines
def __repr__(self):
return "<Code: {}>".format(len(self.lines))
from .utils import get_raw_contents