Add hints to get_service in integrations (2/2) (#86693)

This commit is contained in:
epenet 2023-01-26 16:55:57 +01:00 committed by GitHub
parent 8a5a1b810a
commit 0d579f6ac3
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
20 changed files with 180 additions and 20 deletions

View file

@ -1,4 +1,6 @@
"""Support for the Mailgun mail notifications."""
from __future__ import annotations
import logging
from pymailgunner import (
@ -17,6 +19,8 @@ from homeassistant.components.notify import (
BaseNotificationService,
)
from homeassistant.const import CONF_API_KEY, CONF_DOMAIN, CONF_RECIPIENT, CONF_SENDER
from homeassistant.core import HomeAssistant
from homeassistant.helpers.typing import ConfigType, DiscoveryInfoType
from . import CONF_SANDBOX, DOMAIN as MAILGUN_DOMAIN
@ -33,7 +37,11 @@ PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend(
)
def get_service(hass, config, discovery_info=None):
def get_service(
hass: HomeAssistant,
config: ConfigType,
discovery_info: DiscoveryInfoType | None = None,
) -> MailgunNotificationService | None:
"""Get the Mailgun notification service."""
data = hass.data[MAILGUN_DOMAIN]
mailgun_service = MailgunNotificationService(