mirror of
https://github.com/home-assistant/supervisor.git
synced 2025-04-20 19:27:16 +00:00
d285fd4ad4
* Fix handling with docker container * Fix lint * update version * fix lint v2 * fix signal handling * fix log output
106 lines
3.1 KiB
Python
106 lines
3.1 KiB
Python
"""Bootstrap HassIO."""
|
|
import logging
|
|
import os
|
|
import stat
|
|
import signal
|
|
|
|
from colorlog import ColoredFormatter
|
|
|
|
from .const import SOCKET_DOCKER
|
|
from .config import CoreConfig
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
|
|
def initialize_system_data(websession):
|
|
"""Setup default config and create folders."""
|
|
config = CoreConfig(websession)
|
|
|
|
# homeassistant config folder
|
|
if not os.path.isdir(config.path_config):
|
|
_LOGGER.info(
|
|
"Create Home-Assistant config folder %s", config.path_config)
|
|
os.mkdir(config.path_config)
|
|
|
|
# homeassistant ssl folder
|
|
if not os.path.isdir(config.path_ssl):
|
|
_LOGGER.info("Create Home-Assistant ssl folder %s", config.path_ssl)
|
|
os.mkdir(config.path_ssl)
|
|
|
|
# homeassistant addon data folder
|
|
if not os.path.isdir(config.path_addons_data):
|
|
_LOGGER.info("Create Home-Assistant addon data folder %s",
|
|
config.path_addons_data)
|
|
os.mkdir(config.path_addons_data)
|
|
|
|
if not os.path.isdir(config.path_addons_custom):
|
|
_LOGGER.info("Create Home-Assistant addon custom folder %s",
|
|
config.path_addons_custom)
|
|
os.mkdir(config.path_addons_custom)
|
|
|
|
return config
|
|
|
|
|
|
def initialize_logging():
|
|
"""Setup the logging."""
|
|
logging.basicConfig(level=logging.INFO)
|
|
fmt = ("%(asctime)s %(levelname)s (%(threadName)s) "
|
|
"[%(name)s] %(message)s")
|
|
colorfmt = "%(log_color)s{}%(reset)s".format(fmt)
|
|
datefmt = '%y-%m-%d %H:%M:%S'
|
|
|
|
# suppress overly verbose logs from libraries that aren't helpful
|
|
logging.getLogger("aiohttp.access").setLevel(logging.WARNING)
|
|
|
|
logging.getLogger().handlers[0].setFormatter(ColoredFormatter(
|
|
colorfmt,
|
|
datefmt=datefmt,
|
|
reset=True,
|
|
log_colors={
|
|
'DEBUG': 'cyan',
|
|
'INFO': 'green',
|
|
'WARNING': 'yellow',
|
|
'ERROR': 'red',
|
|
'CRITICAL': 'red',
|
|
}
|
|
))
|
|
|
|
|
|
def check_environment():
|
|
"""Check if all environment are exists."""
|
|
for key in ('SUPERVISOR_SHARE', 'SUPERVISOR_NAME',
|
|
'HOMEASSISTANT_REPOSITORY'):
|
|
try:
|
|
os.environ[key]
|
|
except KeyError:
|
|
_LOGGER.fatal("Can't find %s in env!", key)
|
|
return False
|
|
|
|
mode = os.stat(SOCKET_DOCKER)[stat.ST_MODE]
|
|
if not stat.S_ISSOCK(mode):
|
|
_LOGGER.fatal("Can't find docker socket!")
|
|
return False
|
|
|
|
return True
|
|
|
|
|
|
def reg_signal(loop, hassio):
|
|
"""Register SIGTERM, SIGKILL to stop system."""
|
|
try:
|
|
loop.add_signal_handler(
|
|
signal.SIGTERM, lambda: loop.create_task(hassio.stop()))
|
|
except (ValueError, RuntimeError):
|
|
_LOGGER.warning("Could not bind to SIGTERM")
|
|
|
|
try:
|
|
loop.add_signal_handler(
|
|
signal.SIGHUP, lambda: loop.create_task(hassio.stop()))
|
|
except (ValueError, RuntimeError):
|
|
_LOGGER.warning("Could not bind to SIGHUP")
|
|
|
|
try:
|
|
loop.add_signal_handler(
|
|
signal.SIGINT, lambda: loop.create_task(hassio.stop()))
|
|
except (ValueError, RuntimeError):
|
|
_LOGGER.warning("Could not bind to SIGINT")
|