tesorotools-python 0.0.11__tar.gz → 0.0.13__tar.gz

{tesorotools_python-0.0.11 → tesorotools_python-0.0.13}/PKG-INFO

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

{tesorotools_python-0.0.11 → tesorotools_python-0.0.13}/pyproject.toml

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

{tesorotools_python-0.0.11 → tesorotools_python-0.0.13}/tesorotools/render/content/table.py

@@ -120,7 +120,9 @@ def _fill_index_names(
     start_row: int = 2 if horizontal else 1
 
     index_names: pd.Index = (
-        index if horizontal else index.get_level_values(level=1)
+        index
+        if (horizontal or isinstance(index, pd.Index))
+        else index.get_level_values(level=1)
     )
 
     for idx, name in enumerate(index_names, start=start_row):
@@ -192,8 +194,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
@@ -279,6 +285,10 @@ class Table:
         )
 
     def render(self, document: Document) -> Document:
+        heading = document.add_heading(self._title, level=2)
+        heading.alignment = CENTER
+        heading.runs[0].font.size = 10
+
         document = render_table(
             self._data,
             self._color,