kafka-python 3.0.0__py3-none-any.whl

kafka/protocol/metadata/find_coordinator.pyi

@@ -0,0 +1,105 @@
+# Generated by generate_stubs.py (Python 3.14)
+import uuid
+from typing import Any, Self
+
+from enum import IntEnum
+from kafka.protocol.api_message import ApiMessage
+from kafka.protocol.data_container import DataContainer
+
+__all__ = ['CoordinatorType', 'FindCoordinatorRequest', 'FindCoordinatorResponse']
+
+class CoordinatorType(EnumHelper, IntEnum):
+    GROUP: int
+    TRANSACTION: int
+    SHARE: int
+
+class FindCoordinatorRequest(ApiMessage):
+    key: str
+    key_type: int
+    coordinator_keys: list[str]
+    def __init__(
+        self,
+        *args: Any,
+        key: str = ...,
+        key_type: int = ...,
+        coordinator_keys: 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 FindCoordinatorResponse(ApiMessage):
+    class Coordinator(DataContainer):
+        key: str
+        node_id: int
+        host: str
+        port: int
+        error_code: int
+        error_message: str | None
+        def __init__(
+            self,
+            *args: Any,
+            key: str = ...,
+            node_id: int = ...,
+            host: str = ...,
+            port: int = ...,
+            error_code: int = ...,
+            error_message: str | None = ...,
+            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
+    error_message: str | None
+    node_id: int
+    host: str
+    port: int
+    coordinators: list[Coordinator]
+    def __init__(
+        self,
+        *args: Any,
+        throttle_time_ms: int = ...,
+        error_code: int = ...,
+        error_message: str | None = ...,
+        node_id: int = ...,
+        host: str = ...,
+        port: int = ...,
+        coordinators: list[Coordinator] = ...,
+        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/metadata/metadata.py

@@ -0,0 +1,34 @@
+from __future__ import annotations
+
+from ..api_message import ApiMessage
+
+from kafka.util import classproperty
+
+
+class MetadataRequest(ApiMessage):
+    ALL_TOPICS = None
+    NO_TOPICS = []
+
+    def encode(self, version=None, header=False, framed=False):
+        # Fixup v0 ALL_TOPICS => []
+        if version == 0 or self.API_VERSION == 0:
+            if self.topics is None: # pylint: disable=E0203
+                self.topics = []
+        return super().encode(version=version, header=header, framed=framed)
+
+
+class MetadataResponse(ApiMessage):
+    @classmethod
+    def json_patch(cls, json):
+        # cluster_authorized_operations
+        json['fields'][5]['name'] = 'authorized_operations'
+        json['fields'][5]['type'] = 'bitfield'
+        # topic_authorized_operations
+        json['fields'][4]['fields'][5]['name'] = 'authorized_operations'
+        json['fields'][4]['fields'][5]['type'] = 'bitfield'
+        return json
+
+
+__all__ = [
+    'MetadataRequest', 'MetadataResponse',
+]

kafka/protocol/metadata/metadata.pyi

@@ -0,0 +1,160 @@
+# 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.data_container import DataContainer
+
+__all__ = ['MetadataRequest', 'MetadataResponse']
+
+class MetadataRequest(ApiMessage):
+    class MetadataRequestTopic(DataContainer):
+        topic_id: uuid.UUID
+        name: str | None
+        def __init__(
+            self,
+            *args: Any,
+            topic_id: uuid.UUID = ...,
+            name: str | None = ...,
+            version: int | None = None,
+            **kwargs: Any,
+        ) -> None: ...
+        @property
+        def version(self) -> int | None: ...
+        def to_dict(self, meta: bool = False, json: bool = True) -> dict: ...
+
+    topics: list[MetadataRequestTopic] | None
+    allow_auto_topic_creation: bool
+    include_cluster_authorized_operations: bool
+    include_topic_authorized_operations: bool
+    def __init__(
+        self,
+        *args: Any,
+        topics: list[MetadataRequestTopic] | None = ...,
+        allow_auto_topic_creation: bool = ...,
+        include_cluster_authorized_operations: bool = ...,
+        include_topic_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 MetadataResponse(ApiMessage):
+    class MetadataResponseBroker(DataContainer):
+        node_id: int
+        host: str
+        port: int
+        rack: str | None
+        def __init__(
+            self,
+            *args: Any,
+            node_id: int = ...,
+            host: str = ...,
+            port: int = ...,
+            rack: str | None = ...,
+            version: int | None = None,
+            **kwargs: Any,
+        ) -> None: ...
+        @property
+        def version(self) -> int | None: ...
+        def to_dict(self, meta: bool = False, json: bool = True) -> dict: ...
+
+    class MetadataResponseTopic(DataContainer):
+        class MetadataResponsePartition(DataContainer):
+            error_code: int
+            partition_index: int
+            leader_id: int
+            leader_epoch: int
+            replica_nodes: list[int]
+            isr_nodes: list[int]
+            offline_replicas: list[int]
+            def __init__(
+                self,
+                *args: Any,
+                error_code: int = ...,
+                partition_index: int = ...,
+                leader_id: int = ...,
+                leader_epoch: int = ...,
+                replica_nodes: list[int] = ...,
+                isr_nodes: list[int] = ...,
+                offline_replicas: list[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: ...
+
+        error_code: int
+        name: str | None
+        topic_id: uuid.UUID
+        is_internal: bool
+        partitions: list[MetadataResponsePartition]
+        authorized_operations: set[int]
+        def __init__(
+            self,
+            *args: Any,
+            error_code: int = ...,
+            name: str | None = ...,
+            topic_id: uuid.UUID = ...,
+            is_internal: bool = ...,
+            partitions: list[MetadataResponsePartition] = ...,
+            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
+    brokers: list[MetadataResponseBroker]
+    cluster_id: str | None
+    controller_id: int
+    topics: list[MetadataResponseTopic]
+    authorized_operations: set[int]
+    error_code: int
+    def __init__(
+        self,
+        *args: Any,
+        throttle_time_ms: int = ...,
+        brokers: list[MetadataResponseBroker] = ...,
+        cluster_id: str | None = ...,
+        controller_id: int = ...,
+        topics: list[MetadataResponseTopic] = ...,
+        authorized_operations: set[int] = ...,
+        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: ...
+    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/old/abstract.py

@@ -0,0 +1,17 @@
+from abc import ABC, abstractmethod
+
+
+class AbstractType(ABC):
+    @classmethod
+    @abstractmethod
+    def encode(cls, value): # pylint: disable=no-self-argument
+        pass
+
+    @classmethod
+    @abstractmethod
+    def decode(cls, data): # pylint: disable=no-self-argument
+        pass
+
+    @classmethod
+    def repr(cls, value):
+        return repr(value)

kafka/protocol/old/add_offsets_to_txn.py

@@ -0,0 +1,54 @@
+from .api import Request, Response
+from .types import Int16, Int32, Int64, Schema, String
+
+
+class AddOffsetsToTxnResponse_v0(Response):
+    API_KEY = 25
+    API_VERSION = 0
+    SCHEMA = Schema(
+        ('throttle_time_ms', Int32),
+        ('error_code', Int16),
+    )
+
+
+class AddOffsetsToTxnResponse_v1(Response):
+    API_KEY = 25
+    API_VERSION = 1
+    SCHEMA = AddOffsetsToTxnResponse_v0.SCHEMA
+
+
+class AddOffsetsToTxnResponse_v2(Response):
+    API_KEY = 25
+    API_VERSION = 2
+    SCHEMA = AddOffsetsToTxnResponse_v1.SCHEMA
+
+
+class AddOffsetsToTxnRequest_v0(Request):
+    API_KEY = 25
+    API_VERSION = 0
+    SCHEMA = Schema(
+        ('transactional_id', String('utf-8')),
+        ('producer_id', Int64),
+        ('producer_epoch', Int16),
+        ('group_id', String('utf-8')),
+    )
+
+
+class AddOffsetsToTxnRequest_v1(Request):
+    API_KEY = 25
+    API_VERSION = 1
+    SCHEMA = AddOffsetsToTxnRequest_v0.SCHEMA
+
+
+class AddOffsetsToTxnRequest_v2(Request):
+    API_KEY = 25
+    API_VERSION = 2
+    SCHEMA = AddOffsetsToTxnRequest_v1.SCHEMA
+
+
+AddOffsetsToTxnRequest = [
+    AddOffsetsToTxnRequest_v0, AddOffsetsToTxnRequest_v1, AddOffsetsToTxnRequest_v2,
+]
+AddOffsetsToTxnResponse = [
+    AddOffsetsToTxnResponse_v0, AddOffsetsToTxnResponse_v1, AddOffsetsToTxnResponse_v2,
+]

kafka/protocol/old/add_partitions_to_txn.py

@@ -0,0 +1,71 @@
+from .api import Request, Response
+from .types import Array, Int16, Int32, Int64, Schema, String
+
+
+class AddPartitionsToTxnResponse_v0(Response):
+    API_KEY = 24
+    API_VERSION = 0
+    SCHEMA = Schema(
+        ('throttle_time_ms', Int32),
+        ('results_by_topic_v3_and_below', Array(
+            ('name', String('utf-8')),
+            ('results_by_partition', Array(
+                ('partition_index', Int32),
+                ('partition_error_code', Int16))))))
+    ALIASES = {
+        'results': 'results_by_topic_v3_and_below',
+    }
+
+
+class AddPartitionsToTxnResponse_v1(Response):
+    API_KEY = 24
+    API_VERSION = 1
+    SCHEMA = AddPartitionsToTxnResponse_v0.SCHEMA
+    ALIASES = AddPartitionsToTxnResponse_v0.ALIASES
+
+
+class AddPartitionsToTxnResponse_v2(Response):
+    API_KEY = 24
+    API_VERSION = 2
+    SCHEMA = AddPartitionsToTxnResponse_v1.SCHEMA
+    ALIASES = AddPartitionsToTxnResponse_v1.ALIASES
+
+
+class AddPartitionsToTxnRequest_v0(Request):
+    API_KEY = 24
+    API_VERSION = 0
+    SCHEMA = Schema(
+        ('v3_and_below_transactional_id', String('utf-8')),
+        ('v3_and_below_producer_id', Int64),
+        ('v3_and_below_producer_epoch', Int16),
+        ('v3_and_below_topics', Array(
+            ('name', String('utf-8')),
+            ('partitions', Array(Int32)))))
+    ALIASES = {
+        'transactional_id': 'v3_and_below_transactional_id',
+        'producer_id': 'v3_and_below_producer_id',
+        'producer_epoch': 'v3_and_below_producer_epoch',
+        'topics': 'v3_and_below_topics',
+    }
+
+
+class AddPartitionsToTxnRequest_v1(Request):
+    API_KEY = 24
+    API_VERSION = 1
+    SCHEMA = AddPartitionsToTxnRequest_v0.SCHEMA
+    ALIASES = AddPartitionsToTxnRequest_v0.ALIASES
+
+
+class AddPartitionsToTxnRequest_v2(Request):
+    API_KEY = 24
+    API_VERSION = 2
+    SCHEMA = AddPartitionsToTxnRequest_v1.SCHEMA
+    ALIASES = AddPartitionsToTxnRequest_v1.ALIASES
+
+
+AddPartitionsToTxnRequest = [
+    AddPartitionsToTxnRequest_v0, AddPartitionsToTxnRequest_v1, AddPartitionsToTxnRequest_v2,
+]
+AddPartitionsToTxnResponse = [
+    AddPartitionsToTxnResponse_v0, AddPartitionsToTxnResponse_v1, AddPartitionsToTxnResponse_v2,
+]