Helpers type hint additions and improvements (#27986)

* Helpers type hint additions and improvements

* Fix async setup dump callback signature
This commit is contained in:
Ville Skyttä 2019-10-21 17:54:59 +03:00 committed by GitHub
parent ba10d5d604
commit 70ddab2f3c
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
3 changed files with 57 additions and 41 deletions

View file

@ -6,7 +6,7 @@ import os
from typing import Dict, List, Optional, Callable, Union, Any, Type
from homeassistant.const import EVENT_HOMEASSISTANT_STOP
from homeassistant.core import HomeAssistant, callback
from homeassistant.core import HomeAssistant, callback, CALLBACK_TYPE
from homeassistant.loader import bind_hass
from homeassistant.util import json as json_util
from homeassistant.helpers.event import async_call_later
@ -72,8 +72,8 @@ class Store:
self.hass = hass
self._private = private
self._data: Optional[Dict[str, Any]] = None
self._unsub_delay_listener = None
self._unsub_stop_listener = None
self._unsub_delay_listener: Optional[CALLBACK_TYPE] = None
self._unsub_stop_listener: Optional[CALLBACK_TYPE] = None
self._write_lock = asyncio.Lock()
self._load_task: Optional[asyncio.Future] = None
self._encoder = encoder
@ -137,9 +137,7 @@ class Store:
await self._async_handle_write_data()
@callback
def async_delay_save(
self, data_func: Callable[[], Dict], delay: Optional[int] = None
) -> None:
def async_delay_save(self, data_func: Callable[[], Dict], delay: float = 0) -> None:
"""Save data with an optional delay."""
self._data = {"version": self.version, "key": self.key, "data_func": data_func}