artefacts-cli 0.6.16__py3-none-any.whl → 0.6.17__py3-none-any.whl

artefacts/cli/__init__.py

@@ -15,7 +15,7 @@ logging.basicConfig(level=logging.INFO)
 
 
 try:
-    __version__ = version("package-name")
+    __version__ = version("artefacts-cli")
 except PackageNotFoundError:
     try:
         # Package is not installed, most likely dev/test mode

artefacts/cli/app.py

@@ -231,11 +231,64 @@ def hello(project_name):
     is_eager=True,  # Necessary for callbacks to see it.
     help="Skip configuration validation, so that unsupported settings can be tried out, e.g. non-ROS settings or simulators like SAPIEN.",
 )
+@click.option(
+    "--in-container",
+    is_flag=True,
+    default=False,
+    help='[Experimental] Run the job inside a package container. The container image is build if it does not exist yet, with default name as "artefacts" (please use --with-image to override the image name). This option overrides (for now) --dryrun, --nosim, --noisolation and --description.',
+)
+@click.option(
+    "--with-image",
+    default="artefacts",
+    help="[Experimental] Run the job using the image name passed here. Only used when running with --in-container set.",
+)
+@click.option(
+    "--rebuild-container",
+    is_flag=True,
+    default=False,
+    help="[Experimental] Rebuild the container image before running. This flag guarantees that the run uses the latest code available when building the image, and usually takes more time before the run can start (time to compile and generate the image).",
+)
+@click.option(
+    "--with-gui",
+    is_flag=True,
+    default=False,
+    help="Show any GUI if any is created by the test runs. By default, UI elements are run but hidden---only test logs are returned. Please note GUI often assume X11 (e.g. ROS), typically with Qt, so this may not work without a appropriate environment.",
+)
 @click.argument("jobname")
+@click.pass_context
 def run(
-    config, jobname, dryrun, nosim, noisolation, description="", skip_validation=False
+    ctx: click.Context,
+    config,
+    jobname,
+    dryrun,
+    nosim,
+    noisolation,
+    description="",
+    skip_validation=False,
+    in_container: bool = False,
+    with_image: str = "artefacts",
+    rebuild_container: bool = False,
+    with_gui: bool = False,
 ):
-    """Run JOBNAME locally"""
+    """
+    Run JOBNAME locally
+
+    * Directly in the shell by default.
+    * Inside a packaged container when using the --in-container option.
+    """
+    if in_container:
+        if rebuild_container or not ctx.invoke(containers.check, name=with_image):
+            ctx.invoke(
+                containers.build, path=".", dockerfile="Dockerfile", name=with_image
+            )
+        return ctx.invoke(
+            containers.run,
+            image=with_image,
+            jobname=jobname,
+            config=config,
+            with_gui=with_gui,
+        )
+
     warpconfig = read_config(config)
 
     project_id = warpconfig["project"]
@@ -573,7 +626,7 @@ artefacts.add_command(config)
 artefacts.add_command(hello)
 artefacts.add_command(run)
 artefacts.add_command(run_remote)
-artefacts.add_command(containers.package)
+artefacts.add_command(containers.containers)
 
 
 if __name__ == "__main__":

artefacts/cli/app_containers.py

@@ -10,12 +10,12 @@ from artefacts.cli.containers.utils import ContainerMgr
 @click.group()
 @click.option("--debug/--no-debug", default=False)
 @click.pass_context
-def package(ctx: click.Context, debug: bool):
+def containers(ctx: click.Context, debug: bool):
     ctx.ensure_object(dict)
     ctx.obj["debug"] = debug
 
 
