59 lines
1.5 KiB
Python
59 lines
1.5 KiB
Python
"""Support for Zabbix."""
|
|
import logging
|
|
from urllib.parse import urljoin
|
|
|
|
from pyzabbix import ZabbixAPI, ZabbixAPIException
|
|
import voluptuous as vol
|
|
|
|
from homeassistant.const import (
|
|
CONF_HOST,
|
|
CONF_PASSWORD,
|
|
CONF_PATH,
|
|
CONF_SSL,
|
|
CONF_USERNAME,
|
|
)
|
|
import homeassistant.helpers.config_validation as cv
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
DEFAULT_SSL = False
|
|
DEFAULT_PATH = "zabbix"
|
|
DOMAIN = "zabbix"
|
|
|
|
CONFIG_SCHEMA = vol.Schema(
|
|
{
|
|
DOMAIN: vol.Schema(
|
|
{
|
|
vol.Required(CONF_HOST): cv.string,
|
|
vol.Optional(CONF_PASSWORD): cv.string,
|
|
vol.Optional(CONF_PATH, default=DEFAULT_PATH): cv.string,
|
|
vol.Optional(CONF_SSL, default=DEFAULT_SSL): cv.boolean,
|
|
vol.Optional(CONF_USERNAME): cv.string,
|
|
}
|
|
)
|
|
},
|
|
extra=vol.ALLOW_EXTRA,
|
|
)
|
|
|
|
|
|
def setup(hass, config):
|
|
"""Set up the Zabbix component."""
|
|
|
|
conf = config[DOMAIN]
|
|
protocol = "https" if config[CONF_SSL] else "http"
|
|
|
|
url = urljoin(f"{protocol}://{conf[CONF_HOST]}", conf[CONF_PATH])
|
|
username = conf.get(CONF_USERNAME)
|
|
password = conf.get(CONF_PASSWORD)
|
|
|
|
zapi = ZabbixAPI(url)
|
|
try:
|
|
zapi.login(username, password)
|
|
_LOGGER.info("Connected to Zabbix API Version %s", zapi.api_version())
|
|
except ZabbixAPIException as login_exception:
|
|
_LOGGER.error("Unable to login to the Zabbix API: %s", login_exception)
|
|
return False
|
|
|
|
hass.data[DOMAIN] = zapi
|
|
return True
|