fix mypy check

This commit is contained in:
huangsong
2021-12-21 17:49:22 +08:00
parent 15ed00a62f
commit 0fd533da9e

View File

@ -1,6 +1,6 @@
import asyncio import asyncio
import inspect import inspect
from typing import Any, Callable, Dict, List, TYPE_CHECKING, Tuple, Type, Union from typing import Any, Callable, Dict, TYPE_CHECKING, Tuple, Type, Union
from ormar.exceptions import SignalDefinitionError from ormar.exceptions import SignalDefinitionError
@ -45,7 +45,7 @@ class Signal:
""" """
def __init__(self) -> None: def __init__(self) -> None:
self._receivers: Dict[Tuple[Union[int, Tuple[int, int]]], Callable] = {} self._receivers: Dict[Union[int, Tuple[int, int]], Callable] = {}
def connect(self, receiver: Callable) -> None: def connect(self, receiver: Callable) -> None:
""" """
@ -76,8 +76,9 @@ class Signal:
:rtype: bool :rtype: bool
""" """
new_receiver_key = make_id(receiver) new_receiver_key = make_id(receiver)
receiver = self._receivers.pop(new_receiver_key, None) receiver_func: Union[Callable, None] = self._receivers.pop(
return True if receiver is not None else False new_receiver_key, None)
return True if receiver_func is not None else False
async def send(self, sender: Type["Model"], **kwargs: Any) -> None: async def send(self, sender: Type["Model"], **kwargs: Any) -> None:
""" """