mirror of
https://github.com/home-assistant/supervisor.git
synced 2025-12-23 11:58:49 +00:00
116 lines
3.3 KiB
Python
116 lines
3.3 KiB
Python
"""Init file for HassIO docker object."""
|
|
from contextlib import suppress
|
|
import logging
|
|
|
|
import attr
|
|
import docker
|
|
|
|
from .network import DockerNetwork
|
|
from ..const import SOCKET_DOCKER
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
# pylint: disable=invalid-name
|
|
CommandReturn = attr.make_class('CommandReturn', ['exit_code', 'output'])
|
|
|
|
|
|
class DockerAPI:
|
|
"""Docker hassio wrapper.
|
|
|
|
This class is not AsyncIO safe!
|
|
"""
|
|
|
|
def __init__(self):
|
|
"""Initialize docker base wrapper."""
|
|
self.docker = docker.DockerClient(
|
|
base_url="unix:/{}".format(str(SOCKET_DOCKER)),
|
|
version='auto', timeout=900)
|
|
self.network = DockerNetwork(self.docker)
|
|
|
|
@property
|
|
def images(self):
|
|
"""Return api images."""
|
|
return self.docker.images
|
|
|
|
@property
|
|
def containers(self):
|
|
"""Return api containers."""
|
|
return self.docker.containers
|
|
|
|
@property
|
|
def api(self):
|
|
"""Return api containers."""
|
|
return self.docker.api
|
|
|
|
def run(self, image, **kwargs):
|
|
""""Create a docker and run it.
|
|
|
|
Need run inside executor.
|
|
"""
|
|
name = kwargs.get('name', image)
|
|
network_mode = kwargs.get('network_mode')
|
|
hostname = kwargs.get('hostname')
|
|
|
|
# setup network
|
|
kwargs['dns_search'] = ["."]
|
|
if network_mode:
|
|
kwargs['dns'] = [str(self.network.supervisor)]
|
|
kwargs['dns_opt'] = ["ndots:0"]
|
|
else:
|
|
kwargs['network'] = None
|
|
|
|
# create container
|
|
try:
|
|
container = self.docker.containers.create(image, **kwargs)
|
|
except docker.errors.DockerException as err:
|
|
_LOGGER.error("Can't create container from %s: %s", name, err)
|
|
return False
|
|
|
|
# attach network
|
|
if not network_mode:
|
|
alias = [hostname] if hostname else None
|
|
if self.network.attach_container(container, alias=alias):
|
|
self.network.detach_default_bridge(container)
|
|
else:
|
|
_LOGGER.warning("Can't attach %s to hassio-net!", name)
|
|
|
|
# run container
|
|
try:
|
|
container.start()
|
|
except docker.errors.DockerException as err:
|
|
_LOGGER.error("Can't start %s: %s", name, err)
|
|
return False
|
|
|
|
return True
|
|
|
|
def run_command(self, image, command=None, **kwargs):
|
|
"""Create a temporary container and run command.
|
|
|
|
Need run inside executor.
|
|
"""
|
|
stdout = kwargs.get('stdout', True)
|
|
stderr = kwargs.get('stderr', True)
|
|
|
|
_LOGGER.info("Run command '%s' on %s", command, image)
|
|
try:
|
|
container = self.docker.containers.run(
|
|
image,
|
|
command=command,
|
|
network=self.network.name,
|
|
**kwargs
|
|
)
|
|
|
|
# wait until command is done
|
|
result = container.wait()
|
|
output = container.logs(stdout=stdout, stderr=stderr)
|
|
|
|
except docker.errors.DockerException as err:
|
|
_LOGGER.error("Can't execute command: %s", err)
|
|
return CommandReturn(None, b"")
|
|
|
|
# cleanup container
|
|
with suppress(docker.errors.DockerException):
|
|
container.remove(force=True)
|
|
|
|
return CommandReturn(result.get('StatusCode'), output)
|