def get_compose(name):
try:
files = find_yml_files(settings.COMPOSE_DIR + name)
except Exception as exc:
raise HTTPException(exc.status_code, exc.detail)
for project, file in files.items():
if name == project:
networks = []
volumes = []
services = {}
compose = open(file)
loaded_compose = yaml.load(compose, Loader=yaml.SafeLoader)
if loaded_compose.get("volumes"):
for volume in loaded_compose.get("volumes"):
volumes.append(volume)
if loaded_compose.get("networks"):
for network in loaded_compose.get("networks"):
networks.append(network)
if loaded_compose.get("services"):
for service in loaded_compose.get("services"):
services[service] = loaded_compose["services"][service]
_content = open(file)
content = _content.read()
compose_object = {
"name": project,
"path": file,
"version": loaded_compose.get("version", "-"),
"services": services,
"volumes": volumes,
"networks": networks,
"content": content,
}
return compose_object
else:
raise HTTPException(404, "Project " + name + " not found")
def get_compose_projects():
files = find_yml_files(settings.COMPOSE_DIR)
projects = []
for project, file in files.items():
volumes = []
networks = []
services = {}
compose = open(file)
loaded_compose = yaml.load(compose, Loader=yaml.SafeLoader)
if loaded_compose:
if loaded_compose.get("volumes"):
for volume in loaded_compose.get("volumes"):
volumes.append(volume)
if loaded_compose.get("networks"):
for network in loaded_compose.get("networks"):
networks.append(network)
if loaded_compose.get("services"):
for service in loaded_compose.get("services"):
services[service] = loaded_compose["services"][service]
_project = {
"name": project,
"path": file,
"version": loaded_compose.get("version", "3.9"),
"services": services,
"volumes": volumes,
"networks": networks,
}
projects.append(_project)
else:
print("ERROR: " + file + " is invalid or empty!")
return projects
def generate_support_bundle(project_name):
files = find_yml_files(settings.COMPOSE_DIR + project_name)
if project_name in files:
dclient = docker.from_env()
stream = io.BytesIO()
with zipfile.ZipFile(stream, "w") as zf, open(files[project_name],
"r") as fp:
compose = yaml.load(fp, Loader=yaml.SafeLoader)
# print(compose)
# print(compose.get("services"))
for _service in compose.get("services"):
print()
if len(compose.get("services").keys()) < 2:
try:
if compose.get("services")[_service].get(
"container_name"):
service = dclient.containers.get(
compose.get("services")[_service].get(
"container_name"))
else:
service = dclient.containers.get(_service)
except docker.errors.NotFound as exc:
raise HTTPException(
exc.status_code,
detail="container " + _service + " not found",
)
else:
try:
if compose.get("services")[_service].get(
"container_name"):
service = dclient.containers.get(
compose.get("services")[_service].get(
"container_name"))
else:
service = dclient.containers.get(
project_name.lower() + "_" + _service + "_1")
except docker.errors.NotFound as exc:
raise HTTPException(
exc.status_code,
detail="container " + _service + " not found",
)
service_log = service.logs()
zf.writestr(f"{_service}.log", service_log)
fp.seek(0)
# It is possible that ".write(...)" has better memory management here.
zf.writestr("docker-compose.yml", fp.read())
stream.seek(0)
return StreamingResponse(
stream,
media_type="application/x-zip-compressed",
headers={
"Content-Disposition":
f"attachment;filename={project_name}_bundle.zip"
},
)
else:
raise HTTPException(404, f"Project {project_name} not found.")
def compose_action(name, action):
files = find_yml_files(settings.COMPOSE_DIR)
compose = get_compose(name)
env = os.environ.copy()
if action == "up":
try:
_action = docker_compose(
action,
"-d",
_cwd=os.path.dirname(compose["path"]),
_env=check_dockerhost(env),
)
except Exception as exc:
if hasattr(exc, "stderr"):
raise HTTPException(400, exc.stderr.decode("UTF-8").rstrip())
else:
raise HTTPException(400, exc)
elif action == "create":
try:
_action = docker_compose(
"up",
"--no-start",
_cwd=os.path.dirname(compose["path"]),
_env=check_dockerhost(env),
)
except Exception as exc:
if hasattr(exc, "stderr"):
raise HTTPException(400, exc.stderr.decode("UTF-8").rstrip())
else:
raise HTTPException(400, exc)
else:
try:
_action = docker_compose(
action,
_cwd=os.path.dirname(compose["path"]),
_env=check_dockerhost(env),
)
except Exception as exc:
if hasattr(exc, "stderr"):
raise HTTPException(400, exc.stderr.decode("UTF-8").rstrip())
else:
raise HTTPException(400, exc)
if _action.stdout.decode("UTF-8").rstrip():
_output = _action.stdout.decode("UTF-8").rstrip()
elif _action.stderr.decode("UTF-8").rstrip():
_output = _action.stderr.decode("UTF-8").rstrip()
else:
_output = "No Output"
print(f"""Project {compose['name']} {action} successful.""")
print(f"""Output: """)
print(_output)
return get_compose_projects()
