Improve decorator type annotations [sabnzbd] (#104823)
parent
6ffc298986
commit
46ba62a3c0
|
@ -1,8 +1,9 @@
|
|||
"""Support for monitoring an SABnzbd NZB client."""
|
||||
from __future__ import annotations
|
||||
|
||||
from collections.abc import Callable
|
||||
from collections.abc import Callable, Coroutine
|
||||
import logging
|
||||
from typing import Any
|
||||
|
||||
from pysabnzbd import SabnzbdApiException
|
||||
import voluptuous as vol
|
||||
|
@ -189,7 +190,9 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
|
|||
update_device_identifiers(hass, entry)
|
||||
|
||||
@callback
|
||||
def extract_api(func: Callable) -> Callable:
|
||||
def extract_api(
|
||||
func: Callable[[ServiceCall, SabnzbdApiData], Coroutine[Any, Any, None]]
|
||||
) -> Callable[[ServiceCall], Coroutine[Any, Any, None]]:
|
||||
"""Define a decorator to get the correct api for a service call."""
|
||||
|
||||
async def wrapper(call: ServiceCall) -> None:
|
||||
|
|
Loading…
Reference in New Issue