Function run_command

cli/casp/src/casp/utils/docker_utils.py:111–178

Runs a command in a docker container and streams logs. Args: command: The command to run. volumes: A dictionary of volumes to mount. image: The docker image to use. privileged: Whether to run the container as privileged. Returns: True on success, False otherwise.

(
    command: list[str],
    volumes: dict,
    image: str,
    privileged: bool = False,
)

def run_command(
    command: list[str],
    volumes: dict,
    image: str,
    privileged: bool = False,
) -> bool:
  """Runs a command in a docker container and streams logs.

  Args:
    command: The command to run.
    volumes: A dictionary of volumes to mount.
    image: The docker image to use.
    privileged: Whether to run the container as privileged.

  Returns:
    True on success, False otherwise.
  """
  client = check_docker_setup()
  if not client:
    return False

  if not pull_image(image):
    return False

  container_instance = None
  try:
    click.echo(f'Running command in Docker container: {command}')
    container_instance = client.containers.run(
        image,
        command,
        volumes=volumes,
        working_dir=_DEFAULT_WORKING_DIR,
        privileged=privileged,
        detach=True,
        remove=False)  # Can't auto-remove if we want to stream logs

    for line in container_instance.logs(stream=True, follow=True):
      click.echo(line.decode('utf-8').strip())

    result = container_instance.wait()
    if result['StatusCode'] != 0:
      # Get final logs in case of error
      error_logs = container_instance.logs().decode('utf-8')
      click.secho(
          'Error: Command failed in Docker container with exit code '
          f'{result["StatusCode"]}.',
          fg='red')
      click.secho(error_logs, fg='red')
      return False

    return True
  except docker.errors.ContainerError as e:
    click.secho(f'Error: Command failed in Docker container: {e}', fg='red')
    if e.stderr:
      click.secho(e.stderr.decode('utf-8'), fg='red')
    return False
  except docker.errors.ImageNotFound as e:
    click.secho(f'Error: Docker image {image} not found: {e}', fg='red')