tesorotools-python 0.0.10__tar.gz → 0.0.12__tar.gz

{tesorotools_python-0.0.10 → tesorotools_python-0.0.12}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tesorotools-python
-Version: 0.0.10
+Version: 0.0.12
 Requires-Dist: babel
 Requires-Dist: eikon
 Requires-Dist: lseg-data

{tesorotools_python-0.0.10 → tesorotools_python-0.0.12}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "tesorotools-python"
-version = "0.0.10"
+version = "0.0.12"
 dependencies = ["psycopg2", "SQLAlchemy", "pandas", "matplotlib", "pyarrow", "python-docx", "openpyxl", "PyYAML", "babel", "eikon", "lseg-data"]
 
 [build-system]

{tesorotools_python-0.0.10 → tesorotools_python-0.0.12}/tesorotools/render/__init__.py

@@ -1,6 +1,7 @@
 from tesorotools.render.content.images import Image, Images
 from tesorotools.render.content.section import Section
 from tesorotools.render.content.table import Table
+from tesorotools.render.content.text import Text
 from tesorotools.render.report import Report
 from tesorotools.utils.template import TemplateLoader
 
@@ -9,3 +10,4 @@ TemplateLoader.add_constructor("!section", Section.from_yaml)
 TemplateLoader.add_constructor("!image", Image.from_yaml)
 TemplateLoader.add_constructor("!images", Images.from_yaml)
 TemplateLoader.add_constructor("!table", Table.from_yaml)
+TemplateLoader.add_constructor("!text", Text.from_yaml)

{tesorotools_python-0.0.10 → tesorotools_python-0.0.12}/tesorotools/render/content/table.py

@@ -192,8 +192,12 @@ def _fill_content(
     for (x, y), value in np.ndenumerate(values):
         cell: TableCell = table_docx.cell(x + start_row, y + 1)
         cell.text = value if value is not None else ""
-        color: str | None = color_table.values[x, y]
-        shade: str | None = shade_table.values[x, y]
+        color: str | None = (
+            color_table.values[x, y] if color_table is not None else None
+        )
+        shade: str | None = (
+            shade_table.values[x, y] if shade_table is not None else None
+        )
         if color is not None:
             cell.paragraphs[0].runs[0].font.color.rgb = RGBColor.from_string(
                 color
@@ -239,9 +243,9 @@ class Table:
 
     def __init__(
         self,
-        data_file: Path | None,
-        color_file: Path | None,
-        shade_file: Path | None,
+        data_file: Path | None = None,
+        color_file: Path | None = None,
+        shade_file: Path | None = None,
         block_sep: bool = False,
         title: str | None = None,
     ):
@@ -271,7 +275,12 @@ class Table:
         data_file: Path = root_path / f"{file_prefix}_data.feather"
         color_file: Path = root_path / f"{file_prefix}_color.feather"
         shade_file: Path = root_path / f"{file_prefix}_shade.feather"
-        return cls(data_file, color_file, shade_file, **table_cfg)
+        return cls(
+            data_file,
+            color_file=color_file if color_file.exists() else None,
+            shade_file=shade_file if shade_file.exists() else None,
+            **table_cfg,
+        )
 
     def render(self, document: Document) -> Document:
         document = render_table(

tesorotools_python-0.0.12/tesorotools/render/content/text.py

@@ -0,0 +1,23 @@
+from typing import Any, Self
+
+from docx.document import Document
+from yaml import Loader, MappingNode
+
+from tesorotools.render.content.content import Content
+
+
+class Text:
+    def __init__(self, text: str) -> None:
+        self._text = text
+
+    @classmethod
+    def from_yaml(cls, loader: Loader, node: MappingNode) -> Self:
+        values: dict[str, Any] = loader.construct_mapping(node, deep=True)
+        values.pop("id")
+        text: str = values.pop("text", None)
+        section: Self = cls(text=text)
+        return section
+
+    def render(self, document: Document) -> Document:
+        document.add_paragraph(self._text)
+        return document