fix types

This commit is contained in:
J. Nick Koston
2023-12-05 13:42:37 -10:00
parent a3253173d0
commit 5cb7865309
2 changed files with 7 additions and 5 deletions

View File

@@ -16,10 +16,10 @@ from homeassistant.core import (
callback as hass_callback, callback as hass_callback,
) )
from .const import DATA_MANAGER from . import models
if TYPE_CHECKING: if TYPE_CHECKING:
from .manager import HomeAssistantBluetoothManager pass
class HomeAssistantRemoteScanner(BaseHaRemoteScanner): class HomeAssistantRemoteScanner(BaseHaRemoteScanner):
@@ -46,7 +46,9 @@ class HomeAssistantRemoteScanner(BaseHaRemoteScanner):
) -> None: ) -> None:
"""Initialize the scanner.""" """Initialize the scanner."""
self.hass = hass self.hass = hass
manager: HomeAssistantBluetoothManager = hass.data[DATA_MANAGER] manager = models.MANAGER
if TYPE_CHECKING:
assert manager is not None
self._storage = manager.storage self._storage = manager.storage
self._cancel_stop: CALLBACK_TYPE | None = None self._cancel_stop: CALLBACK_TYPE | None = None
super().__init__(scanner_id, name, new_info_callback, connector, connectable) super().__init__(scanner_id, name, new_info_callback, connector, connectable)

View File

@@ -8,10 +8,10 @@ from typing import TYPE_CHECKING
from home_assistant_bluetooth import BluetoothServiceInfoBleak from home_assistant_bluetooth import BluetoothServiceInfoBleak
if TYPE_CHECKING: if TYPE_CHECKING:
from .manager import BluetoothManager from .manager import HomeAssistantBluetoothManager
MANAGER: BluetoothManager | None = None MANAGER: HomeAssistantBluetoothManager | None = None
BluetoothChange = Enum("BluetoothChange", "ADVERTISEMENT") BluetoothChange = Enum("BluetoothChange", "ADVERTISEMENT")