-@package.command()
+@containers.command()
 @click.option(
     "--path",
     default=".",
@@ -29,13 +29,13 @@ def package(ctx: click.Context, debug: bool):
 @click.option(
     "--name",
     required=False,
-    help="Name for the generated package",
+    help="Name for the generated container",
 )
 @click.pass_context
 def build(ctx: click.Context, path: str, dockerfile: str, name: str):
     if not os.path.exists(os.path.join(path, dockerfile)):
         raise click.ClickException(
-            f"No {dockerfile} found here. I cannot build the package."
+            f"No {dockerfile} found here. I cannot build the container."
         )
     if name is None:
         name = "artefacts"
@@ -44,7 +44,21 @@ def build(ctx: click.Context, path: str, dockerfile: str, name: str):
     print(f"Package complete in image: {image}")
 
 
-@package.command()
+@containers.command()
+@click.argument("name")
+@click.pass_context
+def check(ctx: click.Context, name: str):
+    if name is None:
+        name = "artefacts"
+    handler = ContainerMgr()
+    result = handler.check(name)
+    if ctx.parent is None:
+        # Print only if the command is called directly.
+        print(f"Package {name} exists and ready to use.")
+    return result
+
+
+@containers.command()
 @click.argument("image")
 @click.argument("jobname")
 @click.option(
@@ -53,8 +67,14 @@ def build(ctx: click.Context, path: str, dockerfile: str, name: str):
     default="artefacts.yaml",
     help="Artefacts config file.",
 )
+@click.option(
+    "--with-gui",
+    "with_gui",
+    default=False,
+    help="Show any GUI if any is created by the test runs. By default, UI elements are run but hidden---only test logs are returned. Please note GUI often assume an X11 environment, typically with Qt, so this may not work without a appropriate environment.",
+)
 @click.pass_context
-def run(ctx: click.Context, image: str, jobname: str, config: str):
+def run(ctx: click.Context, image: str, jobname: str, config: str, with_gui: bool):
     try:
         artefacts_config = read_config(config)
     except FileNotFoundError:
@@ -67,6 +87,7 @@ def run(ctx: click.Context, image: str, jobname: str, config: str):
         jobname=jobname,
         # Hidden setting primarily useful to Artefacts developers
         api_url=os.environ.get("ARTEFACTS_API_URL", DEFAULT_API_URL),
+        with_gui=with_gui,
     )
     container, logs = handler.run(**params)
     if container:

artefacts/cli/containers/__init__.py

@@ -15,6 +15,12 @@ class CMgr:
         """
         raise NotImplemented()
 
+    def check(self, image: str) -> bool:
+        """
+        Checks whether a target image exists locally.
+        """
+        raise NotImplemented()
+
     def run(
         self,
         image: str,
@@ -22,6 +28,7 @@ class CMgr:
         jobname: str = None,
         artefacts_dir: str = Path("~/.artefacts").expanduser(),
         api_url: str = DEFAULT_API_URL,
+        with_gui: bool = False,
     ) -> Tuple[Any, Generator]:
         """
         Returns a container (Any type as depends on the framework)
