types-kafka-python 2.3.0.20260523__py3-none-any.whl

kafka-stubs/protocol/list_offsets.pyi

@@ -0,0 +1,85 @@
+from _typeshed import Incomplete
+
+from kafka.protocol.api import Request, Response
+
+UNKNOWN_OFFSET: int
+
+class OffsetResetStrategy:
+    LATEST: int
+    EARLIEST: int
+    NONE: int
+
+class ListOffsetsResponse_v0(Response):
+    API_KEY: int
+    API_VERSION: int
+    SCHEMA: Incomplete
+
+class ListOffsetsResponse_v1(Response):
+    API_KEY: int
+    API_VERSION: int
+    SCHEMA: Incomplete
+
+class ListOffsetsResponse_v2(Response):
+    API_KEY: int
+    API_VERSION: int
+    SCHEMA: Incomplete
+
+class ListOffsetsResponse_v3(Response):
+    API_KEY: int
+    API_VERSION: int
+    SCHEMA: Incomplete
+
+class ListOffsetsResponse_v4(Response):
+    API_KEY: int
+    API_VERSION: int
+    SCHEMA: Incomplete
+
+class ListOffsetsResponse_v5(Response):
+    API_KEY: int
+    API_VERSION: int
+    SCHEMA: Incomplete
+
+class ListOffsetsRequest_v0(Request):
+    API_KEY: int
+    API_VERSION: int
+    RESPONSE_TYPE = ListOffsetsResponse_v0
+    SCHEMA: Incomplete
+    DEFAULTS: Incomplete
+
+class ListOffsetsRequest_v1(Request):
+    API_KEY: int
+    API_VERSION: int
+    RESPONSE_TYPE = ListOffsetsResponse_v1
+    SCHEMA: Incomplete
+    DEFAULTS: Incomplete
+
+class ListOffsetsRequest_v2(Request):
+    API_KEY: int
+    API_VERSION: int
+    RESPONSE_TYPE = ListOffsetsResponse_v2
+    SCHEMA: Incomplete
+    DEFAULTS: Incomplete
+
+class ListOffsetsRequest_v3(Request):
+    API_KEY: int
+    API_VERSION: int
+    RESPONSE_TYPE = ListOffsetsResponse_v3
+    SCHEMA: Incomplete
+    DEFAULTS: Incomplete
+
+class ListOffsetsRequest_v4(Request):
+    API_KEY: int
+    API_VERSION: int
+    RESPONSE_TYPE = ListOffsetsResponse_v4
+    SCHEMA: Incomplete
+    DEFAULTS: Incomplete
+
+class ListOffsetsRequest_v5(Request):
+    API_KEY: int
+    API_VERSION: int
+    RESPONSE_TYPE = ListOffsetsResponse_v5
+    SCHEMA: Incomplete
+    DEFAULTS: Incomplete
+
+ListOffsetsRequest: Incomplete
+ListOffsetsResponse: Incomplete

kafka-stubs/protocol/message.pyi

@@ -0,0 +1,43 @@
+from _typeshed import Incomplete
+
+from kafka.protocol.struct import Struct
+from kafka.protocol.types import AbstractType
+
+class Message(Struct):
+    SCHEMAS: Incomplete
+    SCHEMA: Incomplete
+    CODEC_MASK: int
+    CODEC_GZIP: int
+    CODEC_SNAPPY: int
+    CODEC_LZ4: int
+    CODEC_ZSTD: int
+    TIMESTAMP_TYPE_MASK: int
+    HEADER_SIZE: int
+    timestamp: Incomplete
+    crc: Incomplete
+    magic: Incomplete
+    attributes: Incomplete
+    key: Incomplete
+    value: Incomplete
+    encode: Incomplete
+    def __init__(self, value, key=None, magic: int = 0, attributes: int = 0, crc: int = 0, timestamp=None) -> None: ...
+    @property
+    def timestamp_type(self): ...
+    @classmethod
+    def decode(cls, data): ...
+    def validate_crc(self): ...
+    def is_compressed(self): ...
+    def decompress(self): ...
+    def __hash__(self): ...
+
+class PartialMessage(bytes): ...
+
+class MessageSet(AbstractType):
+    ITEM: Incomplete
+    HEADER_SIZE: int
+    @classmethod
+    def encode(cls, items, prepend_size: bool = True): ...
+    @classmethod
+    def decode(cls, data, bytes_to_read=None): ...
+    @classmethod
+    def repr(cls, messages): ...

kafka-stubs/protocol/metadata.pyi

