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

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

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

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

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

tesorotools_python-0.0.20/tesorotools/database/__init__.py

@@ -0,0 +1 @@
+from tesorotools.database.local import LocalDatabase, ShortcutDatabase

tesorotools_python-0.0.20/tesorotools/database/local.py

@@ -0,0 +1,46 @@
+"""
+Este módulo trata de recoger todas las comunalidades de las bases de datos locales que tenemos en OneDrive, para estandarizar el trabajo con ellas y evitarnos dolores de cabeza en el futuro.
+"""
+
+from pathlib import Path
+
+from tesorotools.utils import SYSTEM
+from tesorotools.utils.shortcuts import resolve_shortcut
+
+
+class LocalDatabase:
+    """
+    Todas las bases de datos locales que tenemos deben ser una instancia de esta, ya sea directamente o a través de una subclase
+    """
+
+    def __init__(self, root_path: Path):
+        self.root_path: Path = root_path
+
+    def get_year_path(self, year: int) -> Path:
+        return self.root_path / str(year)
+
+    def get_raw_path(self, year: int) -> Path:
+        return self.get_year_path(year) / "raw"
+
+    def get_processed_path(self, year: int) -> Path:
+        processed_path: Path = self.get_year_path(year) / "processed"
+        processed_path.mkdir(parents=True, exist_ok=True)
+        return processed_path
+
+    def get_products_path(self, year: int) -> Path:
+        products_path: Path = self.get_year_path(year) / "products"
+        products_path.mkdir(parents=True, exist_ok=True)
+        return products_path
+
+
+class ShortcutDatabase(LocalDatabase):
+    """Base de datos local cuya ruta viene data a través de un acceso directo (Windows) o enlace simbólico (Linux)"""
+
+    def __init__(self, root_path: Path, shortcut: str):
+        db_path = (
+            root_path / f"{shortcut}.lnk"
+            if SYSTEM == "Windows"
+            else root_path / shortcut
+        )
+        db_resolved_path: Path = resolve_shortcut(db_path)
+        super().__init__(root_path=db_resolved_path)

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

@@ -127,6 +127,7 @@ 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)
 
@@ -251,6 +252,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 +269,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
 
@@ -298,6 +306,9 @@ 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.20/tesorotools/utils/__init__.py

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

{tesorotools_python-0.0.18 → tesorotools_python-0.0.20}/tesorotools/utils/globals.py

@@ -1,5 +1,9 @@
+import platform
 from pathlib import Path
 
+SYSTEM: str = platform.system()
+
+
 BASE_PATH: Path = Path(__file__).parent.parent
 
 DEBUG: Path = Path("debug")

{tesorotools_python-0.0.18 → tesorotools_python-0.0.20}/tesorotools/utils/shortcuts.py

@@ -5,7 +5,6 @@ import win32com.client
 from win32com.client import CDispatch
 
 
-# this a good candidate to be added to tesorotools-python
 def resolve_shortcut(shortcut: Path) -> Path:
     """Returns the real path behind a shortcut