40 lines
1.2 KiB
Python
40 lines
1.2 KiB
Python
import logging
|
|
from pathlib import Path
|
|
|
|
import docker
|
|
from docker.models.containers import Container
|
|
from injectable import injectable
|
|
|
|
from app.model.containers import Tree, Compose, SimpleContainer
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
@injectable(singleton=True)
|
|
class DockerService:
|
|
def __init__(self):
|
|
self._client = docker.from_env()
|
|
# logger.info(f"Docker client initialized. Plugins: {self._client.plugins()}")
|
|
self._tree = self._init_tree()
|
|
|
|
def _init_tree(self) -> Tree:
|
|
tree = Tree()
|
|
container: Container
|
|
for container in self._client.containers.list():
|
|
labels = container.labels
|
|
working_dir = labels.get("com.docker.compose.project.working_dir")
|
|
if working_dir:
|
|
if tree.composes.get(working_dir) is None:
|
|
tree.composes[working_dir] = Compose(working_dir)
|
|
tree.composes[working_dir].containers.append(SimpleContainer.from_container(container))
|
|
else:
|
|
tree.containers.append(SimpleContainer.from_container(container))
|
|
return tree
|
|
|
|
@property
|
|
def tree(self) -> Tree:
|
|
return self._tree
|
|
|
|
def reload(self, compose_path: Path):
|
|
cmd = ["sudo", "docker", "compose", "-f", str(compose_path), "up", "-d"]
|
|
# TODO: subprocess
|