mirror of
https://github.com/home-assistant/supervisor.git
synced 2025-06-23 18:36:29 +00:00
84 lines
2.5 KiB
Python
84 lines
2.5 KiB
Python
"""Init file for HassIO docker object."""
|
|
import logging
|
|
import os
|
|
|
|
import docker
|
|
|
|
from . import DockerBase
|
|
from ..const import RESTART_EXIT_CODE
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
|
|
class DockerSupervisor(DockerBase):
|
|
"""Docker hassio wrapper for HomeAssistant."""
|
|
|
|
def __init__(self, config, loop, dock, hassio, image=None):
|
|
"""Initialize docker base wrapper."""
|
|
super().__init__(config, loop, dock, image=image)
|
|
|
|
self.hassio = hassio
|
|
|
|
@property
|
|
def docker_name(self):
|
|
"""Return name of docker container."""
|
|
return os.environ['SUPERVISOR_NAME']
|
|
|
|
async def update(self, tag):
|
|
"""Update a supervisor docker image."""
|
|
if self._lock.locked():
|
|
_LOGGER.error("Can't excute update while a task is in progress")
|
|
return False
|
|
|
|
_LOGGER.info("Update supervisor docker to %s:%s", self.image, tag)
|
|
old_version = self.version
|
|
|
|
async with self._lock:
|
|
if await self.loop.run_in_executor(None, self._install, tag):
|
|
self.config.hassio_cleanup = old_version
|
|
self.loop.create_task(self.hassio.stop(RESTART_EXIT_CODE))
|
|
return True
|
|
|
|
return False
|
|
|
|
async def cleanup(self):
|
|
"""Check if old supervisor version exists and cleanup."""
|
|
if not self.config.hassio_cleanup:
|
|
return
|
|
|
|
async with self._lock:
|
|
if await self.loop.run_in_executor(None, self._cleanup):
|
|
self.config.hassio_cleanup = None
|
|
|
|
def _cleanup(self):
|
|
"""Remove old image.
|
|
|
|
Need run inside executor.
|
|
"""
|
|
old_image = "{}:{}".format(self.image, self.config.hassio_cleanup)
|
|
|
|
_LOGGER.info("Old supervisor docker found %s", old_image)
|
|
try:
|
|
self.dock.images.remove(image=old_image, force=True)
|
|
except docker.errors.DockerException as err:
|
|
_LOGGER.warning("Can't remove old image %s -> %s", old_image, err)
|
|
return False
|
|
|
|
return True
|
|
|
|
async def run(self):
|
|
"""Run docker image."""
|
|
raise RuntimeError("Not support on supervisor docker container!")
|
|
|
|
async def install(self, tag):
|
|
"""Pull docker image."""
|
|
raise RuntimeError("Not support on supervisor docker container!")
|
|
|
|
async def stop(self):
|
|
"""Stop/remove docker container."""
|
|
raise RuntimeError("Not support on supervisor docker container!")
|
|
|
|
async def remove(self):
|
|
"""Remove docker image."""
|
|
raise RuntimeError("Not support on supervisor docker container!")
|