From bfdf27cc9a89ddc99255061a4be1dbf5c20abf5e Mon Sep 17 00:00:00 2001 From: epenet Date: Thu, 13 Jan 2022 09:47:51 +0000 Subject: [PATCH 1/2] Add typing to async_get_handler in demo --- 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 860524dfd7cc0..6a7b986979ad6 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 @@ 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 | None: """Set up the Demo mailbox.""" return DemoMailbox(hass, MAILBOX_NAME) From 8011ab92ac93d87140d0763774688b1adbc5fa5f Mon Sep 17 00:00:00 2001 From: epenet <6771947+epenet@users.noreply.github.com> Date: Thu, 13 Jan 2022 12:29:35 +0100 Subject: [PATCH 2/2] Update homeassistant/components/demo/mailbox.py Co-authored-by: Franck Nijhof --- homeassistant/components/demo/mailbox.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/homeassistant/components/demo/mailbox.py b/homeassistant/components/demo/mailbox.py index 6a7b986979ad6..bc5467faada2e 100644 --- a/homeassistant/components/demo/mailbox.py +++ b/homeassistant/components/demo/mailbox.py @@ -19,7 +19,7 @@ async def async_get_handler( hass: HomeAssistant, config: ConfigType, discovery_info: DiscoveryInfoType | None = None, -) -> Mailbox | None: +) -> Mailbox: """Set up the Demo mailbox.""" return DemoMailbox(hass, MAILBOX_NAME)