@@ -0,0 +1,123 @@
+from _typeshed import Incomplete
+
+from kafka.protocol.api import Request, Response
+
+class MetadataResponse_v0(Response):
+    API_KEY: int
+    API_VERSION: int
+    SCHEMA: Incomplete
+
+class MetadataResponse_v1(Response):
+    API_KEY: int
+    API_VERSION: int
+    SCHEMA: Incomplete
+
+class MetadataResponse_v2(Response):
+    API_KEY: int
+    API_VERSION: int
+    SCHEMA: Incomplete
+
+class MetadataResponse_v3(Response):
+    API_KEY: int
+    API_VERSION: int
+    SCHEMA: Incomplete
+
+class MetadataResponse_v4(Response):
+    API_KEY: int
+    API_VERSION: int
+    SCHEMA: Incomplete
+
+class MetadataResponse_v5(Response):
+    API_KEY: int
+    API_VERSION: int
+    SCHEMA: Incomplete
+
+class MetadataResponse_v6(Response):
+    API_KEY: int
+    API_VERSION: int
+    SCHEMA: Incomplete
+
+class MetadataResponse_v7(Response):
+    API_KEY: int
+    API_VERSION: int
+    SCHEMA: Incomplete
+
+class MetadataResponse_v8(Response):
+    API_KEY: int
+    API_VERSION: int
+    SCHEMA: Incomplete
+
+class MetadataRequest_v0(Request):
+    API_KEY: int
+    API_VERSION: int
+    RESPONSE_TYPE = MetadataResponse_v0
+    SCHEMA: Incomplete
+    ALL_TOPICS: Incomplete
+    NO_TOPICS: Incomplete
+
+class MetadataRequest_v1(Request):
+    API_KEY: int
+    API_VERSION: int
+    RESPONSE_TYPE = MetadataResponse_v1
+    SCHEMA: Incomplete
+    ALL_TOPICS: Incomplete
+    NO_TOPICS: Incomplete
+
+class MetadataRequest_v2(Request):
+    API_KEY: int
+    API_VERSION: int
+    RESPONSE_TYPE = MetadataResponse_v2
+    SCHEMA: Incomplete
+    ALL_TOPICS: Incomplete
+    NO_TOPICS: Incomplete
+
+class MetadataRequest_v3(Request):
+    API_KEY: int
+    API_VERSION: int
+    RESPONSE_TYPE = MetadataResponse_v3
+    SCHEMA: Incomplete
+    ALL_TOPICS: Incomplete
+    NO_TOPICS: Incomplete
+
+class MetadataRequest_v4(Request):
+    API_KEY: int
+    API_VERSION: int
+    RESPONSE_TYPE = MetadataResponse_v4
+    SCHEMA: Incomplete
+    ALL_TOPICS: Incomplete
+    NO_TOPICS: Incomplete
+
+class MetadataRequest_v5(Request):
+    API_KEY: int
+    API_VERSION: int
+    RESPONSE_TYPE = MetadataResponse_v5
+    SCHEMA: Incomplete
+    ALL_TOPICS: Incomplete
+    NO_TOPICS: Incomplete
+
+class MetadataRequest_v6(Request):
+    API_KEY: int
+    API_VERSION: int
+    RESPONSE_TYPE = MetadataResponse_v6
+    SCHEMA: Incomplete
+    ALL_TOPICS: Incomplete
+    NO_TOPICS: Incomplete
+
+class MetadataRequest_v7(Request):
+    API_KEY: int
+    API_VERSION: int
+    RESPONSE_TYPE = MetadataResponse_v7
+    SCHEMA: Incomplete
+    ALL_TOPICS: Incomplete
+    NO_TOPICS: Incomplete
+
+class MetadataRequest_v8(Request):
+    API_KEY: int
+    API_VERSION: int
+    RESPONSE_TYPE = MetadataResponse_v8
+    SCHEMA: Incomplete
+    ALL_TOPICS: Incomplete
+    NO_TOPICS: Incomplete
+
+MetadataRequest: Incomplete
+MetadataResponse: Incomplete

kafka-stubs/protocol/offset_for_leader_epoch.pyi

