Пример #1
0
def _get_ezviz_client_instance(entry: ConfigEntry) -> EzvizClient:
    """Initialize a new instance of EzvizClientApi."""
    ezviz_client = EzvizClient(
        entry.data[CONF_USERNAME], entry.data[CONF_PASSWORD], entry.data[CONF_REGION]
    )
    # , entry.options.get(CONF_TIMEOUT, DEFAULT_TIMEOUT)
    ezviz_client.login()
    return ezviz_client
Пример #2
0
def setup_platform(hass, config, add_entities, discovery_info=None):
    """Set up the Ezviz IP Cameras."""

    conf_cameras = config[CONF_CAMERAS]

    account = config[CONF_USERNAME]
    password = config[CONF_PASSWORD]

    try:
        ezviz_client = EzvizClient(account, password)
        ezviz_client.login()
        cameras = ezviz_client.load_cameras()

    except PyEzvizError as exp:
        _LOGGER.error(exp)
        return

    # now, let's build the HASS devices
    camera_entities = []

    # Add the cameras as devices in HASS
    for camera in cameras:

        camera_username = DEFAULT_CAMERA_USERNAME
        camera_password = ""
        camera_rtsp_stream = ""
        camera_serial = camera["serial"]

        # There seem to be a bug related to localRtspPort in Ezviz API...
        local_rtsp_port = DEFAULT_RTSP_PORT
        if camera["local_rtsp_port"] and camera["local_rtsp_port"] != 0:
            local_rtsp_port = camera["local_rtsp_port"]

        if camera_serial in conf_cameras:
            camera_username = conf_cameras[camera_serial][CONF_USERNAME]
            camera_password = conf_cameras[camera_serial][CONF_PASSWORD]
            camera_rtsp_stream = f"rtsp://{camera_username}:{camera_password}@{camera['local_ip']}:{local_rtsp_port}"
            _LOGGER.debug("Camera %s source stream: %s", camera["serial"],
                          camera_rtsp_stream)

        else:
            _LOGGER.info(
                "Found camera with serial %s without configuration. Add it to configuration.yaml to see the camera stream",
                camera_serial,
            )

        camera["username"] = camera_username
        camera["password"] = camera_password
        camera["rtsp_stream"] = camera_rtsp_stream

        camera["ezviz_camera"] = EzvizCamera(ezviz_client, camera_serial)

        camera_entities.append(HassEzvizCamera(**camera))

    add_entities(camera_entities)
Пример #3
0
def _get_ezviz_client_instance(data):
    """Initialize a new instance of EzvizClientApi."""

    ezviz_client = EzvizClient(
        data[CONF_USERNAME],
        data[CONF_PASSWORD],
        data.get(CONF_URL, EU_URL),
        data.get(CONF_TIMEOUT, DEFAULT_TIMEOUT),
    )

    ezviz_client.login()
    return ezviz_client
Пример #4
0
def _validate_and_create_auth(data: dict) -> dict[str, Any]:
    """Try to login to ezviz cloud account and return token."""
    # Verify cloud credentials by attempting a login request with username and password.
    # Return login token.

    ezviz_client = EzvizClient(
        data[CONF_USERNAME],
        data[CONF_PASSWORD],
        data[CONF_URL],
        data.get(CONF_TIMEOUT, DEFAULT_TIMEOUT),
    )

    ezviz_token = ezviz_client.login()

    auth_data = {
        CONF_SESSION_ID: ezviz_token[CONF_SESSION_ID],
        CONF_RFSESSION_ID: ezviz_token[CONF_RFSESSION_ID],
        CONF_URL: ezviz_token["api_url"],
        CONF_TYPE: ATTR_TYPE_CLOUD,
    }

    return auth_data