@@ -45,7 +52,7 @@ class CMgr:
         """
         if not path:
             raise Exception(
-                "`path` must be a string, a Path object, or exclusded from the kwargs"
+                "`path` must be a string, a Path object, or excluded from the kwargs"
             )
         if os.environ.get("ARTEFACTS_KEY", None):
             return True

artefacts/cli/containers/docker.py

@@ -2,6 +2,7 @@ from collections.abc import Generator
 import json
 import os
 from pathlib import Path
+import platform
 from typing import Any, Tuple
 from uuid import uuid4
 
@@ -39,6 +40,12 @@ class DockerManager(CMgr):
             img_id = self.client.inspect_image(kwargs["tag"])["Id"]
         return img_id, iter(logs)
 
+    def check(
+        self,
+        image: str,
+    ) -> bool:
+        return len(self.client.images(name=image)) > 0
+
     def run(
         self,
         image: str,
@@ -46,6 +53,7 @@ class DockerManager(CMgr):
         jobname: str = None,
         artefacts_dir: str = Path("~/.artefacts").expanduser(),
         api_url: str = DEFAULT_API_URL,
+        with_gui: bool = False,
     ) -> Tuple[Any, Generator]:
         if not self._valid_artefacts_api_key(project, artefacts_dir):
             return None, iter(
@@ -60,6 +68,15 @@ class DockerManager(CMgr):
                 "ARTEFACTS_API_URL": api_url,
             }
 
+            if platform.system() in ["Darwin", "Windows"]:
+                # Assume we run in Docker Desktop
+                env["DISPLAY"] = "host.docker.internal:0"
+            else:
+                env["DISPLAY"] = os.environ.get("DISPLAY", ":0")
+
+            if not with_gui:
+                env["QT_QPA_PLATFORM"] = "offscreen"
+
             container = self.client.create_container(
                 image,
                 environment=env,
@@ -72,6 +89,7 @@ class DockerManager(CMgr):
                             "mode": "ro",
                         },
                     },
+                    network_mode="host",
                 ),
             )
             self.client.start(container=container.get("Id"))

artefacts/cli/containers/utils.py

@@ -50,5 +50,8 @@ class ContainerMgr:
     def build(self, **kwargs) -> Tuple[str, Generator]:
         return self.mgr.build(**kwargs)
 
+    def check(self, image: str) -> bool:
+        return self.mgr.check(image)
+
     def run(self, **kwargs) -> Tuple[Any, Generator]:
         return self.mgr.run(**kwargs)

artefacts/cli/version.py

@@ -12,5 +12,5 @@ __version__: str
 __version_tuple__: VERSION_TUPLE
 version_tuple: VERSION_TUPLE
 
-__version__ = version = '0.6.16'
-__version_tuple__ = version_tuple = (0, 6, 16)
+__version__ = version = '0.6.17'
+__version_tuple__ = version_tuple = (0, 6, 17)

{artefacts_cli-0.6.16.dist-info → artefacts_cli-0.6.17.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: artefacts_cli
-Version: 0.6.16
+Version: 0.6.17
 Author-email: FD <fabian@artefacts.com>, AGC <alejandro@artefacts.com>, TN <tomo@artefacts.com>, EP <eric@artefacts.com>
 Project-URL: Homepage, https://github.com/art-e-fact/artefacts-client
 Project-URL: Bug Tracker, https://github.com/art-e-fact/artefacts-client/issues
@@ -18,6 +18,7 @@ Requires-Dist: mcap-ros2-support
 Requires-Dist: PyYAML>=6.0
 Requires-Dist: requests>=2.27.1
 Requires-Dist: setuptools-scm
+Requires-Dist: setuptools>=74
 Provides-Extra: dev
 Requires-Dist: awscli; extra == "dev"
 Requires-Dist: build; extra == "dev"

artefacts_cli-0.6.17.dist-info/RECORD

@@ -0,0 +1,21 @@
+artefacts/cli/__init__.py,sha256=--1soc8cqq1JWhz4DuRUvJWgSeDCPGqm8mEmdqtclsI,11644
+artefacts/cli/app.py,sha256=cxVVHD89_eVCHm5fqPq3ANuy6jTPfw3se-FST_7wILY,22093
+artefacts/cli/app_containers.py,sha256=tHdGkxtzjexLhgcx5YXy_OOJsPcBi1jmwKZN01OAcmU,3044
+artefacts/cli/bagparser.py,sha256=FE_QaztC9pg4hQzTjGSdyve6mzZbHJbyqa3wqvZSbxE,3702
+artefacts/cli/constants.py,sha256=bvsVDwqkAc49IZN7j6k6IL6EG87bECHd_VINtKJqbv8,320
+artefacts/cli/other.py,sha256=7NvzlspvG0zF7sryR-QznwdLupXLln1BKWxHB9VuEcc,1160
+artefacts/cli/parameters.py,sha256=MDhrM7ur95wKTLDteqz2f-sLdCPcepi5wk0XjeLo6TU,788
+artefacts/cli/ros1.py,sha256=RbtirCGarD9a0ikfuGK-pdpWYSXfqJhEt4rpA0uFsyU,9625
+artefacts/cli/ros2.py,sha256=YaCi3HRuCdvCTjM0Ftnm-SLgMOy1OneIr0aU7KVTiAM,4476
+artefacts/cli/utils.py,sha256=6yQJRzv-xaLbtcA73Tty9C9QZeV06n5pjXMYl3xsLPE,3156
+artefacts/cli/utils_ros.py,sha256=pYbhAU9fK2VbnWm3nSLBwUMVBzsZuCn10SFZmVW3-Zo,2090
+artefacts/cli/version.py,sha256=q7Xb4Wt4eH2VCIhiAa6149XDeUKQFfphqlwqiTFiurk,413
+artefacts/cli/containers/__init__.py,sha256=y6NVB0efBfRQBlHb71xhRu1hEh7t2bkKNs2vtsLJaO8,2239
+artefacts/cli/containers/docker.py,sha256=A9jqs0WU-g8umJzPpEACi0_DU5N8rglR-V3dmwxDuBA,3507
+artefacts/cli/containers/utils.py,sha256=6Bw6s1ceEnTbWSdE5_1CReNW6RB29VeHsq-BAwcJ_fY,1895
+artefacts/wrappers/artefacts_ros1_meta.launch,sha256=9tN7_0xLH8jW27KYFerhF3NuWDx2dED3ks_qoGVZAPw,1412
+artefacts_cli-0.6.17.dist-info/METADATA,sha256=aiDZPGqyOGAS_pS3XnjIk4IAStv_vS3fUOW3qPyEH-s,3051
+artefacts_cli-0.6.17.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
+artefacts_cli-0.6.17.dist-info/entry_points.txt,sha256=nlTXRzilNjccbi53FgaRWCQPkG-pv61HRkaCkrKjlec,58
+artefacts_cli-0.6.17.dist-info/top_level.txt,sha256=FdaMV1C9m36MWa-2Stm5xVODv7hss_nRYNwR83j_7ow,10
+artefacts_cli-0.6.17.dist-info/RECORD,,

artefacts_cli-0.6.16.dist-info/RECORD

@@ -1,21 +0,0 @@
-artefacts/cli/__init__.py,sha256=9rTAfBoS9ALJxKSqaUo32ThbDshbvNN4-EmoyUsPhRI,11643
-artefacts/cli/app.py,sha256=597uPtK4A_wujvqhk5h4KkrXV_XVkDLpymedd2lZB5k,20140
-artefacts/cli/app_containers.py,sha256=7OAtc4n9YeK8ZYUKvteOMbZr7lYjx_YMa6tu8Pnu9Yc,2280
-artefacts/cli/bagparser.py,sha256=FE_QaztC9pg4hQzTjGSdyve6mzZbHJbyqa3wqvZSbxE,3702
-artefacts/cli/constants.py,sha256=bvsVDwqkAc49IZN7j6k6IL6EG87bECHd_VINtKJqbv8,320
-artefacts/cli/other.py,sha256=7NvzlspvG0zF7sryR-QznwdLupXLln1BKWxHB9VuEcc,1160
-artefacts/cli/parameters.py,sha256=MDhrM7ur95wKTLDteqz2f-sLdCPcepi5wk0XjeLo6TU,788
-artefacts/cli/ros1.py,sha256=RbtirCGarD9a0ikfuGK-pdpWYSXfqJhEt4rpA0uFsyU,9625
-artefacts/cli/ros2.py,sha256=YaCi3HRuCdvCTjM0Ftnm-SLgMOy1OneIr0aU7KVTiAM,4476
-artefacts/cli/utils.py,sha256=6yQJRzv-xaLbtcA73Tty9C9QZeV06n5pjXMYl3xsLPE,3156
-artefacts/cli/utils_ros.py,sha256=pYbhAU9fK2VbnWm3nSLBwUMVBzsZuCn10SFZmVW3-Zo,2090
-artefacts/cli/version.py,sha256=zbqzBVUYJbK8o4vMJuNyIN5tBM9QXVZa7XiEM2PNdR8,413
-artefacts/cli/containers/__init__.py,sha256=S3DXLsb0csUdfjOZIngTfhMHrmYAaWBeu5EctLEEXbM,2057
-artefacts/cli/containers/docker.py,sha256=RtUOMbtWHQumTtDRa0ALjJ29tSaa7Pp497Ah1aJ7nZ4,2964
-artefacts/cli/containers/utils.py,sha256=5yUoREHM9AxnTkDNCY-DFUPSR5OWtu1rNxIVs6n5zZo,1816
-artefacts/wrappers/artefacts_ros1_meta.launch,sha256=9tN7_0xLH8jW27KYFerhF3NuWDx2dED3ks_qoGVZAPw,1412
-artefacts_cli-0.6.16.dist-info/METADATA,sha256=OLy7Jmo_RMo9CiktBZSKiZRN1HVILJkuhz-W24YB0RA,3021
-artefacts_cli-0.6.16.dist-info/WHEEL,sha256=PZUExdf71Ui_so67QXpySuHtCi3-J3wvF4ORK6k_S8U,91
-artefacts_cli-0.6.16.dist-info/entry_points.txt,sha256=nlTXRzilNjccbi53FgaRWCQPkG-pv61HRkaCkrKjlec,58
-artefacts_cli-0.6.16.dist-info/top_level.txt,sha256=FdaMV1C9m36MWa-2Stm5xVODv7hss_nRYNwR83j_7ow,10
-artefacts_cli-0.6.16.dist-info/RECORD,,