BigUpdate

md2gost/parser_.py

@@ -0,0 +1,36 @@
+from collections.abc import Generator
+
+from docx import Document
+from marko.block import BlankLine
+
+from .extended_markdown import markdown, Caption
+from .renderable.caption import CaptionInfo
+from .renderable.renderable import Renderable
+from .renderable_factory import RenderableFactory
+
+
+class Parser:
+    """Parses given markdown string and returns Renderable elements"""
+
+    def __init__(self, document: Document, text: str):
+        self._document = document
+        self._parsed = markdown.parse(text)
+        self._caption_info: CaptionInfo | None = None
+
+    def parse(self) -> Generator[Renderable, None, None]:
+        factory = RenderableFactory(self._document._body)
+
+        for marko_element in self._parsed.children:
+            if isinstance(marko_element, BlankLine):
+                continue
+
+            if isinstance(marko_element, Caption):
+                self._caption_info = CaptionInfo(
+                    marko_element.unique_name,
+                    marko_element.text,
+                    getattr(marko_element, "with_listing", False),
+                )
+                continue
+
+            yield factory.create(marko_element, self._caption_info)
+            self._caption_info = None