Пример #5
0
def setup_platform(hass, config, add_entities, discovery_info=None):
    """Set up the Ezviz IP Cameras."""

    conf_cameras = config[ATTR_CAMERAS]

    account = config[CONF_USERNAME]
    password = config[CONF_PASSWORD]
    region = config[CONF_REGION]

    try:
        ezviz_client = EzvizClient(account, password, region)
        ezviz_client.login()
        cameras = ezviz_client.load_cameras()

    except PyEzvizError as exp:
        _LOGGER.error(exp)
        return

    # now, let's build the HASS devices
    camera_entities = []

    # Add the cameras as devices in HASS
    for camera in cameras:

        camera_username = DEFAULT_CAMERA_USERNAME
        camera_password = ""
        camera_rtsp_stream = ""
        camera_serial = camera["serial"]

        # There seem to be a bug related to localRtspPort in Ezviz API...
        local_rtsp_port = DEFAULT_RTSP_PORT
        if camera["local_rtsp_port"] and camera["local_rtsp_port"] != 0:
            local_rtsp_port = camera["local_rtsp_port"]

        if camera_serial in conf_cameras:
            camera_username = conf_cameras[camera_serial][CONF_USERNAME]
            camera_password = conf_cameras[camera_serial][CONF_PASSWORD]
            camera_rtsp_stream = f"rtsp://{camera_username}:{camera_password}@{camera['local_ip']}:{local_rtsp_port}"
            _LOGGER.debug("Camera %s source stream: %s", camera["serial"],
                          camera_rtsp_stream)

        else:
            _LOGGER.info(
                "Found camera with serial %s without configuration. Add it to configuration.yaml to see the camera stream",
                camera_serial,
            )

        camera["username"] = camera_username
        camera["password"] = camera_password
        camera["rtsp_stream"] = camera_rtsp_stream
        camera["ezviz_client"] = ezviz_client

        camera["ezviz_camera"] = EzvizCamera(ezviz_client, camera_serial)

        camera_entities.append(HassEzvizCamera(hass, **camera))

    add_entities(camera_entities)
    """Setup Services"""
    def ezviz_wake_device(service):
        """Basicaly queries device to wake."""
        ezviz_client.get_detection_sensibility(str(service.data['serial']))

    def ezviz_switch_set(service):
        """Set camera switch service."""
        service_switch = getattr(DeviceSwitchType, service.data[ATTR_SWITCH])

        ezviz_client.switch_status(service.data[ATTR_SERIAL],
                                   service_switch.value,
                                   service.data[ATTR_ENABLE])

    def ezviz_alarm_sound(service):
        """Enable/Disable movement sound alarm."""
        ezviz_client.alarm_sound(str(service.data[ATTR_SERIAL]),
                                 int(service.data['level']), 1)

    def ezviz_set_alarm_detection_sensibility(service):
        """Set camera detection sensibility level service."""
        ezviz_client.detection_sensibility(str(service.data[ATTR_SERIAL]),
                                           int(service.data['level']),
                                           int(service.data['type']))

    def ezviz_ptz(service):
        """Camera PTZ service."""
        ezviz_client.ptzControl(
            str(service.data[ATTR_DIRECTION]).upper(),
            service.data[ATTR_SERIAL], 'START', service.data[ATTR_SPEED])
        ezviz_client.ptzControl(
            str(service.data[ATTR_DIRECTION]).upper(),
            service.data[ATTR_SERIAL], 'STOP', service.data[ATTR_SPEED])

    hass.services.register(DOMAIN, "ezviz_wake_device", ezviz_wake_device)
    hass.services.register(DOMAIN,
                           "ezviz_switch_set",
                           ezviz_switch_set,
                           schema=SERVICE_SET_SWITCH_SCHEMA)
    hass.services.register(DOMAIN, "ezviz_ptz", ezviz_ptz, SERVICE_PTZ_SCHEMA)
    hass.services.register(DOMAIN, "ezviz_alarm_sound", ezviz_alarm_sound)
    hass.services.register(DOMAIN, "ezviz_set_alarm_detection_sensibility",
                           ezviz_set_alarm_detection_sensibility)