jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-22 20:57:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

Enable basic type checking for azure_event_hub (#55047)

Browse Source 
* Enable basic type checking for azure_event_hub

* Update homeassistant/components/azure_event_hub/__init__.py

Co-authored-by: Martin Hjelmare <marhje52@gmail.com>

* Disable false pylint positive

Co-authored-by: Martin Hjelmare <marhje52@gmail.com>
This commit is contained in:
Erik Montnemery 2021-08-23 17:04:42 +02:00 committed by GitHub
parent b76e8c5722
commit 68f1c19049
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
4 changed files with 14 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
homeassistant/components/azure_event_hub/__init__.py
View File

@ -5,9 +5,9 @@ import asyncio
import json import json
import logging import logging
import time import time
from typing import Any from typing import Any, Callable
from azure.eventhub import EventData from azure.eventhub import EventData, EventDataBatch
from azure.eventhub.aio import EventHubProducerClient, EventHubSharedKeyCredential from azure.eventhub.aio import EventHubProducerClient, EventHubSharedKeyCredential
from azure.eventhub.exceptions import EventHubError from azure.eventhub.exceptions import EventHubError
import voluptuous as vol import voluptuous as vol
@ -109,14 +109,16 @@ class AzureEventHub:
) -> None: ) -> None:
"""Initialize the listener.""" """Initialize the listener."""
self.hass = hass self.hass = hass
self.queue = asyncio.PriorityQueue() self.queue: asyncio.PriorityQueue[ # pylint: disable=unsubscriptable-object
tuple[int, tuple[float, Event | None]]
] = asyncio.PriorityQueue()
self._client_args = client_args self._client_args = client_args
self._conn_str_client = conn_str_client self._conn_str_client = conn_str_client
self._entities_filter = entities_filter self._entities_filter = entities_filter
self._send_interval = send_interval self._send_interval = send_interval
self._max_delay = max_delay + send_interval self._max_delay = max_delay + send_interval
self._listener_remover = None self._listener_remover: Callable[[], None] | None = None
self._next_send_remover = None self._next_send_remover: Callable[[], None] | None = None
self.shutdown = False self.shutdown = False
async def async_start(self) -> None: async def async_start(self) -> None:
@ -169,7 +171,7 @@ class AzureEventHub:
self.hass, self._send_interval, self.async_send self.hass, self._send_interval, self.async_send
) )
async def fill_batch(self, client) -> None: async def fill_batch(self, client) -> tuple[EventDataBatch, int]:
"""Return a batch of events formatted for writing. """Return a batch of events formatted for writing.
Uses get_nowait instead of await get, because the functions batches and doesn't wait for each single event, the send function is called. Uses get_nowait instead of await get, because the functions batches and doesn't wait for each single event, the send function is called.
@ -207,7 +209,7 @@ class AzureEventHub:
return event_batch, dequeue_count return event_batch, dequeue_count
def _event_to_filtered_event_data(self, event: Event) -> None: def _event_to_filtered_event_data(self, event: Event) -> EventData | None:
"""Filter event states and create EventData object.""" """Filter event states and create EventData object."""
state = event.data.get("new_state") state = event.data.get("new_state")
if ( if (

6
homeassistant/components/azure_event_hub/const.py
View File

@ -1,4 +1,8 @@
"""Constants and shared schema for the Azure Event Hub integration.""" """Constants and shared schema for the Azure Event Hub integration."""
from __future__ import annotations
from typing import Any
DOMAIN = "azure_event_hub" DOMAIN = "azure_event_hub"
CONF_EVENT_HUB_NAMESPACE = "event_hub_namespace" CONF_EVENT_HUB_NAMESPACE = "event_hub_namespace"
@ -10,4 +14,4 @@ CONF_SEND_INTERVAL = "send_interval"
CONF_MAX_DELAY = "max_delay" CONF_MAX_DELAY = "max_delay"
CONF_FILTER = "filter" CONF_FILTER = "filter"
ADDITIONAL_ARGS = {"logging_enable": False} ADDITIONAL_ARGS: dict[str, Any] = {"logging_enable": False}

3
mypy.ini
View File

@ -1268,9 +1268,6 @@ warn_unreachable = false
[mypy-homeassistant.components.awair.*] [mypy-homeassistant.components.awair.*]
ignore_errors = true ignore_errors = true
[mypy-homeassistant.components.azure_event_hub.*]
ignore_errors = true
[mypy-homeassistant.components.blueprint.*] [mypy-homeassistant.components.blueprint.*]
ignore_errors = true ignore_errors = true

1
script/hassfest/mypy_config.py
View File

@ -15,7 +15,6 @@ from .model import Config, Integration
# Do your best to not add anything new here. # Do your best to not add anything new here.
IGNORED_MODULES: Final[list[str]] = [ IGNORED_MODULES: Final[list[str]] = [
"homeassistant.components.awair.*", "homeassistant.components.awair.*",
"homeassistant.components.azure_event_hub.*",
"homeassistant.components.blueprint.*", "homeassistant.components.blueprint.*",
"homeassistant.components.bmw_connected_drive.*", "homeassistant.components.bmw_connected_drive.*",
"homeassistant.components.cert_expiry.*", "homeassistant.components.cert_expiry.*",