bump version, more tests, update docs

ormar/__init__.py

@@ -1,6 +1,18 @@
-from ormar.decorators import property_field
-from ormar.exceptions import ModelDefinitionError, MultipleMatches, NoMatch
+from ormar.decorators import (
+    post_delete,
+    post_save,
+    post_update,
+    pre_delete,
+    pre_save,
+    pre_update,
+    property_field,
+)
 from ormar.protocols import QuerySetProtocol, RelationProtocol  # noqa: I100
+from ormar.exceptions import (  # noqa: I100
+    ModelDefinitionError,
+    MultipleMatches,
+    NoMatch,
+)
 from ormar.fields import (  # noqa: I100
     BigInteger,
     Boolean,
@@ -22,6 +34,7 @@ from ormar.models import Model
 from ormar.models.metaclass import ModelMeta
 from ormar.queryset import QuerySet
 from ormar.relations import RelationType
+from ormar.signals import Signal
 
 
 class UndefinedType:  # pragma no cover
@@ -31,7 +44,7 @@ class UndefinedType:  # pragma no cover
 
 Undefined = UndefinedType()
 
-__version__ = "0.6.2"
+__version__ = "0.7.0"
 __all__ = [
     "Integer",
     "BigInteger",
@@ -59,4 +72,11 @@ __all__ = [
     "RelationProtocol",
     "ModelMeta",
     "property_field",
+    "post_delete",
+    "post_save",
+    "post_update",
+    "pre_delete",
+    "pre_save",
+    "pre_update",
+    "Signal",
 ]

ormar/decorators/__init__.py

@@ -1,5 +1,19 @@
 from ormar.decorators.property_field import property_field
+from ormar.decorators.signals import (
+    post_delete,
+    post_save,
+    post_update,
+    pre_delete,
+    pre_save,
+    pre_update,
+)
 
 __all__ = [
     "property_field",
+    "post_delete",
+    "post_save",
+    "post_update",
+    "pre_delete",
+    "pre_save",
+    "pre_update",
 ]

ormar/decorators/signals.py

@@ -1,11 +1,11 @@
-from typing import Any, Callable, List, TYPE_CHECKING, Type, Union
+from typing import Callable, List, TYPE_CHECKING, Type, Union
 
 if TYPE_CHECKING:  # pragma: no cover
     from ormar import Model
 
 
 def receiver(
-    signal: str, senders: Union[Type["Model"], List[Type["Model"]]], **kwargs: Any
+    signal: str, senders: Union[Type["Model"], List[Type["Model"]]]
 ) -> Callable:
     def _decorator(func: Callable) -> Callable:
         if not isinstance(senders, list):
@@ -14,43 +14,31 @@ def receiver(
             _senders = senders
         for sender in _senders:
             signals = getattr(sender.Meta.signals, signal)
-            signals.connect(func, **kwargs)
+            signals.connect(func)
         return func
 
     return _decorator
 
 
-def post_save(
-    senders: Union[Type["Model"], List[Type["Model"]]], **kwargs: Any
-) -> Callable:
-    return receiver(signal="post_save", senders=senders, **kwargs)
+def post_save(senders: Union[Type["Model"], List[Type["Model"]]],) -> Callable:
+    return receiver(signal="post_save", senders=senders)
 
 
-def post_update(
-    senders: Union[Type["Model"], List[Type["Model"]]], **kwargs: Any
-) -> Callable:
-    return receiver(signal="post_update", senders=senders, **kwargs)
+def post_update(senders: Union[Type["Model"], List[Type["Model"]]],) -> Callable:
+    return receiver(signal="post_update", senders=senders)
 
 
-def post_delete(
-    senders: Union[Type["Model"], List[Type["Model"]]], **kwargs: Any
-) -> Callable:
-    return receiver(signal="post_delete", senders=senders, **kwargs)
+def post_delete(senders: Union[Type["Model"], List[Type["Model"]]],) -> Callable:
+    return receiver(signal="post_delete", senders=senders)
 
 
-def pre_save(
-    senders: Union[Type["Model"], List[Type["Model"]]], **kwargs: Any
-) -> Callable:
-    return receiver(signal="pre_save", senders=senders, **kwargs)
+def pre_save(senders: Union[Type["Model"], List[Type["Model"]]],) -> Callable:
+    return receiver(signal="pre_save", senders=senders)
 
 
-def pre_update(
-    senders: Union[Type["Model"], List[Type["Model"]]], **kwargs: Any
-) -> Callable:
-    return receiver(signal="pre_update", senders=senders, **kwargs)
+def pre_update(senders: Union[Type["Model"], List[Type["Model"]]]) -> Callable:
+    return receiver(signal="pre_update", senders=senders)
 
 
-def pre_delete(
-    senders: Union[Type["Model"], List[Type["Model"]]], **kwargs: Any
-) -> Callable:
-    return receiver(signal="pre_delete", senders=senders, **kwargs)
+def pre_delete(senders: Union[Type["Model"], List[Type["Model"]]]) -> Callable:
+    return receiver(signal="pre_delete", senders=senders)