tesorotools-python 0.0.20__tar.gz → 0.0.21__tar.gz

{tesorotools_python-0.0.20 → tesorotools_python-0.0.21}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tesorotools-python
-Version: 0.0.20
+Version: 0.0.21
 Requires-Python: >=3.13
 Requires-Dist: babel
 Requires-Dist: eikon

{tesorotools_python-0.0.20 → tesorotools_python-0.0.21}/pyproject.toml

@@ -1,7 +1,7 @@
 [project]
 name = "tesorotools-python"
 requires-python = ">=3.13"
-version = "0.0.20"
+version = "0.0.21"
 dependencies = [
     "psycopg2",
     "SQLAlchemy",

{tesorotools_python-0.0.20 → tesorotools_python-0.0.21}/tesorotools/render/content/table.py

@@ -127,7 +127,6 @@ def _fill_index_names(
 
     for idx, name in enumerate(index_names, start=start_row):
         cell: TableCell = table_docx.cell(idx, 0)
-        print(name)
         cell.text = name
         _style_index_names(cell)
 
@@ -306,9 +305,6 @@ class Table:
         heading.alignment = CENTER
         heading.runs[0].font.size = Pt(10)
 
-        print(self._data)
-        print(self._data.index)
-
         render_table(
             self._data,
             self._color,

tesorotools_python-0.0.21/tesorotools/utils/__init__.py

@@ -0,0 +1,3 @@
+from tesorotools.utils.format import format_table
+from tesorotools.utils.globals import SYSTEM
+from tesorotools.utils.matplotlib import format_annotation

tesorotools_python-0.0.21/tesorotools/utils/format.py

@@ -0,0 +1,38 @@
+from typing import Any
+
+import pandas as pd
+
+from tesorotools.utils.matplotlib import format_annotation
+
+
+def format_table(
+    table: pd.DataFrame, format_dict: dict[str, dict[str, Any]]
+) -> pd.DataFrame:
+    """Formats a pandas dataframe to get it ready for being rendered into a .docx file.
+
+    Parameters
+    ----------
+    table : pd.DataFrame
+    format_dict : dict[str, dict[str, Any]]
+        Each key in the dictionary is a column in the dataframe, each value is a dictionary with 3 possible entries: "scale" (integer, default=1), "decimals" (integer, default=0), "units" (string, default="").
+
+    Returns
+    -------
+    pd.DataFrame
+        The formatted dataframe
+    """
+    for column, fmt in format_dict.items():
+        if pd.api.types.is_numeric_dtype(table[column]):
+            scale = fmt.get("scale", 1)
+            table[column] = table[column] / scale
+            table[column] = table[column].apply(
+                lambda x: format_annotation(
+                    value=x,
+                    decimals=fmt.get("decimals", 0),
+                    units=fmt.get("units", ""),
+                )
+            )
+
+    for column in table.columns:
+        table[column] = table[column].astype(str)
+    return table

tesorotools_python-0.0.20/tesorotools/utils/__init__.py

@@ -1 +0,0 @@
-from tesorotools.utils.globals import SYSTEM