jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-19 19:27:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

Correct return typing for catch_log_exception (#78399)

Browse Source
This commit is contained in:
Jan Bouwhuis 2022-09-17 17:27:22 +02:00 committed by GitHub
parent 5a6a474bbe
commit 35221e9a61
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 9 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
homeassistant/util/logging.py
View File

@ -120,27 +120,30 @@ def log_exception(format_err: Callable[..., Any], *args: Any) -> None:
def catch_log_exception( def catch_log_exception(
func: Callable[..., Coroutine[Any, Any, Any]], format_err: Callable[..., Any] func: Callable[..., Coroutine[Any, Any, Any]], format_err: Callable[..., Any]
) -> Callable[..., Coroutine[Any, Any, None]]: ) -> Callable[..., Coroutine[Any, Any, None]]:
"""Overload for Callables that return a Coroutine.""" """Overload for Coroutine that returns a Coroutine."""
@overload @overload
def catch_log_exception( def catch_log_exception(
func: Callable[..., Any], format_err: Callable[..., Any] func: Callable[..., Any], format_err: Callable[..., Any]
) -> Callable[..., None | Coroutine[Any, Any, None]]: ) -> Callable[..., None] | Callable[..., Coroutine[Any, Any, None]]:
"""Overload for Callables that return Any.""" """Overload for a callback that returns a callback."""
def catch_log_exception( def catch_log_exception(
func: Callable[..., Any], format_err: Callable[..., Any] func: Callable[..., Any], format_err: Callable[..., Any]
) -> Callable[..., None | Coroutine[Any, Any, None]]: ) -> Callable[..., None] | Callable[..., Coroutine[Any, Any, None]]:
"""Decorate a callback to catch and log exceptions.""" """Decorate a function func to catch and log exceptions.
If func is a coroutine function, a coroutine function will be returned.
If func is a callback, a callback will be returned.
"""
# Check for partials to properly determine if coroutine function # Check for partials to properly determine if coroutine function
check_func = func check_func = func
while isinstance(check_func, partial): while isinstance(check_func, partial):
check_func = check_func.func check_func = check_func.func
wrapper_func: Callable[..., None | Coroutine[Any, Any, None]] wrapper_func: Callable[..., None] | Callable[..., Coroutine[Any, Any, None]]
if asyncio.iscoroutinefunction(check_func): if asyncio.iscoroutinefunction(check_func):
async_func = cast(Callable[..., Coroutine[Any, Any, None]], func) async_func = cast(Callable[..., Coroutine[Any, Any, None]], func)