diff --git a/org_rw/org_rw.py b/org_rw/org_rw.py index 30fe7f2..f31abe1 100644 --- a/org_rw/org_rw.py +++ b/org_rw/org_rw.py @@ -818,7 +818,7 @@ def parse_headline(hl) -> Headline: ) -class OrgDom: +class OrgDoc: def __init__(self, headlines, keywords, contents): self.headlines: List[Headline] = list(map(parse_headline, headlines)) self.keywords: List[Property] = keywords @@ -976,16 +976,16 @@ class OrgDom: yield from self.dump_headline(headline) -class OrgDomReader: +class OrgDocReader: def __init__(self): self.headlines: List[Headline] = [] self.keywords: List[Property] = [] - self.headline_hierarchy: List[OrgDom] = [] + self.headline_hierarchy: List[OrgDoc] = [] self.contents: List[RawLine] = [] self.delimiters: List[DelimiterLine] = [] def finalize(self): - return OrgDom(self.headlines, self.keywords, self.contents) + return OrgDoc(self.headlines, self.keywords, self.contents) ## Construction def add_headline(self, linenum: int, match: re.Match) -> int: @@ -1131,7 +1131,7 @@ class OrgDomReader: def loads(s, environment=BASE_ENVIRONMENT, extra_cautious=True): - reader = OrgDomReader() + reader = OrgDocReader() reader.read(s, environment) doc = reader.finalize() if extra_cautious: # Check that all options can be properly re-serialized diff --git a/tests/test_org.py b/tests/test_org.py index 63d94b9..563a9d6 100644 --- a/tests/test_org.py +++ b/tests/test_org.py @@ -6,7 +6,7 @@ from datetime import datetime as DT from org_rw import dumps, load, loads from utils.assertions import (BOLD, CODE, HL, ITALIC, SPAN, STRIKE, UNDERLINED, - VERBATIM, WEB_LINK, Dom, Tokens) + VERBATIM, WEB_LINK, Doc, Tokens) DIR = os.path.dirname(os.path.abspath(__file__)) @@ -16,7 +16,7 @@ class TestSerde(unittest.TestCase): with open(os.path.join(DIR, "01-simple.org")) as f: doc = load(f) - ex = Dom( + ex = Doc( props=[ ("TITLE", "01-Simple"), ("DESCRIPTION", "Simple org file"), @@ -70,7 +70,7 @@ class TestSerde(unittest.TestCase): with open(os.path.join(DIR, "02-markup.org")) as f: doc = load(f) - ex = Dom( + ex = Doc( props=[ ("TITLE", "02-Markup"), ("DESCRIPTION", "Simple org file to test markup"), @@ -213,7 +213,7 @@ class TestSerde(unittest.TestCase): self.assertEqual(links[3].value, "id:03-markup-first-level-id") self.assertEqual(links[3].description, "a link to a section by id") - ex = Dom( + ex = Doc( props=[ ("TITLE", "03-Links"), ("DESCRIPTION", "Simple org file to test links"), @@ -293,7 +293,7 @@ class TestSerde(unittest.TestCase): links[3].value = "id:03-markup-non-existent-level-id" links[3].description = None - ex = Dom( + ex = Doc( props=[ ("TITLE", "03-Links"), ("DESCRIPTION", "Simple org file to test links"), diff --git a/tests/utils/assertions.py b/tests/utils/assertions.py index ebcdcf1..0e21424 100644 --- a/tests/utils/assertions.py +++ b/tests/utils/assertions.py @@ -19,7 +19,7 @@ def get_raw(doc): return doc.get_raw() -class Dom: +class Doc: def __init__(self, *, props=None, children=None): self.props = props self.children = children