buz 2.22.1__py3-none-any.whl → 2.23.0__py3-none-any.whl

buz/command/synchronous/command_handler.py

@@ -3,11 +3,18 @@ from typing import Generic, Type, TypeVar
 
 from buz import Handler
 from buz.command import Command
+from buz.wrapper.async_unsafe import async_unsafe
 
 TCommand = TypeVar("TCommand", bound=Command)
 
 
 class CommandHandler(Generic[TCommand], Handler[TCommand], ABC):
+    def __new__(cls, *args, **kwargs):
+        instance = super().__new__(cls)
+        original_handle = instance.handle
+        instance.handle = async_unsafe()(original_handle)
+        return instance
+
     @classmethod
     @abstractmethod
     def handles(cls) -> Type[TCommand]:

buz/query/synchronous/query_handler.py

@@ -3,12 +3,19 @@ from typing import Generic, Type, TypeVar
 
 from buz import Handler
 from buz.query import Query, QueryResponse
+from buz.wrapper.async_unsafe import async_unsafe
 
 TQuery = TypeVar("TQuery", bound=Query)
 TQueryResponse = TypeVar("TQueryResponse", bound=QueryResponse)
 
 
 class QueryHandler(Generic[TQuery, TQueryResponse], Handler[TQuery], ABC):
+    def __new__(cls, *args, **kwargs):
+        instance = super().__new__(cls)
+        original_handle = instance.handle
+        instance.handle = async_unsafe()(original_handle)
+        return instance
+
     @classmethod
     @abstractmethod
     def handles(cls) -> Type[TQuery]:

buz/wrapper/async_unsafe.py

@@ -0,0 +1,67 @@
+import logging
+import os
+from asyncio import get_running_loop
+from functools import wraps
+from typing import Callable, Any
+
+from buz.wrapper.synchronous_only_operation_exception import SynchronousOnlyOperationException
+
+logger = logging.getLogger(__name__)
+
+
+def async_unsafe() -> Callable[[Callable], Callable]:
+    """
+    Copy-paste from Django source code. If BUZ_SYNC_HANDLERS_FAIL_ON_ASYNC env var is True then an exception is raised to
+    prevent event loop from being blocked. If the env var is False, a warning is logged.
+    """
+
+    def decorator(func: Callable) -> Callable:
+        @wraps(func)
+        def inner(*args: Any, **kwargs: Any) -> Any:
+            try:
+                get_running_loop()
+            except RuntimeError:
+                pass
+            else:
+                if __get_bool_env("BUZ_SYNC_HANDLERS_FAIL_ON_ASYNC") is True:
+                    raise SynchronousOnlyOperationException()
+
+                logger.warning(
+                    f"You should not call this from an async context - use a thread or sync_to_async. Class {__get_class_name(func, args)}"
+                )
+
+            return func(*args, **kwargs)
+
+        return inner
+
+    return decorator
+
+
+def __get_bool_env(var_name: str) -> bool:
+    """
+    >>> get_bool_env("MY_BOOL_VAR")
+    False
+    >>> os.environ["MY_BOOL_VAR"] = "1"
+    >>> get_bool_env("MY_BOOL_VAR")
+    True
+    >>> os.environ["MY_BOOL_VAR"] = "true"
+    >>> get_bool_env("MY_BOOL_VAR")
+    True
+    >>> os.environ["MY_BOOL_VAR"] = "True"
+    >>> get_bool_env("MY_BOOL_VAR")
+    True
+    >>> os.environ["MY_BOOL_VAR"] = "0"
+    >>> get_bool_env("MY_BOOL_VAR")
+    False
+    >>> os.environ["MY_BOOL_VAR"] = "false"
+    >>> get_bool_env("MY_BOOL_VAR")
+    False
+    """
+    return os.environ.get(var_name, "0").lower() in ("true", "1")
+
+
+def __get_class_name(func: Callable, args: tuple) -> str:
+    if hasattr(func, "__self__"):
+        return func.__self__.__class__.__name__
+
+    return args[0].__class__.__name__ if len(args) > 0 else "Unknown"

buz/wrapper/synchronous_only_operation_exception.py

@@ -0,0 +1,3 @@
+class SynchronousOnlyOperationException(Exception):
+    def __init__(self):
+        super().__init__("You cannot call this from an async context - use a thread or sync_to_async.")

