kafka-python 3.0.0__py3-none-any.whl

kafka/protocol/admin/groups.pyi

@@ -0,0 +1,261 @@
+# Generated by generate_stubs.py (Python 3.14)
+import uuid
+from typing import Any, Self
+
+from kafka.protocol.api_message import ApiMessage
+from kafka.protocol.api_data import ApiData
+from kafka.protocol.data_container import DataContainer
+
+__all__ = ['DescribeGroupsRequest', 'DescribeGroupsResponse', 'ListGroupsRequest', 'ListGroupsResponse', 'DeleteGroupsRequest', 'DeleteGroupsResponse']
+
+class DescribeGroupsRequest(ApiMessage):
+    groups: list[str]
+    include_authorized_operations: bool
+    def __init__(
+        self,
+        *args: Any,
+        groups: list[str] = ...,
+        include_authorized_operations: bool = ...,
+        version: int | None = None,
+        **kwargs: Any,
+    ) -> None: ...
+    @property
+    def version(self) -> int | None: ...
+    def to_dict(self, meta: bool = False, json: bool = True) -> dict: ...
+    name: str
+    type: str
+    API_KEY: int
+    API_VERSION: int
+    valid_versions: tuple[int, int]
+    min_version: int
+    max_version: int
+    @property
+    def header(self) -> Any: ...
+    @classmethod
+    def is_request(cls) -> bool: ...
+    def expect_response(self) -> bool: ...
+    def with_header(self, correlation_id: int = 0, client_id: str = "kafka-python") -> None: ...
+
+class DescribeGroupsResponse(ApiMessage):
+    class DescribedGroup(DataContainer):
+        class DescribedGroupMember(DataContainer):
+            member_id: str
+            group_instance_id: str | None
+            client_id: str
+            client_host: str
+            member_metadata: bytes | ApiData
+            member_assignment: bytes | ApiData
+            def __init__(
+                self,
+                *args: Any,
+                member_id: str = ...,
+                group_instance_id: str | None = ...,
+                client_id: str = ...,
+                client_host: str = ...,
+                member_metadata: bytes | ApiData = ...,
+                member_assignment: bytes | ApiData = ...,
+                version: int | None = None,
+                **kwargs: Any,
+            ) -> None: ...
+            @property
+            def version(self) -> int | None: ...
+            def to_dict(self, meta: bool = False, json: bool = True) -> dict: ...
+
+        error_code: int
+        error_message: str | None
+        group_id: str
+        group_state: str
+        protocol_type: str
+        protocol_data: str
+        members: list[DescribedGroupMember]
+        authorized_operations: set[int]
+        def __init__(
+            self,
+            *args: Any,
+            error_code: int = ...,
+            error_message: str | None = ...,
+            group_id: str = ...,
+            group_state: str = ...,
+            protocol_type: str = ...,
+            protocol_data: str = ...,
+            members: list[DescribedGroupMember] = ...,
+            authorized_operations: set[int] = ...,
+            version: int | None = None,
+            **kwargs: Any,
+        ) -> None: ...
+        @property
+        def version(self) -> int | None: ...
+        def to_dict(self, meta: bool = False, json: bool = True) -> dict: ...
+
+    throttle_time_ms: int
+    groups: list[DescribedGroup]
+    def __init__(
+        self,
+        *args: Any,
+        throttle_time_ms: int = ...,
+        groups: list[DescribedGroup] = ...,
+        version: int | None = None,
+        **kwargs: Any,
+    ) -> None: ...
+    @property
+    def version(self) -> int | None: ...
+    def to_dict(self, meta: bool = False, json: bool = True) -> dict: ...
+    name: str
+    type: str
+    API_KEY: int
+    API_VERSION: int
+    valid_versions: tuple[int, int]
+    min_version: int
+    max_version: int
+    @property
+    def header(self) -> Any: ...
+    @classmethod
+    def is_request(cls) -> bool: ...
+    def expect_response(self) -> bool: ...
+    def with_header(self, correlation_id: int = 0, client_id: str = "kafka-python") -> None: ...
+
+class ListGroupsRequest(ApiMessage):
+    states_filter: list[str]
+    types_filter: list[str]
+    def __init__(
+        self,
+        *args: Any,
+        states_filter: list[str] = ...,
+        types_filter: list[str] = ...,
+        version: int | None = None,
+        **kwargs: Any,
+    ) -> None: ...
+    @property
+    def version(self) -> int | None: ...
+    def to_dict(self, meta: bool = False, json: bool = True) -> dict: ...
+    name: str
+    type: str
+    API_KEY: int
+    API_VERSION: int
+    valid_versions: tuple[int, int]
+    min_version: int
+    max_version: int
+    @property
+    def header(self) -> Any: ...
+    @classmethod
+    def is_request(cls) -> bool: ...
+    def expect_response(self) -> bool: ...
+    def with_header(self, correlation_id: int = 0, client_id: str = "kafka-python") -> None: ...
+
+class ListGroupsResponse(ApiMessage):
+    class ListedGroup(DataContainer):
+        group_id: str
+        protocol_type: str
+        group_state: str
+        group_type: str
+        def __init__(
+            self,
+            *args: Any,
+            group_id: str = ...,
+            protocol_type: str = ...,
+            group_state: str = ...,
+            group_type: str = ...,
+            version: int | None = None,
+            **kwargs: Any,
+        ) -> None: ...
+        @property
+        def version(self) -> int | None: ...
+        def to_dict(self, meta: bool = False, json: bool = True) -> dict: ...
+
+    throttle_time_ms: int
+    error_code: int
+    groups: list[ListedGroup]
+    def __init__(
+        self,
+        *args: Any,
+        throttle_time_ms: int = ...,
+        error_code: int = ...,
+        groups: list[ListedGroup] = ...,
+        version: int | None = None,
+        **kwargs: Any,
+    ) -> None: ...
+    @property
+    def version(self) -> int | None: ...
+    def to_dict(self, meta: bool = False, json: bool = True) -> dict: ...
+    name: str
+    type: str
+    API_KEY: int
+    API_VERSION: int
+    valid_versions: tuple[int, int]
+    min_version: int
+    max_version: int
+    @property
+    def header(self) -> Any: ...
+    @classmethod
+    def is_request(cls) -> bool: ...
+    def expect_response(self) -> bool: ...
+    def with_header(self, correlation_id: int = 0, client_id: str = "kafka-python") -> None: ...
+
+class DeleteGroupsRequest(ApiMessage):
+    groups_names: list[str]
+    def __init__(
+        self,
+        *args: Any,
+        groups_names: list[str] = ...,
+        version: int | None = None,
+        **kwargs: Any,
+    ) -> None: ...
+    @property
+    def version(self) -> int | None: ...
+    def to_dict(self, meta: bool = False, json: bool = True) -> dict: ...
+    name: str
+    type: str
+    API_KEY: int
+    API_VERSION: int
+    valid_versions: tuple[int, int]
+    min_version: int
+    max_version: int
+    @property
+    def header(self) -> Any: ...
+    @classmethod
+    def is_request(cls) -> bool: ...
+    def expect_response(self) -> bool: ...
+    def with_header(self, correlation_id: int = 0, client_id: str = "kafka-python") -> None: ...
+
+class DeleteGroupsResponse(ApiMessage):
+    class DeletableGroupResult(DataContainer):
+        group_id: str
+        error_code: int
+        def __init__(
+            self,
+            *args: Any,
+            group_id: str = ...,
+            error_code: int = ...,
+            version: int | None = None,
+            **kwargs: Any,
+        ) -> None: ...
+        @property
+        def version(self) -> int | None: ...
+        def to_dict(self, meta: bool = False, json: bool = True) -> dict: ...
+
+    throttle_time_ms: int
+    results: list[DeletableGroupResult]
+    def __init__(
+        self,
+        *args: Any,
+        throttle_time_ms: int = ...,
+        results: list[DeletableGroupResult] = ...,
+        version: int | None = None,
+        **kwargs: Any,
+    ) -> None: ...
+    @property
+    def version(self) -> int | None: ...
+    def to_dict(self, meta: bool = False, json: bool = True) -> dict: ...
+    name: str
+    type: str
+    API_KEY: int
+    API_VERSION: int
+    valid_versions: tuple[int, int]
+    min_version: int
+    max_version: int
+    @property
+    def header(self) -> Any: ...
+    @classmethod
+    def is_request(cls) -> bool: ...
+    def expect_response(self) -> bool: ...
+    def with_header(self, correlation_id: int = 0, client_id: str = "kafka-python") -> None: ...

