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

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

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

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

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

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

@@ -251,6 +251,7 @@ class Table:
         shade_file: Path | None = None,
         block_sep: bool = False,
         title: str | None = None,
+        columns: list[str] | None = None,
     ):
         if (
             (data_file is None)
@@ -267,6 +268,12 @@ class Table:
         self._shade: pd.DataFrame | None = (
             pd.read_feather(shade_file) if shade_file is not None else None
         )
+
+        if columns is not None:
+            self._data = self._data[columns]
+            self._color = self._color[columns]
+            self._shade = self._shade[columns]
+
         self._title: str | None = title
         self._block_sep: bool = block_sep
 

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.19/tesorotools/utils/__init__.py

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