mirror of
https://github.com/home-assistant/supervisor.git
synced 2025-10-04 01:09:31 +00:00
1b887e38d6
* Add new data Layer for addons * draft v2 * part 3 * Cleanups for new addon layout * cleanup part 5 * fix lint p1 * fix lint p2 * fix api bug * Fix lint p3 * fix lint p4 * fix lint p5 * fix lint p6 * fix update * fix lint * Update repo add code part 1 * new repository load code * reorder code * Code cleanup p1 * Don't allow change options for not installed addons * Cleanup error handling * Fix addon restart config bug * minimize timeout for bad addons * set core timeout for docker to 15 * fix lint * fix start bug * Add startuptype * change names / fix update bug * fix update bug p2 * Cleanup arch * Ignore built-in repositories * fix lint * fix bug * fix bug p4 * fix arch handling / better bugfix for boot option * fix lint * fix arch * fix options * fix close is no coro * update api description & fix lint * Fix error * fix api validate config * cleanup api * Fix repo loader * cleanup slugs * fix lint
108 lines
3.2 KiB
Python
108 lines
3.2 KiB
Python
"""Init file for HassIO addons git."""
|
|
import asyncio
|
|
import logging
|
|
from pathlib import Path
|
|
import shutil
|
|
|
|
import git
|
|
|
|
from .util import get_hash_from_repository
|
|
from ..const import URL_HASSIO_ADDONS
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
|
|
class GitRepo(object):
|
|
"""Manage addons git repo."""
|
|
|
|
def __init__(self, config, loop, path, url):
|
|
"""Initialize git base wrapper."""
|
|
self.config = config
|
|
self.loop = loop
|
|
self.repo = None
|
|
self.path = path
|
|
self.url = url
|
|
self._lock = asyncio.Lock(loop=loop)
|
|
|
|
async def load(self):
|
|
"""Init git addon repo."""
|
|
if not self.path.is_dir():
|
|
return await self.clone()
|
|
|
|
async with self._lock:
|
|
try:
|
|
_LOGGER.info("Load addon %s repository", self.path)
|
|
self.repo = await self.loop.run_in_executor(
|
|
None, git.Repo, str(self.path))
|
|
|
|
except (git.InvalidGitRepositoryError, git.NoSuchPathError,
|
|
git.GitCommandError) as err:
|
|
_LOGGER.error("Can't load %s repo: %s.", self.path, err)
|
|
return False
|
|
|
|
return True
|
|
|
|
async def clone(self):
|
|
"""Clone git addon repo."""
|
|
async with self._lock:
|
|
try:
|
|
_LOGGER.info("Clone addon %s repository", self.url)
|
|
self.repo = await self.loop.run_in_executor(
|
|
None, git.Repo.clone_from, self.url, str(self.path))
|
|
|
|
except (git.InvalidGitRepositoryError, git.NoSuchPathError,
|
|
git.GitCommandError) as err:
|
|
_LOGGER.error("Can't clone %s repo: %s.", self.url, err)
|
|
return False
|
|
|
|
return True
|
|
|
|
async def pull(self):
|
|
"""Pull git addon repo."""
|
|
if self._lock.locked():
|
|
_LOGGER.warning("It is already a task in progress.")
|
|
return False
|
|
|
|
async with self._lock:
|
|
try:
|
|
_LOGGER.info("Pull addon %s repository", self.url)
|
|
await self.loop.run_in_executor(
|
|
None, self.repo.remotes.origin.pull)
|
|
|
|
except (git.InvalidGitRepositoryError, git.NoSuchPathError,
|
|
git.exc.GitCommandError) as err:
|
|
_LOGGER.error("Can't pull %s repo: %s.", self.url, err)
|
|
return False
|
|
|
|
return True
|
|
|
|
|
|
class GitRepoHassIO(GitRepo):
|
|
"""HassIO addons repository."""
|
|
|
|
def __init__(self, config, loop):
|
|
"""Initialize git hassio addon repository."""
|
|
super().__init__(
|
|
config, loop, config.path_addons_core, URL_HASSIO_ADDONS)
|
|
|
|
|
|
class GitRepoCustom(GitRepo):
|
|
"""Custom addons repository."""
|
|
|
|
def __init__(self, config, loop, url):
|
|
"""Initialize git hassio addon repository."""
|
|
path = Path(config.path_addons_git, get_hash_from_repository(url))
|
|
|
|
super().__init__(config, loop, path, url)
|
|
|
|
def remove(self):
|
|
"""Remove a custom addon."""
|
|
if self.path.is_dir():
|
|
_LOGGER.info("Remove custom addon repository %s", self.url)
|
|
|
|
def log_err(funct, path, _):
|
|
"""Log error."""
|
|
_LOGGER.warning("Can't remove %s", path)
|
|
|
|
shutil.rmtree(str(self.path), onerror=log_err)
|