mirror of
https://github.com/katanemo/plano.git
synced 2026-06-17 15:25:17 +02:00
remove unused code
This commit is contained in:
parent
3abd1a7261
commit
8987d099e0
2 changed files with 0 additions and 101 deletions
|
|
@ -1,12 +1,10 @@
|
|||
import click
|
||||
import os
|
||||
import pkg_resources
|
||||
import sys
|
||||
import subprocess
|
||||
import multiprocessing
|
||||
import importlib.metadata
|
||||
from cli import targets
|
||||
from cli import config_generator
|
||||
from cli.utils import (
|
||||
getLogger,
|
||||
get_llm_provider_access_keys,
|
||||
|
|
|
|||
|
|
@ -1,10 +1,4 @@
|
|||
import subprocess
|
||||
import os
|
||||
import time
|
||||
import select
|
||||
import shlex
|
||||
import yaml
|
||||
import json
|
||||
import logging
|
||||
import docker
|
||||
|
||||
|
|
@ -60,74 +54,6 @@ def validate_schema(arch_config_file: str) -> None:
|
|||
raise ValueError(f"Failed to create container: {e}")
|
||||
|
||||
|
||||
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.
|
||||
# This should be a non-blocking call so using subprocess.Popen(...)
|
||||
ps_process = subprocess.Popen(
|
||||
[
|
||||
"docker",
|
||||
"compose",
|
||||
"-p",
|
||||
"arch",
|
||||
"ps",
|
||||
"--format",
|
||||
"table{{.Service}}\t{{.State}}\t{{.Ports}}",
|
||||
],
|
||||
cwd=os.path.dirname(compose_file),
|
||||
stdout=subprocess.PIPE,
|
||||
stderr=subprocess.PIPE,
|
||||
text=True,
|
||||
start_new_session=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:
|
||||
log.info(
|
||||
f"Error while checking service status:\n{error_output}",
|
||||
file=os.sys.stderr,
|
||||
)
|
||||
return {}
|
||||
|
||||
services = parse_docker_compose_ps_output(services_status)
|
||||
return services
|
||||
|
||||
except subprocess.CalledProcessError as e:
|
||||
log.info(f"Failed to check service status. Error:\n{e.stderr}")
|
||||
return e
|
||||
|
||||
|
||||
# Helper method to print service status
|
||||
def print_service_status(services):
|
||||
log.info(f"{'Service Name':<25} {'State':<20} {'Ports'}")
|
||||
log.info("=" * 72)
|
||||
for service_name, info in services.items():
|
||||
status = info["STATE"]
|
||||
ports = info["PORTS"]
|
||||
log.info(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[
|
||||
"STATE"
|
||||
].lower() # Convert status to lowercase for easier comparison
|
||||
if any(state in status for state in states):
|
||||
return True
|
||||
|
||||
return False
|
||||
|
||||
|
||||
def get_llm_provider_access_keys(arch_config_file):
|
||||
with open(arch_config_file, "r") as file:
|
||||
arch_config = file.read()
|
||||
|
|
@ -165,28 +91,3 @@ def load_env_file_to_dict(file_path):
|
|||
env_dict[key] = value
|
||||
|
||||
return env_dict
|
||||
|
||||
|
||||
def parse_docker_compose_ps_output(output):
|
||||
# Split the output into lines
|
||||
lines = output.strip().splitlines()
|
||||
|
||||
# Extract the headers (first row) and the rest of the data
|
||||
headers = lines[0].split()
|
||||
service_data = lines[1:]
|
||||
|
||||
# Initialize the result dictionary
|
||||
services = {}
|
||||
|
||||
# Iterate over each line of data after the headers
|
||||
for line in service_data:
|
||||
# Split the line by tabs or multiple spaces
|
||||
parts = line.split()
|
||||
|
||||
# Create a dictionary entry using the header names
|
||||
service_info = {headers[1]: parts[1], headers[2]: parts[2]} # State # Ports
|
||||
|
||||
# Add to the result dictionary using the service name as the key
|
||||
services[parts[0]] = service_info
|
||||
|
||||
return services
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue