iatoolkit 0.7.15__tar.gz → 0.8.1__tar.gz

{iatoolkit-0.7.15 → iatoolkit-0.8.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: iatoolkit
-Version: 0.7.15
+Version: 0.8.1
 Summary: IAToolkit
 Author: Fernando Libedinsky
 License-Expression: MIT

{iatoolkit-0.7.15 → iatoolkit-0.8.1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "iatoolkit"
-version = "0.7.15"
+version = "0.8.1"
 requires-python = ">=3.11"
 description = "IAToolkit"
 readme = "readme.md"

{iatoolkit-0.7.15 → iatoolkit-0.8.1}/src/iatoolkit/common/routes.py

@@ -3,7 +3,7 @@
 #
 # IAToolkit is open source software.
 
-from flask import render_template, redirect, flash, url_for,send_from_directory, current_app
+from flask import render_template, redirect, flash, url_for,send_from_directory, current_app, abort
 from iatoolkit.common.session_manager import SessionManager
 from flask import jsonify
 from iatoolkit.views.history_view import HistoryView
@@ -80,7 +80,22 @@ def register_views(injector, app):
 
     @app.route('/download/<path:filename>')
     def download_file(filename):
-        temp_dir = os.path.join(current_app.root_path, 'static', 'temp')
-        return send_from_directory(temp_dir, filename, as_attachment=True)
-
+        """
+        Esta vista sirve un archivo previamente generado desde el directorio
+        configurado en IATOOLKIT_DOWNLOAD_DIR.
+        """
+        # Valida que la configuración exista
+        if 'IATOOLKIT_DOWNLOAD_DIR' not in current_app.config:
+            abort(500, "Error de configuración: IATOOLKIT_DOWNLOAD_DIR no está definido.")
+
+        download_dir = current_app.config['IATOOLKIT_DOWNLOAD_DIR']
+
+        try:
+            return send_from_directory(
+                download_dir,
+                filename,
+                as_attachment=True  # Fuerza la descarga en lugar de la visualización
+            )
+        except FileNotFoundError:
+            abort(404)
 

{iatoolkit-0.7.15 → iatoolkit-0.8.1}/src/iatoolkit/iatoolkit.py

@@ -97,6 +97,9 @@ class IAToolkit:
         self._setup_cli_commands()
         self._setup_context_processors()
 
+        # Step 8: define the download_dir for excel's
+        self._setup_download_dir()
+
         try:
             self.version = _pkg_version("iatoolkit")
         except PackageNotFoundError:
@@ -404,6 +407,26 @@ class IAToolkit:
             )
         return self.db_manager
 
+    def _setup_download_dir(self):
+        # 1. set the default download directory
+        default_download_dir = os.path.join(os.getcwd(), 'downloads')
+
+        # 3. if user specified one, use it
+        download_dir = self.app.config.get('IATOOLKIT_DOWNLOAD_DIR', default_download_dir)
+
+        # 3. save it in the app config
+        self.app.config['IATOOLKIT_DOWNLOAD_DIR'] = download_dir
+
+        # 4. make sure the directory exists
+        try:
+            os.makedirs(download_dir, exist_ok=True)
+        except OSError as e:
+            raise IAToolkitException(
+                IAToolkitException.ErrorType.CONFIG_ERROR,
+                "No se pudo crear el directorio de descarga. Verifique que el directorio existe y tenga permisos de escritura."
+            )
+        logging.info(f"✅ download dir created in: {download_dir}")
+
 
 def current_iatoolkit() -> IAToolkit:
     return IAToolkit.get_instance()

{iatoolkit-0.7.15 → iatoolkit-0.8.1}/src/iatoolkit/services/excel_service.py

@@ -55,7 +55,13 @@ class ExcelService:
 
             # 3. create temporary name
             token = f"{uuid4()}.xlsx"
-            filepath = Path("static/temp") / token
+
+            # 4. check that download directory is configured
+            if 'IATOOLKIT_DOWNLOAD_DIR' not in current_app.config:
+                return 'no esta configurado el directorio temporal para guardar excels'
+
+            download_dir = current_app.config['IATOOLKIT_DOWNLOAD_DIR']
+            filepath = Path(download_dir) / token
             filepath.parent.mkdir(parents=True, exist_ok=True)
 
             # 4. save excel file in temporary directory

{iatoolkit-0.7.15 → iatoolkit-0.8.1}/src/iatoolkit.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: iatoolkit
-Version: 0.7.15
+Version: 0.8.1
 Summary: IAToolkit
 Author: Fernando Libedinsky
 License-Expression: MIT

{iatoolkit-0.7.15 → iatoolkit-0.8.1}/src/iatoolkit.egg-info/SOURCES.txt

@@ -77,25 +77,6 @@ src/iatoolkit/static/styles/chat_iatoolkit.css
 src/iatoolkit/static/styles/chat_info.css
 src/iatoolkit/static/styles/chat_modal.css
 src/iatoolkit/static/styles/llm_output.css
-src/iatoolkit/static/temp/024f44a9-cc7e-4bde-9a3c-11903a8f5d3d.xlsx
-src/iatoolkit/static/temp/0b97768e-79e8-43ec-b17e-ba3137f94e93.xlsx
-src/iatoolkit/static/temp/202883ee-763e-4b40-9bb6-bfacfc5e65fe.xlsx
-src/iatoolkit/static/temp/28287491-08b7-4863-a2a3-49fcb64a0906.xlsx
-src/iatoolkit/static/temp/36780cac-7a46-4db4-ac98-7338a51aaf52.xlsx
-src/iatoolkit/static/temp/5c1b66f6-d58f-4684-8a7a-df3bb1a35eaa.xlsx
-src/iatoolkit/static/temp/5d5a3500-ec57-4e07-a554-8799a906d1ab.xlsx
-src/iatoolkit/static/temp/65887e40-cf64-49aa-8d1f-651cb0f8cdf0.xlsx
-src/iatoolkit/static/temp/6fa64b13-e8e5-40ad-8257-00fd1682dad8.xlsx
-src/iatoolkit/static/temp/7ab7071f-ad9b-49e6-8e63-f08410625ce7.xlsx
-src/iatoolkit/static/temp/824346d9-d54d-40c6-b5c2-9e1d84d0ae90.xlsx
-src/iatoolkit/static/temp/9ca80c3d-d196-4dfc-8179-582584fae04c.xlsx
-src/iatoolkit/static/temp/a34cb8a6-85fb-4ea8-aabe-889967cd83b5.xlsx
-src/iatoolkit/static/temp/a7fc9c13-c509-4499-b4be-23bfa57cac31.xlsx
-src/iatoolkit/static/temp/b83084f5-fe54-4580-885e-412b4388cbda.xlsx
-src/iatoolkit/static/temp/c17c6864-34e1-448f-b0e9-380354256ea9.xlsx
-src/iatoolkit/static/temp/customer_clusters.parquet
-src/iatoolkit/static/temp/customer_clusters.xlsx
-src/iatoolkit/static/temp/d1af98b8-18a9-4b94-b9bc-607d19a87d0d.xlsx
 src/iatoolkit/system_prompts/format_styles.prompt
 src/iatoolkit/system_prompts/query_main.prompt
 src/iatoolkit/system_prompts/sql_rules.prompt