python-cqrs 3.0.1__tar.gz → 4.0.0__tar.gz

{python_cqrs-3.0.1 → python_cqrs-4.0.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: python-cqrs
-Version: 3.0.1
+Version: 4.0.0
 Summary: Python CQRS pattern implementation
 Author-email: Vadim Kozyrevskiy <vadikko2@mail.ru>, Dmitry Kutlubaev <kutlubaev00@mail.ru>
 Maintainer-email: Vadim Kozyrevskiy <vadikko2@mail.ru>

{python_cqrs-3.0.1 → python_cqrs-4.0.0}/pyproject.toml

@@ -29,7 +29,7 @@ maintainers = [{name = "Vadim Kozyrevskiy", email = "vadikko2@mail.ru"}]
 name = "python-cqrs"
 readme = "README.md"
 requires-python = ">=3.10"
-version = "3.0.1"
+version = "4.0.0"
 
 [project.optional-dependencies]
 dev = [

python_cqrs-4.0.0/src/cqrs/adapters/amqp.py

@@ -0,0 +1,83 @@
+import asyncio
+import typing
+from functools import partial
+
+from cqrs.adapters import protocol
+
+import aio_pika
+from aio_pika import abc, pool
+
+
+async def connection_pool_factory(url: str) -> abc.AbstractRobustConnection:
+    return await aio_pika.connect_robust(url=url)
+
+
+async def channel_pool_factory(
+    connection_pool: pool.Pool[aio_pika.abc.AbstractConnection],
+) -> aio_pika.abc.AbstractChannel:
+    async with connection_pool.acquire() as connection:
+        return await connection.channel()
+
+
+class AMQPPublisher(protocol.AMQPPublisher):
+    def __init__(self, channel_pool: pool.Pool[aio_pika.abc.AbstractChannel]):
+        self.channel_pool = channel_pool
+
+    async def publish(self, message: abc.AbstractMessage, queue_name: str, exchange_name: str) -> None:
+        async with self.channel_pool.acquire() as channel:
+            queue = await channel.declare_queue(queue_name)
+            exchange = await channel.declare_exchange(exchange_name, type="direct", auto_delete=True)
+            await queue.bind(exchange=exchange, routing_key=queue_name)
+            await exchange.publish(message=message, routing_key=queue_name)
+
+
+class AMQPConsumer(protocol.AMQPConsumer):
+    def __init__(self, channel_pool: pool.Pool[aio_pika.abc.AbstractChannel]):
+        self.channel_pool = channel_pool
+
+    async def consume(
+        self,
+        handler: typing.Callable[[abc.AbstractIncomingMessage], typing.Awaitable[None]],
+        queue_name: str,
+    ) -> None:
+        async with self.channel_pool.acquire() as channel:
+            await channel.set_qos(prefetch_count=1)
+            queue = await channel.declare_queue(queue_name)
+            await queue.consume(handler)
+            await asyncio.Future()
+
+
+def amqp_publisher_factory(
+    url: typing.Text,
+    max_connection_pool_size: int = 2,
+    max_channel_pool_size: int = 10,
+) -> AMQPPublisher:
+    max_connection_pool_size = max_connection_pool_size
+    max_channel_pool_size = max_channel_pool_size
+    connection_pool = pool.Pool(
+        partial(connection_pool_factory, url=url),
+        max_size=max_connection_pool_size,
+    )
+    channel_pool = pool.Pool(
+        partial(channel_pool_factory, connection_pool=connection_pool),
+        max_size=max_channel_pool_size,
+    )
+    return AMQPPublisher(channel_pool=channel_pool)
+
+
+def amqp_consumer_factory(
+    url: typing.Text,
+    max_connection_pool_size: int = 2,
+    max_channel_pool_size: int = 10,
+) -> AMQPConsumer:
+    max_connection_pool_size = max_connection_pool_size
+    max_channel_pool_size = max_channel_pool_size
+    connection_pool = pool.Pool(
+        partial(connection_pool_factory, url=url),
+        max_size=max_connection_pool_size,
+    )
+    channel_pool = pool.Pool(
+        partial(channel_pool_factory, connection_pool=connection_pool),
+        max_size=max_channel_pool_size,
+    )
+    return AMQPConsumer(channel_pool=channel_pool)

{python_cqrs-3.0.1 → python_cqrs-4.0.0}/src/cqrs/adapters/kafka.py

@@ -1,13 +1,16 @@
 import asyncio
 import functools
 import logging
+import ssl
 import typing
 
+from cqrs.adapters import protocol
+from cqrs.serializers import default
+
 import aiokafka
 import retry_async
 from aiokafka import errors
 
-from cqrs.serializers import default
 
 __all__ = (
     "KafkaProducer",
@@ -44,16 +47,7 @@ logger.setLevel(logging.DEBUG)
 Serializer = typing.Callable[[typing.Any], typing.ByteString | None]
 
 
-class _Singleton(type):
-    _instances = {}
-
-    def __call__(cls, *args, **kwargs):
-        if cls not in cls._instances:
-            cls._instances[cls] = super(_Singleton, cls).__call__(*args, **kwargs)
-        return cls._instances[cls]
-
-
-class KafkaProducer(metaclass=_Singleton):
+class KafkaProducer(protocol.KafkaProducer):
     def __init__(
         self,
         producer: aiokafka.AIOKafkaProducer,
@@ -89,6 +83,7 @@ def kafka_producer_factory(
     dsn: typing.Text,
     security_protocol: SecurityProtocol = "PLAINTEXT",
     sasl_mechanism: SaslMechanism = "PLAIN",
+    ssl_context: ssl.SSLContext | None = None,
     retry_count: int = 3,
     retry_delay: int = 1,
     user: typing.Text | None = None,
@@ -105,6 +100,7 @@ def kafka_producer_factory(
         sasl_mechanism=sasl_mechanism,
         sasl_plain_username=user,
         sasl_plain_password=password,
+        ssl_context=ssl_context,
         loop=loop,
     )
     return KafkaProducer(

python_cqrs-4.0.0/src/cqrs/adapters/protocol.py

@@ -0,0 +1,28 @@
+import typing
+
+import aio_pika
+
+
+class KafkaProducer(typing.Protocol):
+    async def produce(
+        self,
+        topic: typing.Text,
+        message: typing.Any,
+    ) -> None: ...
+
+
+class AMQPPublisher(typing.Protocol):
+    async def publish(
+        self,
+        message: aio_pika.abc.AbstractMessage,
+        queue_name: str,
+        exchange_name: str,
+    ) -> None: ...
+
+
+class AMQPConsumer(typing.Protocol):
+    async def consume(
+        self,
+        handler: typing.Callable[[aio_pika.abc.AbstractIncomingMessage], typing.Awaitable[None]],
+        queue_name: str,
+    ) -> None: ...

{python_cqrs-3.0.1 → python_cqrs-4.0.0}/src/cqrs/message_brokers/amqp.py

@@ -3,13 +3,13 @@ import logging
 import aio_pika
 import orjson
 
-from cqrs.adapters import amqp
+from cqrs.adapters import protocol as adapters_protocol
 from cqrs.message_brokers import protocol
 
 
 class AMQPMessageBroker(protocol.MessageBroker):
-    def __init__(self, dsn: str, exchange_name: str, pika_log_level: str = "ERROR"):
-        self.publisher = amqp.AMQPPublisher(url=dsn)
+    def __init__(self, publisher: adapters_protocol.AMQPPublisher, exchange_name: str, pika_log_level: str = "ERROR"):
+        self.publisher = publisher
         self.exchange_name = exchange_name
         logging.getLogger("aiormq").setLevel(pika_log_level)
         logging.getLogger("aio_pika").setLevel(pika_log_level)

{python_cqrs-3.0.1 → python_cqrs-4.0.0}/src/cqrs/message_brokers/kafka.py

@@ -1,14 +1,14 @@
 import logging
 import typing
 
-from cqrs.adapters import kafka
+from cqrs.adapters import protocol as adapters_protocol
 from cqrs.message_brokers import protocol
 
 
 class KafkaMessageBroker(protocol.MessageBroker):
     def __init__(
         self,
-        producer: kafka.KafkaProducer,
+        producer: adapters_protocol.KafkaProducer,
         aiokafka_log_level: typing.Text = "ERROR",
     ):
         self._producer = producer

{python_cqrs-3.0.1 → python_cqrs-4.0.0}/src/cqrs/producer.py

@@ -37,7 +37,7 @@ class EventProducer:
                     message_name=event.event.event_name,
                     message_id=event.event.event_id,
                     topic=event.topic,
-                    payload=event.event,
+                    payload=event.event.model_dump(),
                 ),
             )
         except Exception as error:

{python_cqrs-3.0.1 → python_cqrs-4.0.0}/src/python_cqrs.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: python-cqrs
-Version: 3.0.1
+Version: 4.0.0
 Summary: Python CQRS pattern implementation
 Author-email: Vadim Kozyrevskiy <vadikko2@mail.ru>, Dmitry Kutlubaev <kutlubaev00@mail.ru>
 Maintainer-email: Vadim Kozyrevskiy <vadikko2@mail.ru>

{python_cqrs-3.0.1 → python_cqrs-4.0.0}/src/python_cqrs.egg-info/SOURCES.txt

@@ -8,6 +8,7 @@ src/cqrs/response.py
 src/cqrs/adapters/__init__.py
 src/cqrs/adapters/amqp.py
 src/cqrs/adapters/kafka.py
+src/cqrs/adapters/protocol.py
 src/cqrs/compressors/__init__.py
 src/cqrs/compressors/protocol.py
 src/cqrs/compressors/zlib.py

python_cqrs-3.0.1/src/cqrs/adapters/amqp.py

@@ -1,63 +0,0 @@
-import asyncio
-import typing
-from functools import partial
-
-import aio_pika
-from aio_pika import abc, pool
-
-
-async def connection_pool_factory(url: str) -> abc.AbstractRobustConnection:
-    return await aio_pika.connect_robust(url=url)
-
-
-async def channel_pool_factory(connection_pool: pool.Pool) -> aio_pika.Channel:
-    async with connection_pool.acquire() as connection:
-        return await connection.channel()
-
-
-class AMQPPublisher:
-    def __init__(self, url: str, max_connection_pool_size=2, max_channel_pool_size=10):
-        self.url = url
-        self.max_connection_pool_size = max_connection_pool_size
-        self.max_channel_pool_size = max_channel_pool_size
-        self.connection_pool: pool.Pool = pool.Pool(
-            partial(connection_pool_factory, url=url),
-            max_size=self.max_connection_pool_size,
-        )
-        self.channel_pool: pool.Pool = pool.Pool(
-            partial(channel_pool_factory, connection_pool=self.connection_pool),
-            max_size=self.max_channel_pool_size,
-        )
-
-    async def publish(self, message: abc.AbstractMessage, queue_name: str, exchange_name: str) -> None:
-        async with self.channel_pool.acquire() as channel:
-            queue: aio_pika.Queue = await channel.declare_queue(queue_name)
-            exchange: aio_pika.Exchange = await channel.declare_exchange(exchange_name, type="direct", auto_delete=True)
-            await queue.bind(exchange=exchange, routing_key=queue_name)
-            await exchange.publish(message=message, routing_key=queue_name)
-
-
-class AMQPConsumer:
-    def __init__(self, url: str, max_connection_pool_size=2, max_channel_pool_size=10):
-        self.url = url
-        self.max_connection_pool_size = max_connection_pool_size
-        self.max_channel_pool_size = max_channel_pool_size
-        self.connection_pool: pool.Pool = pool.Pool(
-            partial(connection_pool_factory, url=url),
-            max_size=self.max_connection_pool_size,
-        )
-        self.channel_pool: pool.Pool = pool.Pool(
-            partial(channel_pool_factory, connection_pool=self.connection_pool),
-            max_size=self.max_channel_pool_size,
-        )
-
-    async def consume(
-        self,
-        handler: typing.Callable[[abc.AbstractIncomingMessage], typing.Awaitable[None]],
-        queue_name: str,
-    ) -> None:
-        async with self.channel_pool.acquire() as channel:
-            await channel.set_qos(prefetch_count=1)
-            queue = await channel.declare_queue(queue_name)
-            await queue._consume(handler)
-            await asyncio.Future()