{buz-2.22.1.dist-info → buz-2.23.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: buz
-Version: 2.22.1
+Version: 2.23.0
 Summary: Buz is a set of light, simple and extensible implementations of event, command and query buses.
 License: MIT
 Author: Luis Pintado Lozano

{buz-2.22.1.dist-info → buz-2.23.0.dist-info}/RECORD

@@ -15,7 +15,7 @@ buz/command/more_than_one_command_handler_related_exception.py,sha256=dNCMXCAKVc
 buz/command/synchronous/__init__.py,sha256=ZAH35jpte6zpxb70iQfZfGSg_OsXqVNYwNkCJy7Qnzk,268
 buz/command/synchronous/base_command_handler.py,sha256=GkmxttA1_Wu2X5O7iR-tLRdnmT-JDgeyURPwkjLPGhg,1393
 buz/command/synchronous/command_bus.py,sha256=FHJH4lerThu5fWRISM0VIK8ebyfqgDGqNNCqItbmGPI,175
-buz/command/synchronous/command_handler.py,sha256=pP11Njbth89NQiVPqlALtKmabVtOX4sucdbiCp-sceI,422
+buz/command/synchronous/command_handler.py,sha256=M9nQhW9Ilkix09U6Pv6ZpoqA-gezrm7H_gqjoLDc_A4,676
 buz/command/synchronous/middleware/__init__.py,sha256=wE97veOW2cCGpTZ-3bvc3aU6NdQmSOYfdX9kAUETJ0w,406
 buz/command/synchronous/middleware/base_handle_middleware.py,sha256=qHnbucL7j6Dnw7E8ACBf32pzDD9squYPp6UiK2heb5A,774
 buz/command/synchronous/middleware/handle_middleware.py,sha256=oIsmB30mu-Cd9IQc1qMKbUIaIMHipSEXOw2swVrtbkk,420
@@ -250,7 +250,7 @@ buz/query/synchronous/middleware/base_handle_middleware.py,sha256=4lqJB2C-GHl3T5
 buz/query/synchronous/middleware/handle_middleware.py,sha256=Xr2HSlRrW3slluyUiJ6zH0Rw0oeWD3uMN-1hjuFbC_k,433
 buz/query/synchronous/middleware/handle_middleware_chain_resolver.py,sha256=OIpGLJ_2a8cGsp-LmA3Q1Lvb1pB65MIuA-geiPrQKHM,1070
 buz/query/synchronous/query_bus.py,sha256=eYl_sGH5TcELkOXc4RnV4Tkmcs-VLc1vzd-YMduQ1YI,189
-buz/query/synchronous/query_handler.py,sha256=q8zqXjU9btid_q4wbL73QgaiWjMzGDFvwZ5AQN4q0CA,505
+buz/query/synchronous/query_handler.py,sha256=NuolVPZelmjOe6-q_hFC3Y0gm7ih2A6aanKufo6wHR0,759
 buz/query/synchronous/self_process/__init__.py,sha256=fU1OoXXXH5dMGKz8y7mwTVvyhNj6BCKDTxuxH_q-leM,125
 buz/query/synchronous/self_process/self_process_query_bus.py,sha256=pKGJxXBWtqU4i0fzb30OCNhAVPCkUh7IlfNzgAhCUC8,1157
 buz/query/synchronous/synced_async/__init__.py,sha256=TdFmIBeFIpl3Tvmh_FJpJMXJdPdfRxOstVqnPUi23mo,125
@@ -264,8 +264,10 @@ buz/serializer/message_to_bytes_serializer.py,sha256=uQ6JSTn24mLQvN48alwVXa6lfgf
 buz/serializer/message_to_json_bytes_serializer.py,sha256=RGZJ64t4t4Pz2FCASZZCv-2LiWnertC8scE9ZzQkBsU,764
 buz/wrapper/__init__.py,sha256=GnRdJFcncn-qp0hzDG9dBHLmTJSbHFVjE_yr-MdW_n4,77
 buz/wrapper/async_to_sync.py,sha256=OfK-vrVUhuN-LLLvekLdMbQYtH0ue5lfbvuasj6ovMI,698
+buz/wrapper/async_unsafe.py,sha256=vg-BM8Dh7MqKiMAc8qsMPexLUuHI1bn4APzMXeSrBgk,2026
 buz/wrapper/event_loop.py,sha256=pfBJ1g-8A2a3YgW8Gf9Fg0kkewoh3-wgTy2KIFDyfHk,266
-buz-2.22.1.dist-info/LICENSE,sha256=jcLgcIIVaBqaZNwe0kzGWSU99YgwMcI0IGv142wkYSM,1062
-buz-2.22.1.dist-info/METADATA,sha256=baDIRyxTLM2Z4gKrvVH9War7HIp0V0WFgSlbTR4x6Eo,12580
-buz-2.22.1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-buz-2.22.1.dist-info/RECORD,,
+buz/wrapper/synchronous_only_operation_exception.py,sha256=qCTaNSuIY7_HXTOmzlNGhks1nRApeW5K8W5aHC-dL-4,180
+buz-2.23.0.dist-info/LICENSE,sha256=jcLgcIIVaBqaZNwe0kzGWSU99YgwMcI0IGv142wkYSM,1062
+buz-2.23.0.dist-info/METADATA,sha256=YxHyoXLE0SlA_UxWPBACGbNIeXkeOnEW9_gfe6yDOo8,12580
+buz-2.23.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+buz-2.23.0.dist-info/RECORD,,