buz 2.23.0__tar.gz → 2.25.0__tar.gz

{buz-2.23.0 → buz-2.25.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: buz
-Version: 2.23.0
+Version: 2.25.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.23.0 → buz-2.25.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "buz"
-version = "2.23.0"
+version = "2.25.0"
 description = "Buz is a set of light, simple and extensible implementations of event, command and query buses."
 readme = "README.md"
 authors = [

{buz-2.23.0 → buz-2.25.0}/src/buz/command/synchronous/command_handler.py

@@ -3,18 +3,11 @@ 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-2.23.0 → buz-2.25.0}/src/buz/kafka/domain/services/kafka_admin_client.py

@@ -2,7 +2,7 @@ from __future__ import annotations
 
 from abc import abstractmethod, ABC
 from datetime import datetime
-from typing import Sequence, Any
+from typing import Sequence, Any, Optional
 
 from buz.kafka.domain.models.create_kafka_topic import CreateKafkaTopic
 from buz.kafka.infrastructure.interfaces.connection_manager import ConnectionManager
@@ -63,6 +63,7 @@ class KafkaAdminClient(ConnectionManager, ABC):
         *,
         topic: str,
         new_number_of_partitions: int,
+        consumer_groups_to_ignore: Optional[set[str]] = None,
     ) -> None:
         pass
 

{buz-2.23.0 → buz-2.25.0}/src/buz/kafka/infrastructure/kafka_python/kafka_python_admin_client.py

@@ -271,6 +271,7 @@ class KafkaPythonAdminClient(KafkaAdminClient):
         *,
         topic: str,
         new_number_of_partitions: int,
+        consumer_groups_to_ignore: Optional[set[str]] = None,
     ) -> None:
         self._logger.info(
             f'Increasing topic "{topic}" partitions: Verifying the new number of partitions "{new_number_of_partitions}"'
@@ -290,6 +291,10 @@ class KafkaPythonAdminClient(KafkaAdminClient):
         self._logger.info(f'Increasing topic "{topic}" partitions: Extracting related consumer groups')
         related_consumer_groups = self.__get_consumer_groups_related_to_a_topic(topic_partitions)
 
+        if consumer_groups_to_ignore:
+            consumer_groups_to_ignore_set = set[str](consumer_groups_to_ignore)
+            related_consumer_groups = related_consumer_groups - consumer_groups_to_ignore_set
+
         self._logger.info(
             f'Increasing topic "{topic}" partitions: The following consumer groups will be updated:"{related_consumer_groups}"'
         )

{buz-2.23.0 → buz-2.25.0}/src/buz/query/synchronous/query_handler.py

@@ -3,19 +3,12 @@ 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-2.23.0/src/buz/wrapper/async_unsafe.py

@@ -1,67 +0,0 @@
-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-2.23.0/src/buz/wrapper/synchronous_only_operation_exception.py

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