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,
)
from .const import DATA_MANAGER
from . import models
if TYPE_CHECKING:
from .manager import HomeAssistantBluetoothManager
pass
class HomeAssistantRemoteScanner(BaseHaRemoteScanner):
@@ -46,7 +46,9 @@ class HomeAssistantRemoteScanner(BaseHaRemoteScanner):
) -> None:
"""Initialize the scanner."""
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._cancel_stop: CALLBACK_TYPE | None = None
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
if TYPE_CHECKING:
from .manager import BluetoothManager
from .manager import HomeAssistantBluetoothManager
MANAGER: BluetoothManager | None = None
MANAGER: HomeAssistantBluetoothManager | None = None
BluetoothChange = Enum("BluetoothChange", "ADVERTISEMENT")