buz 2.19.0__tar.gz → 2.20.0__tar.gz

{buz-2.19.0 → buz-2.20.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: buz
-Version: 2.19.0
+Version: 2.20.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.19.0 → buz-2.20.0}/pyproject.toml

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

buz-2.20.0/src/buz/kafka/domain/exceptions/consumer_group_not_found_exception.py

@@ -0,0 +1,6 @@
+from __future__ import annotations
+
+
+class ConsumerGroupNotFoundException(Exception):
+    def __init__(self, consumer_group: str) -> None:
+        super().__init__(f'The consumer group "{consumer_group}" has not been found')

{buz-2.19.0 → buz-2.20.0}/src/buz/kafka/domain/services/kafka_admin_client.py

@@ -65,3 +65,11 @@ class KafkaAdminClient(ConnectionManager, ABC):
         new_number_of_partitions: int,
     ) -> None:
         pass
+
+    @abstractmethod
+    def get_consumer_group_offsets(self, *, consumer_group: str, topic: str) -> dict[int, int]:
+        """
+        Get the committed offsets for a consumer group on a specific topic.
+        Returns a dictionary mapping partition numbers to their committed offsets.
+        """
+        pass

{buz-2.19.0 → buz-2.20.0}/src/buz/kafka/infrastructure/kafka_python/kafka_python_admin_client.py

@@ -12,6 +12,7 @@ from kafka.admin.new_partitions import NewPartitions
 from kafka.errors import TopicAlreadyExistsError
 from kafka.structs import TopicPartition, OffsetAndTimestamp
 
+from buz.kafka.domain.exceptions.consumer_group_not_found_exception import ConsumerGroupNotFoundException
 from buz.kafka.domain.exceptions.not_all_partition_assigned_exception import NotAllPartitionAssignedException
 from buz.kafka.domain.exceptions.not_valid_partition_number_exception import NotValidPartitionNumberException
 from buz.kafka.domain.exceptions.topic_already_created_exception import KafkaTopicsAlreadyCreatedException
@@ -400,3 +401,28 @@ class KafkaPythonAdminClient(KafkaAdminClient):
             related_consumer_groups.add(consumer_group)
 
         return related_consumer_groups
+
+    def get_consumer_group_offsets(self, *, consumer_group: str, topic: str) -> dict[int, int]:
+        self._logger.info(f'Getting consumer group offsets for group "{consumer_group}" on topic "{topic}"')
+
+        if not self.is_topic_created(topic):
+            raise TopicNotFoundException(topic_name=topic)
+
+        cluster_consumer_groups = self.get_cluster_consumer_groups()
+        if consumer_group not in cluster_consumer_groups:
+            raise ConsumerGroupNotFoundException(consumer_group=consumer_group)
+
+        partitions = self.get_number_of_partitions(topic)
+        topic_partitions = [TopicPartition(topic=topic, partition=partition) for partition in range(partitions)]
+
+        offsets_response = self._get_kafka_admin().list_consumer_group_offsets(
+            consumer_group, partitions=topic_partitions
+        )
+
+        # Build the result dictionary, filtering out partitions with no committed offset (-1)
+        result: dict[int, int] = {}
+        for topic_partition, offset_and_metadata in offsets_response.items():
+            if offset_and_metadata.offset >= 0:
+                result[topic_partition.partition] = offset_and_metadata.offset
+
+        return result