feat: add types - topic partition and admin cli (#5)

benbenbang · Aug 11, 2023 · 4c6846e · 4c6846e
1 parent ed314a3
commit 4c6846e
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/confluent_kafka-stubs/cimpl.pyi b/confluent_kafka-stubs/cimpl.pyi
@@ -58,6 +58,9 @@ TIMESTAMP_CREATE_TIME: int
 TIMESTAMP_LOG_APPEND_TIME: int
 TIMESTAMP_NOT_AVAILABLE: int
 
+class _AdminClientImpl:
+    pass
+
 class KafkaError:
     # Class Private Vars
     _ALL_BROKERS_DOWN: int
@@ -302,3 +305,18 @@ class NewPartitions:
             - doc: https://docs.confluent.io/platform/current/clients/confluent-kafka-python/html/index.html#newpartitions
         """
         ...
+
+class TopicPartition:
+    def __init__(
+        self,
+        topic: str,
+        partition: int = -1,
+        offset: int = -1001,
+        metadata: str | None = None,
+        leader_epoch: int | None = None,
+    ) -> None:
+        """
+        Note:
+            - doc: https://docs.confluent.io/platform/current/clients/confluent-kafka-python/html/index.html#topicpartition
+        """
+        ...