@@ -0,0 +1,61 @@
+from _typeshed import Incomplete
+
+from kafka.protocol.api import Request, Response
+
+class OffsetForLeaderEpochResponse_v0(Response):
+    API_KEY: int
+    API_VERSION: int
+    SCHEMA: Incomplete
+
+class OffsetForLeaderEpochResponse_v1(Response):
+    API_KEY: int
+    API_VERSION: int
+    SCHEMA: Incomplete
+
+class OffsetForLeaderEpochResponse_v2(Response):
+    API_KEY: int
+    API_VERSION: int
+    SCHEMA: Incomplete
+
+class OffsetForLeaderEpochResponse_v3(Response):
+    API_KEY: int
+    API_VERSION: int
+    SCHEMA: Incomplete
+
+class OffsetForLeaderEpochResponse_v4(Response):
+    API_KEY: int
+    API_VERSION: int
+    SCHEMA: Incomplete
+
+class OffsetForLeaderEpochRequest_v0(Request):
+    API_KEY: int
+    API_VERSION: int
+    RESPONSE_TYPE = OffsetForLeaderEpochResponse_v0
+    SCHEMA: Incomplete
+
+class OffsetForLeaderEpochRequest_v1(Request):
+    API_KEY: int
+    API_VERSION: int
+    RESPONSE_TYPE = OffsetForLeaderEpochResponse_v1
+    SCHEMA: Incomplete
+
+class OffsetForLeaderEpochRequest_v2(Request):
+    API_KEY: int
+    API_VERSION: int
+    RESPONSE_TYPE = OffsetForLeaderEpochResponse_v2
+    SCHEMA: Incomplete
+
+class OffsetForLeaderEpochRequest_v3(Request):
+    API_KEY: int
+    API_VERSION: int
+    RESPONSE_TYPE = OffsetForLeaderEpochResponse_v3
+    SCHEMA: Incomplete
+
+class OffsetForLeaderEpochRequest_v4(Request):
+    API_KEY: int
+    API_VERSION: int
+    RESPONSE_TYPE = OffsetForLeaderEpochResponse_v4
+    SCHEMA: Incomplete
+
+OffsetForLeaderEpochRequest: Incomplete
+OffsetForLeaderEpochResponse: Incomplete

kafka-stubs/protocol/parser.pyi

@@ -0,0 +1,11 @@
+from _typeshed import Incomplete
+
+log: Incomplete
+
+class KafkaProtocol:
+    in_flight_requests: Incomplete
+    bytes_to_send: Incomplete
+    def __init__(self, client_id=None, api_version=None) -> None: ...
+    def send_request(self, request, correlation_id=None): ...
+    def send_bytes(self): ...
+    def receive_bytes(self, data): ...

kafka-stubs/protocol/produce.pyi

@@ -0,0 +1,103 @@
+import abc
+from _typeshed import Incomplete
+from typing import type_check_only
+
+from kafka.protocol.api import Request, Response
+
+class ProduceResponse_v0(Response):
+    API_KEY: int
+    API_VERSION: int
+    SCHEMA: Incomplete
+
+class ProduceResponse_v1(Response):
+    API_KEY: int
+    API_VERSION: int
+    SCHEMA: Incomplete
+
+class ProduceResponse_v2(Response):
+    API_KEY: int
+    API_VERSION: int
+    SCHEMA: Incomplete
+
+class ProduceResponse_v3(Response):
+    API_KEY: int
+    API_VERSION: int
+    SCHEMA: Incomplete
+
+class ProduceResponse_v4(Response):
+    API_KEY: int
+    API_VERSION: int
+    SCHEMA: Incomplete
+
+class ProduceResponse_v5(Response):
+    API_KEY: int
+    API_VERSION: int
+    SCHEMA: Incomplete
+
+class ProduceResponse_v6(Response):
+    API_KEY: int
+    API_VERSION: int
+    SCHEMA: Incomplete
+
+class ProduceResponse_v7(Response):
+    API_KEY: int
+    API_VERSION: int
+    SCHEMA: Incomplete
+
+class ProduceResponse_v8(Response):
+    API_KEY: int
+    API_VERSION: int
+    SCHEMA: Incomplete
+
+@type_check_only
+class _ProduceRequest(Request, metaclass=abc.ABCMeta):
+    API_KEY: int
+    def expect_response(self): ...
+
+class ProduceRequest_v0(_ProduceRequest):
+    API_VERSION: int
+    RESPONSE_TYPE = ProduceResponse_v0
+    SCHEMA: Incomplete
+
+class ProduceRequest_v1(_ProduceRequest):
+    API_VERSION: int
+    RESPONSE_TYPE = ProduceResponse_v1
+    SCHEMA: Incomplete
+
+class ProduceRequest_v2(_ProduceRequest):
+    API_VERSION: int
+    RESPONSE_TYPE = ProduceResponse_v2
+    SCHEMA: Incomplete
+
+class ProduceRequest_v3(_ProduceRequest):
+    API_VERSION: int
+    RESPONSE_TYPE = ProduceResponse_v3
+    SCHEMA: Incomplete
+
+class ProduceRequest_v4(_ProduceRequest):
+    API_VERSION: int
+    RESPONSE_TYPE = ProduceResponse_v4
+    SCHEMA: Incomplete
+
+class ProduceRequest_v5(_ProduceRequest):
+    API_VERSION: int
+    RESPONSE_TYPE = ProduceResponse_v5
+    SCHEMA: Incomplete
+
+class ProduceRequest_v6(_ProduceRequest):
+    API_VERSION: int
+    RESPONSE_TYPE = ProduceResponse_v6
+    SCHEMA: Incomplete
+
+class ProduceRequest_v7(_ProduceRequest):
+    API_VERSION: int
+    RESPONSE_TYPE = ProduceResponse_v7
+    SCHEMA: Incomplete
+
+class ProduceRequest_v8(_ProduceRequest):
+    API_VERSION: int
+    RESPONSE_TYPE = ProduceResponse_v8
+    SCHEMA: Incomplete
+
+ProduceRequest: list[type[_ProduceRequest]]
+ProduceResponse: Incomplete

