From 7c0dc2ff1f22c87410e12bd8e0f3463256bcb15b Mon Sep 17 00:00:00 2001 From: epenet <6771947+epenet@users.noreply.github.com> Date: Thu, 13 Jan 2022 14:20:05 +0100 Subject: [PATCH] Add typing to async_get_handler in demo (#64029) Co-authored-by: Franck Nijhof Co-authored-by: epenet --- homeassistant/components/demo/mailbox.py | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/homeassistant/components/demo/mailbox.py b/homeassistant/components/demo/mailbox.py index 860524dfd7c..bc5467faada 100644 --- a/homeassistant/components/demo/mailbox.py +++ b/homeassistant/components/demo/mailbox.py @@ -1,9 +1,13 @@ """Support for a demo mailbox.""" +from __future__ import annotations + from hashlib import sha1 import logging import os from homeassistant.components.mailbox import CONTENT_TYPE_MPEG, Mailbox, StreamError +from homeassistant.core import HomeAssistant +from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType from homeassistant.util import dt _LOGGER = logging.getLogger(__name__) @@ -11,7 +15,11 @@ _LOGGER = logging.getLogger(__name__) MAILBOX_NAME = "DemoMailbox" -async def async_get_handler(hass, config, discovery_info=None): +async def async_get_handler( + hass: HomeAssistant, + config: ConfigType, + discovery_info: DiscoveryInfoType | None = None, +) -> Mailbox: """Set up the Demo mailbox.""" return DemoMailbox(hass, MAILBOX_NAME)