Igor20264
37 lines
1.2 KiB
Python
37 lines
1.2 KiB
Python
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
|