kafka-stubs/protocol/sasl_authenticate.pyi

@@ -0,0 +1,28 @@
+from _typeshed import Incomplete
+
+from kafka.protocol.api import Request, Response
+
+class SaslAuthenticateResponse_v0(Response):
+    API_KEY: int
+    API_VERSION: int
+    SCHEMA: Incomplete
+
+class SaslAuthenticateResponse_v1(Response):
+    API_KEY: int
+    API_VERSION: int
+    SCHEMA: Incomplete
+
+class SaslAuthenticateRequest_v0(Request):
+    API_KEY: int
+    API_VERSION: int
+    RESPONSE_TYPE = SaslAuthenticateResponse_v0
+    SCHEMA: Incomplete
+
+class SaslAuthenticateRequest_v1(Request):
+    API_KEY: int
+    API_VERSION: int
+    RESPONSE_TYPE = SaslAuthenticateResponse_v1
+    SCHEMA: Incomplete
+
+SaslAuthenticateRequest: Incomplete
+SaslAuthenticateResponse: Incomplete

kafka-stubs/protocol/sasl_handshake.pyi

@@ -0,0 +1,28 @@
+from _typeshed import Incomplete
+
+from kafka.protocol.api import Request, Response
+
+class SaslHandshakeResponse_v0(Response):
+    API_KEY: int
+    API_VERSION: int
+    SCHEMA: Incomplete
+
+class SaslHandshakeResponse_v1(Response):
+    API_KEY: int
+    API_VERSION: int
+    SCHEMA: Incomplete
+
+class SaslHandshakeRequest_v0(Request):
+    API_KEY: int
+    API_VERSION: int
+    RESPONSE_TYPE = SaslHandshakeResponse_v0
+    SCHEMA: Incomplete
+
+class SaslHandshakeRequest_v1(Request):
+    API_KEY: int
+    API_VERSION: int
+    RESPONSE_TYPE = SaslHandshakeResponse_v1
+    SCHEMA: Incomplete
+
+SaslHandshakeRequest: Incomplete
+SaslHandshakeResponse: Incomplete

kafka-stubs/protocol/struct.pyi

@@ -0,0 +1,14 @@
+from _typeshed import Incomplete
+
+from kafka.protocol.abstract import AbstractType
+
+class Struct(AbstractType):
+    SCHEMA: Incomplete
+    def __init__(self, *args, **kwargs) -> None: ...
+    @classmethod
+    def encode(cls, item): ...
+    @classmethod
+    def decode(cls, data): ...
+    def get_item(self, name): ...
+    def __hash__(self): ...
+    def __eq__(self, other): ...

kafka-stubs/protocol/txn_offset_commit.pyi

