mirror of
https://github.com/katanemo/plano.git
synced 2026-05-01 11:56:29 +02:00
dc57f119a0
* initial commit of the insurange agent demo, with the CLI tool * committing the cli * fixed some field descriptions for generate-prompt-targets * CLI works with buil, up and down commands. Function calling example works stand-alone * fixed README to install archgw cli * fixing based on feedback * fixing based on feedback --------- Co-authored-by: Salman Paracha <salmanparacha@MacBook-Pro-261.local>
79 lines
2.7 KiB
Python
79 lines
2.7 KiB
Python
import subprocess
|
|
import os
|
|
import time
|
|
import select
|
|
import shlex
|
|
|
|
def run_docker_compose_ps(compose_file, env):
|
|
"""
|
|
Check if all Docker Compose services are in a healthy state.
|
|
|
|
Args:
|
|
path (str): The path where the docker-compose.yml file is located.
|
|
"""
|
|
try:
|
|
# Run `docker-compose ps` to get the health status of each service
|
|
ps_process = subprocess.Popen(
|
|
["docker-compose", "ps"],
|
|
cwd=os.path.dirname(compose_file),
|
|
stdout=subprocess.PIPE,
|
|
stderr=subprocess.PIPE,
|
|
text=True,
|
|
env=env
|
|
)
|
|
# Capture the output of `docker-compose ps`
|
|
services_status, error_output = ps_process.communicate()
|
|
|
|
# Check if there is any error output
|
|
if error_output:
|
|
print(f"Error while checking service status:\n{error_output}", file=os.sys.stderr)
|
|
return {}
|
|
|
|
lines = services_status.strip().splitlines()
|
|
services = {}
|
|
|
|
# Skip the header row and parse each service
|
|
for line in lines[1:]:
|
|
parts = shlex.split(line)
|
|
if len(parts) >= 5:
|
|
service_name = parts[0] # Service name
|
|
status_index = 3 # Status is typically at index 3, but may have multiple words
|
|
|
|
# Check if the status has multiple words (e.g., "running (healthy)")
|
|
if '(' in parts[status_index+1] :
|
|
# Combine the status field if it's split over two parts
|
|
status = f"{parts[status_index]} {parts[status_index + 1]}"
|
|
ports = parts[status_index + 2]
|
|
else:
|
|
status = parts[status_index]
|
|
ports = parts[status_index + 1]
|
|
|
|
# Store both status and ports in a dictionary for each service
|
|
services[service_name] = {
|
|
'status': status,
|
|
'ports': ports
|
|
}
|
|
|
|
return services
|
|
|
|
except subprocess.CalledProcessError as e:
|
|
print(f"Failed to check service status. Error:\n{e.stderr}")
|
|
return e
|
|
|
|
#Helper method to print service status
|
|
def print_service_status(services):
|
|
print(f"{'Service Name':<25} {'Status':<20} {'Ports'}")
|
|
print("="*72)
|
|
for service_name, info in services.items():
|
|
status = info['status']
|
|
ports = info['ports']
|
|
print(f"{service_name:<25} {status:<20} {ports}")
|
|
|
|
#check for states based on the states passed in
|
|
def check_services_state(services, states):
|
|
for service_name, service_info in services.items():
|
|
status = service_info['status'].lower() # Convert status to lowercase for easier comparison
|
|
if any(state in status for state in states):
|
|
return True
|
|
|
|
return False
|