kafka/protocol/admin/topics.py

@@ -0,0 +1,53 @@
+from enum import IntEnum
+
+from ..api_message import ApiMessage
+
+
+class CreateTopicsRequest(ApiMessage): pass
+class CreateTopicsResponse(ApiMessage): pass
+
+class DeleteTopicsRequest(ApiMessage):
+    def encode(self, version=None, header=False, framed=False):
+        # convert topics => topic_names for v0-v5
+        version = self.API_VERSION if version is None else version
+        if version is not None and version <= 5:
+            if self.topics and not self.topic_names: # pylint: disable=E0203
+                self.topic_names = [topic.name for topic in self.topics]
+        return super().encode(version=version, header=header, framed=framed)
+
+class DeleteTopicsResponse(ApiMessage): pass
+
+class CreatePartitionsRequest(ApiMessage): pass
+class CreatePartitionsResponse(ApiMessage): pass
+
+class AlterPartitionReassignmentsRequest(ApiMessage): pass
+class AlterPartitionReassignmentsResponse(ApiMessage): pass
+
+class ListPartitionReassignmentsRequest(ApiMessage): pass
+class ListPartitionReassignmentsResponse(ApiMessage): pass
+
+class DescribeTopicPartitionsRequest(ApiMessage): pass
+class DescribeTopicPartitionsResponse(ApiMessage): pass
+
+class DeleteRecordsRequest(ApiMessage): pass
+class DeleteRecordsResponse(ApiMessage): pass
+
+class ElectLeadersRequest(ApiMessage): pass
+class ElectLeadersResponse(ApiMessage): pass
+
+class ElectionType(IntEnum):
+    """Leader election type"""
+    PREFERRED = 0
+    UNCLEAN = 1
+
+
+__all__ = [
+    'CreateTopicsRequest', 'CreateTopicsResponse',
+    'DeleteTopicsRequest', 'DeleteTopicsResponse',
+    'CreatePartitionsRequest', 'CreatePartitionsResponse',
+    'AlterPartitionReassignmentsRequest', 'AlterPartitionReassignmentsResponse',
+    'ListPartitionReassignmentsRequest', 'ListPartitionReassignmentsResponse',
+    'DescribeTopicPartitionsRequest', 'DescribeTopicPartitionsResponse',
+    'DeleteRecordsRequest', 'DeleteRecordsResponse',
+    'ElectLeadersRequest', 'ElectLeadersResponse', 'ElectionType',
+]