BigUpdate

md2gost/bibliography_renderable.py

@@ -0,0 +1,59 @@
+"""Bibliography renderable (docx-dependent)."""
+
+from __future__ import annotations
+
+from copy import copy
+from typing import Generator
+
+from docx.shared import Parented
+
+from .bibliography import BiblioEntry
+from .layout_tracker import LayoutState
+from .renderable import Renderable
+from .renderable.paragraph import Paragraph
+from .rendered_info import RenderedInfo
+from .sub_renderable import SubRenderable
+
+
+class Bibliography(Renderable):
+    def __init__(self, parent: Parented, entries: list[BiblioEntry],
+                 sections: list[tuple[str, list[BiblioEntry]]] | None = None):
+        self._parent = parent
+        self._entries = entries
+        self._sections = sections
+        self._paragraphs: list[Paragraph] = []
+
+        if sections:
+            for title, sect_entries in sections:
+                h = Paragraph(parent)
+                try:
+                    h.style = "Bibliography Heading"
+                except KeyError:
+                    pass
+                h.first_line_indent = 0
+                h.add_run(title.upper())
+                self._paragraphs.append(h)
+                for e in sect_entries:
+                    self._paragraphs.append(self._make_entry(parent, e))
+        else:
+            for e in entries:
+                self._paragraphs.append(self._make_entry(parent, e))
+
+    @staticmethod
+    def _make_entry(parent: Parented, e: BiblioEntry) -> Paragraph:
+        p = Paragraph(parent)
+        try:
+            p.style = "Bibliography"
+        except KeyError:
+            pass
+        # Avoid hyphen-splitting URLs: put whole entry in one run via direct text
+        run = p._docx_paragraph.add_run(f"{e.key}. {e.text}")
+        return p
+
+    def render(self, previous_rendered: RenderedInfo, layout_state: LayoutState) -> Generator[
+            RenderedInfo | SubRenderable, None, None]:
+        for paragraph in self._paragraphs:
+            for x in paragraph.render(previous_rendered, copy(layout_state)):
+                layout_state.add_height(x.height)
+                previous_rendered = x
+                yield x