message-db-py 0.3.0__tar.gz → 0.3.2__tar.gz

{message_db_py-0.3.0 → message_db_py-0.3.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: message-db-py
-Version: 0.3.0
+Version: 0.3.2
 Summary: The Python interface to the MessageDB Event Store and Message Store
 License: MIT
 License-File: LICENSE

{message_db_py-0.3.0 → message_db_py-0.3.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "message-db-py"
-version = "0.3.0"
+version = "0.3.2"
 description = "The Python interface to the MessageDB Event Store and Message Store"
 authors = ["Subhash Bhushan <subhash.bhushan@gmail.com>"]
 license = "MIT"

{message_db_py-0.3.0 → message_db_py-0.3.2}/src/message_db/__init__.py

@@ -1,3 +1,3 @@
-__version__ = "0.3.0"
+__version__ = "0.3.2"
 
 # Re-export client class

{message_db_py-0.3.0 → message_db_py-0.3.2}/src/message_db/client.py

@@ -283,6 +283,51 @@ class MessageDB:
             consumer_group_size=consumer_group_size,
         )
 
+    def stream_identifiers(self, category_name: str) -> List[str]:
+        """Return all unique aggregate identifiers for a stream category.
+
+        Extracts distinct identifiers from stream names matching the given
+        category, excluding snapshot streams.  Stream names follow the pattern
+        ``{category}-{identifier}``; snapshot streams use
+        ``{category}:snapshot-{identifier}``.
+
+        Args:
+            category_name: The stream category (must not contain a hyphen).
+
+        Returns:
+            Sorted list of unique aggregate identifiers.
+
+        Raises:
+            ValueError: If *category_name* contains a hyphen.
+        """
+        if "-" in category_name:
+            raise ValueError(f"{category_name} is not a category")
+
+        conn = self.connection_pool.get_connection()
+        cursor = conn.cursor()
+
+        cursor.execute(
+            """
+            SELECT DISTINCT
+                substring(stream_name from position('-' in stream_name) + 1)
+            FROM message_store.messages
+            WHERE stream_name LIKE %(pattern)s
+              AND stream_name NOT LIKE %(snapshot_pattern)s
+            ORDER BY 1
+            """,
+            {
+                "pattern": f"{category_name}-%",
+                "snapshot_pattern": f"{category_name}:snapshot-%",
+            },
+        )
+        identifiers = [row[0] for row in cursor.fetchall() if row[0]]
+
+        conn.commit()
+        cursor.close()
+        self.connection_pool.release(conn)
+
+        return identifiers
+
     def read_last_message(self, stream_name: str) -> Dict[str, Any] | None:
         """Read the last message from a stream."""
         conn = self.connection_pool.get_connection()