datatailr 0.1.25__py3-none-any.whl → 0.1.27__py3-none-any.whl

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/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}")
 

{datatailr-0.1.25.dist-info → datatailr-0.1.27.dist-info}/METADATA

@@ -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.dist-info → datatailr-0.1.27.dist-info}/RECORD

@@ -4,7 +4,7 @@ datatailr/blob.py,sha256=9lKZKm4eRKVE4_t2zy3hh6Z_Ov4iaTA3-tJFjnQKezg,3313
 datatailr/dt_json.py,sha256=3xmTqDBk68oPl2UW8UVOYPaBw4lAsVg6nDLwcen5nuo,2252
 datatailr/errors.py,sha256=p_e4ao3sFEfz1g4LvEDqw6bVzHJPJSINLjJ8H6_PqOo,751
 datatailr/excel.py,sha256=7_13VFHrjHlIAEeTaGQIlq4ycJvzRo-TvkP_KRBELGs,425
-datatailr/group.py,sha256=ExsrkqUooAfFLWKvnkp1ZxisSJD1yCp9TKqoCXDCwhs,4360
+datatailr/group.py,sha256=AC0nCA44eEWZCJCq2klPqkFg_995mS3C_wu5uSFFLtU,4426
 datatailr/logging.py,sha256=4Rsx3wf2tAr1334E2goBjhC877RwbUNaFgBlh902vU4,3270
 datatailr/user.py,sha256=tjIMbQ-tb2yFhgLBl_eTb47kKiy-RoM3KtYS3zmJiwc,6736
 datatailr/utils.py,sha256=mqnnERMyHNAuAgFY4Ry4O4yW0ZjCRtJbjfI5fXVqt2s,1524
@@ -12,7 +12,7 @@ datatailr/version.py,sha256=N9K8ZxlwFFSz8XSgbgaTWZY4k2J0JKfj698nZ_O2pIU,536
 datatailr/wrapper.py,sha256=K9ZD76cWey_ikA6C5sKejwRaYBDln4QMg-RcoRGiuFc,7991
 datatailr/build/__init__.py,sha256=_dA7b4L6wsaAFaSxUoYSJ1oaRqDHDMR20kqoCocSOss,487
 datatailr/build/image.py,sha256=YC8ML-l-sj6TcIBY-DCx_vaeI_7SmL9fPFhHnuxzRh0,5509
-datatailr/sbin/datatailr_run.py,sha256=uuhzINoBm2O59UcSe7H1iNkPwIlGh6nqor4EwNTKNwg,5952
+datatailr/sbin/datatailr_run.py,sha256=A8H2VigdKUxXQuxfy2Y09KOj5iY0nggZqlS8E9hThYQ,6386
 datatailr/sbin/datatailr_run_app.py,sha256=AOkutzv4DeKfWZs-ZBciAMKnK4A05SfkVf1ZJnSSFwA,1231
 datatailr/sbin/datatailr_run_batch.py,sha256=UWnp96j_G66R_Cape7Bb-rbK6UBLF7Y5_mTlWyGJAVQ,1818
 datatailr/sbin/datatailr_run_excel.py,sha256=Gr_QZgqJrwgRVD9_o4v-2tbvU-QMvNHL7xUvFGhftFc,1163
@@ -25,9 +25,9 @@ datatailr/scheduler/batch_decorator.py,sha256=LqL1bsupWLn-YEQUvFJYae7R3ogrL5-Vod
 datatailr/scheduler/constants.py,sha256=5WWTsfwZ_BA8gVDOTa2AQX9DJ0NzfaWgtY3vrODS2-8,606
 datatailr/scheduler/schedule.py,sha256=0XJJen2nL1xplRs0Xbjwgq3T-0bFCOrJzkSALdio998,3741
 datatailr/scheduler/utils.py,sha256=up6oR2iwe6G52LkvgfO394xchXgCYNjOMGRQW3e8PQk,1082
-datatailr-0.1.25.dist-info/licenses/LICENSE,sha256=ikKP4_O-UD_b8FuNdKmbzTb6odd0JX085ZW_FAPN3VI,1066
-datatailr-0.1.25.dist-info/METADATA,sha256=peUpI8-_NsE4zq2Fv63aUMtgwzqhRhpSfSnRLy-bU-s,5146
-datatailr-0.1.25.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-datatailr-0.1.25.dist-info/entry_points.txt,sha256=YqXfk2At-olW4PUSRkqvy_O3Mbv7uTKCCPuAAiz3Qbg,312
-datatailr-0.1.25.dist-info/top_level.txt,sha256=75gntW0X_SKpqxLL6hAPipvpk28GAhJBvoyqN_HohWU,10
-datatailr-0.1.25.dist-info/RECORD,,
+datatailr-0.1.27.dist-info/licenses/LICENSE,sha256=ikKP4_O-UD_b8FuNdKmbzTb6odd0JX085ZW_FAPN3VI,1066
+datatailr-0.1.27.dist-info/METADATA,sha256=0U7ze0UBwQ6n8KlnE97KKQTlTHjhPrGByhtUupcjtXE,5146
+datatailr-0.1.27.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+datatailr-0.1.27.dist-info/entry_points.txt,sha256=YqXfk2At-olW4PUSRkqvy_O3Mbv7uTKCCPuAAiz3Qbg,312
+datatailr-0.1.27.dist-info/top_level.txt,sha256=75gntW0X_SKpqxLL6hAPipvpk28GAhJBvoyqN_HohWU,10
+datatailr-0.1.27.dist-info/RECORD,,