tesorotools-python 0.0.15__tar.gz → 0.0.16__tar.gz

{tesorotools_python-0.0.15 → tesorotools_python-0.0.16}/PKG-INFO

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

{tesorotools_python-0.0.15 → tesorotools_python-0.0.16}/pyproject.toml

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

{tesorotools_python-0.0.15 → tesorotools_python-0.0.16}/tesorotools/artists/barh_plot.py

@@ -289,7 +289,7 @@ def _normalize_from_flash(
 
 
 def plot_barh_charts_from_flash(
-    flash: pd.DataFrame, config_dicts: dict[str, dict]
+    out_path: Path, flash: pd.DataFrame, config_dicts: dict[str, dict]
 ):
     for name, config in config_dicts.items():
         blocks: dict[str, Any] = config.get("blocks", None)
@@ -306,5 +306,5 @@ def plot_barh_charts_from_flash(
             units_bar=config["format"]["units"],
             units_axis=config.get("axis_format", {"units": ""})["units"],
         )
-        out_file = Path(DEBUG / "barh" / f"{name}.png")
+        out_file = out_path / f"{name}.png"
         _plot_barh_chart(out_file, standard_dict, alias, **config)

{tesorotools_python-0.0.15 → tesorotools_python-0.0.16}/tesorotools/artists/line_plot.py

@@ -100,7 +100,9 @@ def plot_line_chart(
     fig.savefig(out_name)
 
 
-def plot_line_charts(data: pd.DataFrame, config_dicts: dict[str, Any]):
+def plot_line_charts(
+    out_path: Path, data: pd.DataFrame, config_dicts: dict[str, Any]
+):
     for name, config in config_dicts.items():
         start_date: pd.Timestamp = pd.to_datetime(config["start_date"])
         end_date_str: str | None = config["end_date"]
@@ -114,7 +116,7 @@ def plot_line_charts(data: pd.DataFrame, config_dicts: dict[str, Any]):
             slice(start_date, end_date), series.keys()
         ]
         trimmed_data = trimmed_data.rename(columns=series)
-        out_name: Path = DEBUG / "line" / f"{name}.png"
+        out_name: Path = out_path / f"{name}.png"
         plot_line_chart(out_name, trimmed_data, **config)
 
 

{tesorotools_python-0.0.15 → tesorotools_python-0.0.16}/tesorotools/artists/table.py

@@ -1,4 +1,5 @@
 from math import floor
+from pathlib import Path
 from typing import Any
 
 import pandas as pd
@@ -186,7 +187,7 @@ def generate_table(table_data: pd.DataFrame, table_cfg: dict[str, Any]):
 
 
 def generate_tables_from_flash(
-    flash: pd.DataFrame, config_dicts: dict[str, dict]
+    out_path: Path, flash: pd.DataFrame, config_dicts: dict[str, dict]
 ):
     for table_name, table_cfg in config_dicts.items():
         series: list[str] = list(collect_series(table_cfg))
@@ -194,6 +195,6 @@ def generate_tables_from_flash(
         formatted_table, color_table, shade_table = generate_table(
             table_data, table_cfg
         )
-        formatted_table.to_feather(DEBUG / "table" / f"{table_name}.feather")
-        color_table.to_feather(DEBUG / "table" / f"{table_name}_color.feather")
-        shade_table.to_feather(DEBUG / "table" / f"{table_name}_shade.feather")
+        formatted_table.to_feather(out_path / f"{table_name}.feather")
+        color_table.to_feather(out_path / f"{table_name}_color.feather")
+        shade_table.to_feather(out_path / f"{table_name}_shade.feather")

{tesorotools_python-0.0.15 → tesorotools_python-0.0.16}/tesorotools/artists/type_curve.py

@@ -199,7 +199,9 @@ def plot_type_curve(
 
 
 # data is expected to be a simple time series data, columns are series and rows represents dates
-def plot_type_curves(data: pd.DataFrame, config_dicts: dict[str, Any]):
+def plot_type_curves(
+    out_path: Path, data: pd.DataFrame, config_dicts: dict[str, Any]
+):
     for name, config in config_dicts.items():
         if not name.startswith("."):  # aux entries
             series: dict[str, str] = config["series"]
@@ -211,6 +213,6 @@ def plot_type_curves(data: pd.DataFrame, config_dicts: dict[str, Any]):
             trimmed_data: pd.DataFrame = trimmed_data.rename(columns=series)
             plot_type_curve(
                 data=trimmed_data,
-                out_file=DEBUG / "type_curve" / f"{name}.png",
+                out_file=out_path / f"{name}.png",
                 **config,
             )