evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Make CoordinatorEntity generic (#49854)

Browse source
This commit is contained in:
Ruslan Sayfutdinov 2021-04-30 21:24:27 +01:00 • committed by GitHub
parent dc46a213a7
commit f53284b200
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

8
homeassistant/helpers/update_coordinator.py
View file

@ -6,7 +6,7 @@ from collections.abc import Awaitable
from datetime import datetime, timedelta
import logging
from time import monotonic
from typing import Any, Callable, Generic, TypeVar
from typing import Callable, Generic, TypeVar
import urllib.error
import aiohttp
@ -25,8 +25,6 @@ REQUEST_REFRESH_DEFAULT_IMMEDIATE = True
T = TypeVar("T")
# mypy: disallow-any-generics
class UpdateFailed(Exception):
"""Raised when an update has failed."""
@ -294,10 +292,10 @@ class DataUpdateCoordinator(Generic[T]):
self._unsub_refresh = None
class CoordinatorEntity(entity.Entity):
class CoordinatorEntity(Generic[T], entity.Entity):
"""A class for entities using DataUpdateCoordinator."""
def __init__(self, coordinator: DataUpdateCoordinator[Any]) -> None:
def __init__(self, coordinator: DataUpdateCoordinator[T]) -> None:
"""Create the entity with a DataUpdateCoordinator."""
self.coordinator = coordinator