7ec506907c
* Ensure async_get_system_info does not fail if supervisor is unavailable fixes #96470 * fix i/o in the event loop * fix tests * handle some more failure cases * more I/O here * coverage * coverage * Update homeassistant/helpers/system_info.py Co-authored-by: Paulus Schoutsen <balloob@gmail.com> * remove supervisor detection fallback * Update tests/helpers/test_system_info.py --------- Co-authored-by: Paulus Schoutsen <balloob@gmail.com>
86 lines
2.8 KiB
Python
86 lines
2.8 KiB
Python
"""Helper to gather system info."""
|
|
from __future__ import annotations
|
|
|
|
from functools import cache
|
|
from getpass import getuser
|
|
import logging
|
|
import os
|
|
import platform
|
|
from typing import Any
|
|
|
|
from homeassistant.const import __version__ as current_version
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.loader import bind_hass
|
|
from homeassistant.util.package import is_docker_env, is_virtual_env
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
|
|
@cache
|
|
def is_official_image() -> bool:
|
|
"""Return True if Home Assistant is running in an official container."""
|
|
return os.path.isfile("/OFFICIAL_IMAGE")
|
|
|
|
|
|
# Cache the result of getuser() because it can call getpwuid() which
|
|
# can do blocking I/O to look up the username in /etc/passwd.
|
|
cached_get_user = cache(getuser)
|
|
|
|
|
|
@bind_hass
|
|
async def async_get_system_info(hass: HomeAssistant) -> dict[str, Any]:
|
|
"""Return info about the system."""
|
|
is_hassio = hass.components.hassio.is_hassio()
|
|
|
|
info_object = {
|
|
"installation_type": "Unknown",
|
|
"version": current_version,
|
|
"dev": "dev" in current_version,
|
|
"hassio": is_hassio,
|
|
"virtualenv": is_virtual_env(),
|
|
"python_version": platform.python_version(),
|
|
"docker": False,
|
|
"arch": platform.machine(),
|
|
"timezone": str(hass.config.time_zone),
|
|
"os_name": platform.system(),
|
|
"os_version": platform.release(),
|
|
}
|
|
|
|
try:
|
|
info_object["user"] = cached_get_user()
|
|
except KeyError:
|
|
info_object["user"] = None
|
|
|
|
if platform.system() == "Darwin":
|
|
info_object["os_version"] = platform.mac_ver()[0]
|
|
elif platform.system() == "Linux":
|
|
info_object["docker"] = is_docker_env()
|
|
|
|
# Determine installation type on current data
|
|
if info_object["docker"]:
|
|
if info_object["user"] == "root" and is_official_image():
|
|
info_object["installation_type"] = "Home Assistant Container"
|
|
else:
|
|
info_object["installation_type"] = "Unsupported Third Party Container"
|
|
|
|
elif is_virtual_env():
|
|
info_object["installation_type"] = "Home Assistant Core"
|
|
|
|
# Enrich with Supervisor information
|
|
if is_hassio:
|
|
if not (info := hass.components.hassio.get_info()):
|
|
_LOGGER.warning("No Home Assistant Supervisor info available")
|
|
info = {}
|
|
|
|
host = hass.components.hassio.get_host_info() or {}
|
|
info_object["supervisor"] = info.get("supervisor")
|
|
info_object["host_os"] = host.get("operating_system")
|
|
info_object["docker_version"] = info.get("docker")
|
|
info_object["chassis"] = host.get("chassis")
|
|
|
|
if info.get("hassos") is not None:
|
|
info_object["installation_type"] = "Home Assistant OS"
|
|
else:
|
|
info_object["installation_type"] = "Home Assistant Supervised"
|
|
|
|
return info_object
|