mirror of
https://github.com/home-assistant/core.git
synced 2026-05-17 14:01:34 +01:00
Paul Tarjan
88 lines
2.3 KiB
Python
88 lines
2.3 KiB
Python
"""The Hikvision integration."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from dataclasses import dataclass
|
|
import logging
|
|
|
|
from pyhik.hikvision import HikCamera
|
|
import requests
|
|
|
|
from homeassistant.config_entries import ConfigEntry
|
|
from homeassistant.const import (
|
|
CONF_HOST,
|
|
CONF_PASSWORD,
|
|
CONF_PORT,
|
|
CONF_SSL,
|
|
CONF_USERNAME,
|
|
Platform,
|
|
)
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.exceptions import ConfigEntryNotReady
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
PLATFORMS = [Platform.BINARY_SENSOR]
|
|
|
|
|
|
@dataclass
|
|
class HikvisionData:
|
|
"""Data class for Hikvision runtime data."""
|
|
|
|
camera: HikCamera
|
|
device_id: str
|
|
device_name: str
|
|
device_type: str
|
|
|
|
|
|
type HikvisionConfigEntry = ConfigEntry[HikvisionData]
|
|
|
|
|
|
async def async_setup_entry(hass: HomeAssistant, entry: HikvisionConfigEntry) -> bool:
|
|
"""Set up Hikvision from a config entry."""
|
|
host = entry.data[CONF_HOST]
|
|
port = entry.data[CONF_PORT]
|
|
username = entry.data[CONF_USERNAME]
|
|
password = entry.data[CONF_PASSWORD]
|
|
ssl = entry.data[CONF_SSL]
|
|
|
|
protocol = "https" if ssl else "http"
|
|
url = f"{protocol}://{host}"
|
|
|
|
try:
|
|
camera = await hass.async_add_executor_job(
|
|
HikCamera, url, port, username, password
|
|
)
|
|
except requests.exceptions.RequestException as err:
|
|
raise ConfigEntryNotReady(f"Unable to connect to {host}") from err
|
|
|
|
device_id = camera.get_id()
|
|
if device_id is None:
|
|
raise ConfigEntryNotReady(f"Unable to get device ID from {host}")
|
|
|
|
device_name = camera.get_name or host
|
|
device_type = camera.get_type or "Camera"
|
|
|
|
entry.runtime_data = HikvisionData(
|
|
camera=camera,
|
|
device_id=device_id,
|
|
device_name=device_name,
|
|
device_type=device_type,
|
|
)
|
|
|
|
# Start the event stream
|
|
await hass.async_add_executor_job(camera.start_stream)
|
|
|
|
await hass.config_entries.async_forward_entry_setups(entry, PLATFORMS)
|
|
|
|
return True
|
|
|
|
|
|
async def async_unload_entry(hass: HomeAssistant, entry: HikvisionConfigEntry) -> bool:
|
|
"""Unload a config entry."""
|
|
if unload_ok := await hass.config_entries.async_unload_platforms(entry, PLATFORMS):
|
|
# Stop the event stream
|
|
await hass.async_add_executor_job(entry.runtime_data.camera.disconnect)
|
|
|
|
return unload_ok
|