datatailr 0.1.25__tar.gz → 0.1.27__tar.gz

{datatailr-0.1.25/src/datatailr.egg-info → datatailr-0.1.27}/PKG-INFO

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

{datatailr-0.1.25 → datatailr-0.1.27}/pyproject.toml

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

{datatailr-0.1.25 → datatailr-0.1.27}/src/datatailr/group.py

@@ -103,8 +103,8 @@ class Group:
 
     @staticmethod
     def add(name: str) -> Optional["Group"]:
-        __client__.add(name)
-        return Group.get(name)
+        new_group = __client__.add(name, json_enrichened=True)
+        return Group(new_group["name"]) if new_group else None
 
     @staticmethod
     def ls() -> list:

{datatailr-0.1.25 → datatailr-0.1.27/src/datatailr.egg-info}/PKG-INFO

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

{datatailr-0.1.25 → datatailr-0.1.27}/src/sbin/datatailr_run.py

@@ -32,6 +32,7 @@
 #     DATATAILR_JOB_ID - the unique identifier for the job.
 
 
+import subprocess
 import os
 import sys
 from typing import Tuple
@@ -80,18 +81,29 @@ def create_user_and_group() -> Tuple[str, str]:
     return user, group
 
 
-def run_command_as_user(command: str, user: str, env_vars: dict):
+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(" ")
     env_vars.update({"PATH": get_env_var("PATH")})
     env_vars.update({"PYTHONPATH": get_env_var("PYTHONPATH", "")})
-    env_vars_str = " ".join(f"{key}='{value}'" for key, value in env_vars.items())
-    full_command = f"sudo -u {user} {env_vars_str} {command}"
-    logger.debug(f"Running command: {full_command}")
-    status = os.system(full_command)
-    if status != 0:
-        logger.error(f"Command failed with status {status / 256:g}")
+    full_command = ["sudo", "-u", user, "bash", "-c", *command]
+
+    try:
+        result = subprocess.run(
+            full_command,
+            env=env_vars,
+            check=True,  # raises if returncode != 0
+            capture_output=True,
+            text=True,
+        )
+        logger.info(f"stdout: {result.stdout}")
+        logger.debug(f"stderr: {result.stderr}")
+    except subprocess.CalledProcessError as e:
+        logger.error(f"Command failed with exit code {e.returncode}")
+        logger.error(f"stderr: {e.stderr}")
         sys.exit(1)
 
 
@@ -138,7 +150,9 @@ def main():
         } | env
         run_command_as_user("datatailr_run_excel", user, env)
     elif job_type == "IDE":
-        pass
+        run_command_as_user("code server", user, env)
+    elif job_type == "jupyter":
+        run_command_as_user("jupyter lab", user, env)
     else:
         raise ValueError(f"Unknown job type: {job_type}")