org-rw/org_rw/dom.py
Sergio Martínez Portela 8ca480ad77 fix: Apply black formatter.
2024-08-18 22:47:24 +02:00

133 lines
2.6 KiB
Python

from typing import List, Optional, Union
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, arguments):
super().__init__()
self.header = header
self.lines: Optional[List] = None
self.subtype = subtype
self.arguments = arguments
def set_lines(self, lines):
self.lines = lines
def __repr__(self):
return "<Code: {}>".format(len(self.lines or []))
DomNode = Union[
DrawerNode,
PropertyNode,
ListGroupNode,
TableNode,
TableSeparatorRow,
TableRow,
Text,
ListItem,
BlockNode,
]
ContainerDomNode = Union[
DrawerNode,
ListGroupNode,
TableNode,
BlockNode,
]
from .utils import get_raw_contents