tesorotools-python 0.0.26__tar.gz → 0.0.27__tar.gz

{tesorotools_python-0.0.26 → tesorotools_python-0.0.27}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tesorotools-python
-Version: 0.0.26
+Version: 0.0.27
 Requires-Python: >=3.13
 Requires-Dist: babel>=2.17
 Requires-Dist: eikon>=1.1

{tesorotools_python-0.0.26 → tesorotools_python-0.0.27}/pyproject.toml

@@ -1,7 +1,7 @@
 [project]
 name = "tesorotools-python"
 requires-python = ">=3.13"
-version = "0.0.26"
+version = "0.0.27"
 dependencies = [
     # database and ORM
     "psycopg[binary]>=3.1",

{tesorotools_python-0.0.26 → tesorotools_python-0.0.27}/src/tesorotools/artists/stacked.py

@@ -139,6 +139,13 @@ class StackedBarPlot:
 
     Positive and negative values are stacked separately so
     that bars extend in both directions from the baseline.
+
+    The ``plot`` method delegates to overridable hooks so
+    subclasses can customise individual steps without
+    reimplementing the full render pipeline:
+
+    * ``_prepare_data`` — slice, scale, resample
+    * ``_format_xticks`` — tick positions, labels, rotation
     """
 
     def __init__(
@@ -156,6 +163,8 @@ class StackedBarPlot:
         figsize: tuple[float, float] | None = None,
         overlay_series: dict[str, str] | None = None,
         plot_size: tuple[float, float] | None = None,
+        bar_width: float = 0.7,
+        x_rotation: float = 0,
     ) -> None:
         if out_path.suffix != ".png":
             raise ValueError(f"out_path must be .png: {out_path}")
@@ -171,6 +180,8 @@ class StackedBarPlot:
         self.plot_size = plot_size
         self.figsize = figsize
         self.overlay_series = overlay_series or {}
+        self.bar_width = bar_width
+        self.x_rotation = x_rotation
 
     @classmethod
     def from_yaml(cls, loader: TemplateLoader, node: MappingNode) -> Self:
@@ -182,7 +193,13 @@ class StackedBarPlot:
         cfg["data"] = pd.read_feather(cfg.pop("data_path"))
         return cls(**cfg)
 
-    def plot(self) -> Axes:
+    # -- Overridable hooks -----------------------------------
+
+    def _prepare_data(self) -> pd.DataFrame:
+        """Slice, scale, and return plot-ready DataFrame.
+
+        Override to resample, change the index type, etc.
+        """
         start = (
             pd.Timestamp(self.start_date)
             if self.start_date
@@ -193,10 +210,35 @@ class StackedBarPlot:
             if self.end_date
             else self.data.index.max()
         )
-
         all_cols = list(self.series.keys()) + list(self.overlay_series.keys())
         plot_data = self.data.loc[start:end, all_cols].dropna()
-        plot_data = plot_data * self.scale
+        return plot_data * self.scale
+
+    def _format_xticks(
+        self,
+        ax: Axes,
+        plot_data: pd.DataFrame,
+        x: np.ndarray[tuple[int], np.dtype[np.intp]],
+    ) -> None:
+        """Set tick positions, labels, and rotation.
+
+        Override for custom labels (e.g. string index).
+        """
+        dates = plot_data.index
+        step = max(1, len(dates) // 12)
+        tick_pos = list(range(0, len(dates), step))
+        tick_labels = [dates[i].strftime("%Y") for i in tick_pos]
+        ax.set_xticks(  # type: ignore[reportUnknownMemberType]
+            tick_pos
+        )
+        ax.set_xticklabels(  # type: ignore[reportUnknownMemberType]
+            tick_labels, rotation=self.x_rotation
+        )
+
+    # -- Main render -----------------------------------------
+
+    def plot(self) -> Axes:
+        plot_data = self._prepare_data()
 
         fig_kw = dict(FIG_CONFIG)
         if self.figsize is not None:
@@ -210,7 +252,6 @@ class StackedBarPlot:
         labels = list(self.series.values())
 
         x = np.arange(len(plot_data))
-        bar_width = 0.7
 
         pos_bottom: np.ndarray[tuple[int], np.dtype[np.float64]] = np.zeros(
             len(plot_data)
@@ -235,7 +276,7 @@ class StackedBarPlot:
                     x,
                     pos,
                     bottom=pos_bottom,
-                    width=bar_width,
+                    width=self.bar_width,
                     label=label,
                 )
                 .patches[0]
@@ -245,7 +286,7 @@ class StackedBarPlot:
                 x,
                 neg,
                 bottom=neg_bottom,
-                width=bar_width,
+                width=self.bar_width,
                 color=color,
             )
             pos_bottom = pos_bottom + pos
@@ -262,16 +303,7 @@ class StackedBarPlot:
                 zorder=10,
             )
 
-        dates = plot_data.index
-        step = max(1, len(dates) // 12)
-        tick_pos = list(range(0, len(dates), step))
-        tick_labels = [dates[i].strftime("%Y") for i in tick_pos]
-        ax.set_xticks(  # type: ignore[reportUnknownMemberType]
-            tick_pos
-        )
-        ax.set_xticklabels(  # type: ignore[reportUnknownMemberType]
-            tick_labels
-        )
+        self._format_xticks(ax, plot_data, x)
 
         style_spines(
             ax,