iatoolkit 0.7.15__py3-none-any.whl → 0.8.0__py3-none-any.whl

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/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(self.app.instance_path, '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/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.dist-info → iatoolkit-0.8.0.dist-info}/METADATA

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

{iatoolkit-0.7.15.dist-info → iatoolkit-0.8.0.dist-info}/RECORD

@@ -2,11 +2,11 @@ iatoolkit/__init__.py,sha256=yi7Uw68KajDWUUo-v4xV31Nt7hjr_-2gCXbkhdiDl44,1887
 iatoolkit/base_company.py,sha256=gM3X7xedNaVLodOObvjUN9eJKPO5-HmSLaBFa9jowCk,4277
 iatoolkit/cli_commands.py,sha256=G5L9xQXZ0lVFXQWBaE_KEZHyfuiT6PL1nTQRoSdnBzc,2302
 iatoolkit/company_registry.py,sha256=tduqt3oV8iDX_IB1eA7KIgvIxE4edTcy-3qZIXh3Lzw,2549
-iatoolkit/iatoolkit.py,sha256=UTixRTy0HoOMfO01H58IhC6BJFfZFZU_cbyOA25gkmM,15888
+iatoolkit/iatoolkit.py,sha256=S6PQ_rty7wD6XwXXAUmaWc2e2D2Sqj3epqnP4KK3nYE,16831
 iatoolkit/common/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 iatoolkit/common/auth.py,sha256=8NH6MQXfddLQd1GxrO2op3IYrrP4SMQKoKzj1o1jZmc,8486
 iatoolkit/common/exceptions.py,sha256=EXx40n5htp7UiOM6P1xfJ9U6NMcADqm62dlFaKz7ICU,1154
-iatoolkit/common/routes.py,sha256=zs883Xa1zoJgUYS4HKDUnnR4-fa-Sxg0SLWmU1Q1nPo,4378
+iatoolkit/common/routes.py,sha256=8_wvrFiB5w5Sq-5gZfIDz9JVPROd9JED5oxkX1N87gY,4923
 iatoolkit/common/session_manager.py,sha256=7D_RuJs60w-1zDr3fOGEz9JW7IZlSXuUHgUT87CzaUo,472
 iatoolkit/common/util.py,sha256=08js3KLJTXICOd5sgwDp2u_kDaZO_0xG4BIuzWZnLo8,15535
 iatoolkit/infra/__init__.py,sha256=5JqK9sZ6jBuK83zDQokUhxQ0wuJJJ9DXB8pYCLkX7X4,102
@@ -38,7 +38,7 @@ iatoolkit/services/__init__.py,sha256=5JqK9sZ6jBuK83zDQokUhxQ0wuJJJ9DXB8pYCLkX7X
 iatoolkit/services/benchmark_service.py,sha256=CdbFYyS3FHFhNzWQEa9ZNjUlmON10DT1nKNbZQ1EUi8,5880
 iatoolkit/services/dispatcher_service.py,sha256=ykR1ye6McyCCuaBgwH6r3-PqcLAr4v4ApkPazMSBzbs,14040
 iatoolkit/services/document_service.py,sha256=nMXrNtbHQuc9pSaten0LvKY0kT8_WngBDmZJUP3jNPw,5936
-iatoolkit/services/excel_service.py,sha256=etb77aCOiHe3_6hG7BKxXn23P1BF1tXMp6YhYYSVWA4,3579
+iatoolkit/services/excel_service.py,sha256=CJGhu7cQl9J6y_ZWSJ-M63Xm-RXR9Zs66oOR2NJErZQ,3868
 iatoolkit/services/file_processor_service.py,sha256=B1sUUhZNFf-rT4_1wrD38GKNoBFMp2g0dYrXYMCWe2E,4122
 iatoolkit/services/history_service.py,sha256=JwW958tYTrcdWSP5TRoDgDtBdGCvWIG4fnmHsrXLQ-8,1617
 iatoolkit/services/jwt_service.py,sha256=YoZ9h7_o9xBko-arNQv4MbcwnxoSWVNj4VbZmMo_QGY,3908
@@ -69,25 +69,6 @@ iatoolkit/static/styles/chat_iatoolkit.css,sha256=ZPQAgLT0HIeXjZC_EY8xACmVdjEp7s
 iatoolkit/static/styles/chat_info.css,sha256=17DbgoNYE21VYWfb5L9-QLCpD2R1idK4imKRLwXtJLY,1058
 iatoolkit/static/styles/chat_modal.css,sha256=67qwltu-QS-sU0d_fSeTIiXwcNjfAnQQtmsgStx-ezk,3047
 iatoolkit/static/styles/llm_output.css,sha256=AlxgRSOleeCk2dLAqFWVaQ-jwZiJjcpC5rHuUv3T6VU,2312
-iatoolkit/static/temp/024f44a9-cc7e-4bde-9a3c-11903a8f5d3d.xlsx,sha256=_6n2A4apLWkQvORpQOAAKGYlQyCHlCUDPJDzJ4Ons14,8788
-iatoolkit/static/temp/0b97768e-79e8-43ec-b17e-ba3137f94e93.xlsx,sha256=hxkJNPzqfSRcYUO6sxD1X8O8o0ULO0lBCj0ifsDfwBU,6937
-iatoolkit/static/temp/202883ee-763e-4b40-9bb6-bfacfc5e65fe.xlsx,sha256=reGomquStUq8iH3rByXJSPH4DH7MpdGxqy3puQ3jYHM,6103
-iatoolkit/static/temp/28287491-08b7-4863-a2a3-49fcb64a0906.xlsx,sha256=7rQMpkXfXlFdyAhTWM3m_7EO-FCVNXLk1Hexd97mspk,9517
-iatoolkit/static/temp/36780cac-7a46-4db4-ac98-7338a51aaf52.xlsx,sha256=DUKW7DUaMAgFbKHm3vYK6GezRMq58fHBJCNMokyuQ9k,5426
-iatoolkit/static/temp/5c1b66f6-d58f-4684-8a7a-df3bb1a35eaa.xlsx,sha256=c8157lfVgtcbw94fqS1GkW2eO3VZ9MVtkWouC4jC_Lo,7605
-iatoolkit/static/temp/5d5a3500-ec57-4e07-a554-8799a906d1ab.xlsx,sha256=8dnb4nba-5g2aRYXSBwVSJ5EV4Ah-4dCV7cqSxhJVgk,5794
-iatoolkit/static/temp/65887e40-cf64-49aa-8d1f-651cb0f8cdf0.xlsx,sha256=_AGAhHHQo9pCXOgvDZ83K25dOrelfuxwjTJW85QcWGM,9055
-iatoolkit/static/temp/6fa64b13-e8e5-40ad-8257-00fd1682dad8.xlsx,sha256=FYOJNCZOwXFK_2Ork7vL4pGqZyJR5hLv09RrH_tosJk,7339
-iatoolkit/static/temp/7ab7071f-ad9b-49e6-8e63-f08410625ce7.xlsx,sha256=JWbGzhnwErg4V5jrCXGUZafxduCKcPHagM8TTsAMJro,5806
-iatoolkit/static/temp/824346d9-d54d-40c6-b5c2-9e1d84d0ae90.xlsx,sha256=czRFxIL1opDDaqnDePF8aqIu7tQDmDB8hvyuZRes09w,5843
-iatoolkit/static/temp/9ca80c3d-d196-4dfc-8179-582584fae04c.xlsx,sha256=DUQWwcMkq3UkUCr3TxqJuHx3ODwS5jxYCrgHnBaxcCI,6525
-iatoolkit/static/temp/a34cb8a6-85fb-4ea8-aabe-889967cd83b5.xlsx,sha256=3O6b2c29sYOZ1vfW9eHnNl-7bpyxcyX9h2MC6Xlq0JA,8968
-iatoolkit/static/temp/a7fc9c13-c509-4499-b4be-23bfa57cac31.xlsx,sha256=EIO1XPlO-TBQq5gd8pJ3t2SiEwDkFynkkALSXP4yMog,5379
-iatoolkit/static/temp/b83084f5-fe54-4580-885e-412b4388cbda.xlsx,sha256=mQc6ZBWOLLUMhPJmsfqlb0heAVCrwjkt65mqkVg3Y7c,9041
-iatoolkit/static/temp/c17c6864-34e1-448f-b0e9-380354256ea9.xlsx,sha256=WDvGGyxAhl96CeV-1Co9OGCOBmMIOD1UODHzj-wA0a0,5266
-iatoolkit/static/temp/customer_clusters.parquet,sha256=vlzC453Aqu-QXOzCksgeyKleuHFctr-fEcVy4ejJJak,15177
-iatoolkit/static/temp/customer_clusters.xlsx,sha256=7lN6Q0_GDj8VI5_-oMPbFHT9DxBRiK64PVH6js3PWNI,9920
-iatoolkit/static/temp/d1af98b8-18a9-4b94-b9bc-607d19a87d0d.xlsx,sha256=lY4DGo1L-bLcK3j5wioLC0SwDLSw0SjyKUmISaKuSdU,5787
 iatoolkit/system_prompts/format_styles.prompt,sha256=MSMe1qvR3cF_0IbFshn8R0z6Wx6VCHQq1p37rpu5wwk,3576
 iatoolkit/system_prompts/query_main.prompt,sha256=w_9ybgWgiQH4V_RbAXqsvz0M7oOuiyhxcwf-D0CgfA4,3017
 iatoolkit/system_prompts/sql_rules.prompt,sha256=y4nURVnb9AyFwt-lrbMNBHHtZlhk6kC9grYoOhRnrJo,59174
@@ -188,7 +169,7 @@ tests/views/test_tasks_review_view.py,sha256=HXbAY99rbcdg-ZoaU3kxp-jNTPgOhgiTyKm
 tests/views/test_tasks_view.py,sha256=TNdV69Vt17uc-vBI37A7qtRNqyPho2MoW7cgGxQKecU,4551
 tests/views/test_user_feedback_view.py,sha256=IfKOpsncmf5ciMDsGuPtro9GdKXkxhRzsysVlEEM_HA,9181
 tests/views/test_verify_user_view.py,sha256=dxQ3ibOETEuP8M50Gmwbmzj2L6UU7PAyC91Wp9Xs0K0,4772
-iatoolkit-0.7.15.dist-info/METADATA,sha256=2lRc7VvYw7ALxcXjgbUAS0jzvZu1kWF-WACrFGWDyoI,9301
-iatoolkit-0.7.15.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-iatoolkit-0.7.15.dist-info/top_level.txt,sha256=or0Ar3Su6BhTy86zRrUwMAWtsR8Nk-tFEwdC0CZpKCs,16
-iatoolkit-0.7.15.dist-info/RECORD,,
+iatoolkit-0.8.0.dist-info/METADATA,sha256=vlvBB4vIGEe3KZeM1Z0H4eNNExJ_MGsF-u76hXSJRoY,9300
+iatoolkit-0.8.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+iatoolkit-0.8.0.dist-info/top_level.txt,sha256=or0Ar3Su6BhTy86zRrUwMAWtsR8Nk-tFEwdC0CZpKCs,16
+iatoolkit-0.8.0.dist-info/RECORD,,