datatailr 0.1.44__tar.gz → 0.1.46__tar.gz

{datatailr-0.1.44/src/datatailr.egg-info → datatailr-0.1.46}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: datatailr
-Version: 0.1.44
+Version: 0.1.46
 Summary: Ready-to-Use Platform That Drives Business Insights
 Author-email: Datatailr <info@datatailr.com>
 License-Expression: MIT

{datatailr-0.1.44 → datatailr-0.1.46}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "datatailr"
-version = "0.1.44"
+version = "0.1.46"
 description = "Ready-to-Use Platform That Drives Business Insights"
 readme = "README.md"
 requires-python = ">=3.9"

{datatailr-0.1.44 → datatailr-0.1.46}/src/datatailr/scheduler/arguments_cache.py

@@ -161,8 +161,5 @@ class ArgumentsCache:
         :param path: Path in the Blob storage where the blob is stored.
         """
         path = path.replace("/tmp/", "")
-        try:
-            data = __BLOB_STORAGE__.get_blob(path)
-        except RuntimeError as e:
-            raise RuntimeError(f"Failed to retrieve blob from path {path}: {str(e)}")
+        data = __BLOB_STORAGE__.get_blob(path)
         return json.loads(data)

{datatailr-0.1.44 → datatailr-0.1.46}/src/datatailr/scheduler/batch_decorator.py

@@ -78,18 +78,13 @@ def batch_decorator(memory: str = DEFAULT_TASK_MEMORY, cpu: float = DEFAULT_TASK
                                 [None] * (len(all_function_args) - len(final_args))
                             )
                         final_args[all_function_args.index(name)] = value
-                job_name = os.getenv("DATATAILR_JOB_NAME", func.__name__)
-                run_id = os.getenv("DATATAILR_BATCH_RUN_ID")
                 try:
                     final_kwargs = __ARGUMENTS_CACHE__.get_arguments(
                         dag_id(func),
-                        job_name,
-                        run_id,
-                    )
-                except CacheNotFoundError as e:
-                    print(
-                        f"Cache not found for job {job_name} with run id {run_id}. Using provided arguments. Error: {e}"
+                        os.getenv("DATATAILR_JOB_NAME", func.__name__),
+                        os.getenv("DATATAILR_BATCH_RUN_ID"),
                     )
+                except CacheNotFoundError:
                     final_kwargs = kwargs
 
                 if varargs is not None and varkw is None:
@@ -101,7 +96,6 @@ def batch_decorator(memory: str = DEFAULT_TASK_MEMORY, cpu: float = DEFAULT_TASK
                 if len(final_args) == len(parameters.keys()):
                     for i, arg_name in enumerate(parameters.keys()):
                         final_args[i] = final_kwargs.pop(arg_name, final_args[i])
-                print(f"{final_args=}, {final_kwargs=}, {args=}, {kwargs=}")
                 result = func(*final_args, **final_kwargs)
                 __ARGUMENTS_CACHE__.add_result(
                     batch_run_id(),

{datatailr-0.1.44 → datatailr-0.1.46/src/datatailr.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: datatailr
-Version: 0.1.44
+Version: 0.1.46
 Summary: Ready-to-Use Platform That Drives Business Insights
 Author-email: Datatailr <info@datatailr.com>
 License-Expression: MIT

{datatailr-0.1.44 → datatailr-0.1.46}/src/sbin/datatailr_run.py

@@ -38,10 +38,8 @@ import sys
 from typing import Tuple
 from datatailr.logging import DatatailrLogger
 from datatailr.utils import is_dt_installed
-from datatailr import __version__
 
 logger = DatatailrLogger(os.path.abspath(__file__)).get_logger()
-logger.info(f"Datatailr version: {__version__}")
 
 if not is_dt_installed():
     logger.error("Datatailr is not installed.")
@@ -87,15 +85,15 @@ def run_command_as_user(command: str | list, user: str, env_vars: dict):
     """
     Run a command as a specific user with the given environment variables.
     """
-    if isinstance(command, str):
-        command = command.split(" ")
+    if isinstance(command, list):
+        command = " ".join(command)
     env_vars = {
         "PATH": get_env_var("PATH", ""),
         "PYTHONPATH": get_env_var("PYTHONPATH", ""),
     } | env_vars
 
     env_kv = [f"{k}={v}" for k, v in env_vars.items()]
-    argv = ["sudo", "-u", user, "env", *env_kv, "bash", "-lc", *command]
+    argv = ["sudo", "-u", user, "env", *env_kv, "bash", "-lc", command]
 
     try:
         result = subprocess.run(
@@ -149,9 +147,11 @@ def main():
     elif job_type == "excel":
         host = get_env_var("DATATAILR_HOST", "")
         entrypoint = get_env_var("DATATAILR_ENTRYPOINT")
+        local = get_env_var("DATATAILR_LOCAL", False)
         env = {
             "DATATAILR_ENTRYPOINT": entrypoint,
             "DATATAILR_HOST": host,
+            "DATATAILR_LOCAL": local,
         } | env
         run_command_as_user("datatailr_run_excel", user, env)
     elif job_type == "ide":

{datatailr-0.1.44 → datatailr-0.1.46}/src/sbin/datatailr_run_excel.py

@@ -20,15 +20,21 @@ logger = DatatailrLogger(os.path.abspath(__file__)).get_logger()
 
 def run():
     logger.info("Starting Datatailr excel add-in...")
-    entrypoint = os.environ.get("DATATAILR_ENTRYPOINT")
-    # hostname = os.environ.get("DATATAILR_HOST")
 
+    entrypoint = os.environ.get("DATATAILR_ENTRYPOINT")
     if entrypoint is None:
         raise ValueError("Environment variable 'DATATAILR_ENTRYPOINT' is not set.")
 
-    # if hostname is None:
-    #     raise ValueError("Environment variable 'DATATAILR_HOST' is not set.")
-
-    # entrypoint = f'./dt-excel.sh -n -H "{hostname}" -p 8080 "{entrypoint}"'
+    local = os.environ.get("DATATAILR_LOCAL", False) in ("1", "true", "True", "TRUE")
+    if local:
+        hostname = "localhost.excel.datatailr.com"
+        local_flag = "-l"
+    else:
+        hostname = os.environ.get("DATATAILR_HOST")
+        if hostname is None:
+            raise ValueError("Environment variable 'DATATAILR_HOST' is not set.")
+        local_flag = ""
+
+    entrypoint = f'/opt/datatailr/bin/dt-excel.sh -n -H "{hostname}" {local_flag} -p 8080 -w 8000 {entrypoint}'
     logger.info(f"Running entrypoint: {entrypoint}")
     subprocess.run(entrypoint, shell=True)