@@ -0,0 +1,39 @@
+from _typeshed import Incomplete
+
+from kafka.protocol.api import Request, Response
+
+class TxnOffsetCommitResponse_v0(Response):
+    API_KEY: int
+    API_VERSION: int
+    SCHEMA: Incomplete
+
+class TxnOffsetCommitResponse_v1(Response):
+    API_KEY: int
+    API_VERSION: int
+    SCHEMA: Incomplete
+
+class TxnOffsetCommitResponse_v2(Response):
+    API_KEY: int
+    API_VERSION: int
+    SCHEMA: Incomplete
+
+class TxnOffsetCommitRequest_v0(Request):
+    API_KEY: int
+    API_VERSION: int
+    RESPONSE_TYPE = TxnOffsetCommitResponse_v0
+    SCHEMA: Incomplete
+
+class TxnOffsetCommitRequest_v1(Request):
+    API_KEY: int
+    API_VERSION: int
+    RESPONSE_TYPE = TxnOffsetCommitResponse_v1
+    SCHEMA: Incomplete
+
+class TxnOffsetCommitRequest_v2(Request):
+    API_KEY: int
+    API_VERSION: int
+    RESPONSE_TYPE = TxnOffsetCommitResponse_v2
+    SCHEMA: Incomplete
+
+TxnOffsetCommitRequest: Incomplete
+TxnOffsetCommitResponse: Incomplete

kafka-stubs/protocol/types.pyi

@@ -0,0 +1,115 @@
+from _typeshed import Incomplete
+
+from kafka.protocol.abstract import AbstractType as AbstractType
+
+class Int8(AbstractType):
+    @classmethod
+    def encode(cls, value): ...
+    @classmethod
+    def decode(cls, data): ...
+
+class Int16(AbstractType):
+    @classmethod
+    def encode(cls, value): ...
+    @classmethod
+    def decode(cls, data): ...
+
+class Int32(AbstractType):
+    @classmethod
+    def encode(cls, value): ...
+    @classmethod
+    def decode(cls, data): ...
+
+class Int64(AbstractType):
+    @classmethod
+    def encode(cls, value): ...
+    @classmethod
+    def decode(cls, data): ...
+
+class Float64(AbstractType):
+    @classmethod
+    def encode(cls, value): ...
+    @classmethod
+    def decode(cls, data): ...
+
+class String(AbstractType):
+    encoding: Incomplete
+    def __init__(self, encoding: str = "utf-8") -> None: ...
+    def encode(self, value): ...
+    def decode(self, data): ...
+
+class Bytes(AbstractType):
+    @classmethod
+    def encode(cls, value): ...
+    @classmethod
+    def decode(cls, data): ...
+    @classmethod
+    def repr(cls, value): ...
+
+class Boolean(AbstractType):
+    @classmethod
+    def encode(cls, value): ...
+    @classmethod
+    def decode(cls, data): ...
+
+class Schema(AbstractType):
+    def __init__(self, *fields) -> None: ...
+    def encode(self, item): ...
+    def decode(self, data): ...
+    def __len__(self) -> int: ...
+    def repr(self, value): ...
+
+class Array(AbstractType):
+    array_of: Incomplete
+    def __init__(self, *array_of) -> None: ...
+    def encode(self, items): ...
+    def decode(self, data): ...
+    def repr(self, list_of_items): ...
+
+class UnsignedVarInt32(AbstractType):
+    @classmethod
+    def decode(cls, data): ...
+    @classmethod
+    def encode(cls, value): ...
+
+class VarInt32(AbstractType):
+    @classmethod
+    def decode(cls, data): ...
+    @classmethod
+    def encode(cls, value): ...
+
+class VarInt64(AbstractType):
+    @classmethod
+    def decode(cls, data): ...
+    @classmethod
+    def encode(cls, value): ...
+
+class CompactString(String):
+    def decode(self, data): ...
+    def encode(self, value): ...
+
+class TaggedFields(AbstractType):
+    @classmethod
+    def decode(cls, data): ...
+    @classmethod
+    def encode(cls, value): ...
+
+class CompactBytes(AbstractType):
+    @classmethod
+    def decode(cls, data): ...
+    @classmethod
+    def encode(cls, value): ...
+
+class CompactArray(Array):
+    def encode(self, items): ...
+    def decode(self, data): ...
+
+class BitField(AbstractType):
+    @classmethod
+    def decode(cls, data): ...
+    @classmethod
+    def encode(cls, vals): ...
+    @classmethod
+    def to_32_bit_field(cls, vals): ...
+    @classmethod
+    def from_32_bit_field(cls, value): ...

kafka-stubs/record/__init__.pyi

@@ -0,0 +1,3 @@
+from kafka.record.memory_records import MemoryRecords as MemoryRecords, MemoryRecordsBuilder as MemoryRecordsBuilder
+
+__all__ = ["MemoryRecords", "MemoryRecordsBuilder"]

kafka-stubs/record/_crc32c.pyi

@@ -0,0 +1,8 @@
+from _typeshed import Incomplete
+
+CRC_TABLE: Incomplete
+CRC_INIT: int
+
+def crc_update(crc, data): ...
+def crc_finalize(crc): ...
+def crc(data): ...