diff --git a/stubs/kafka-python/METADATA.toml b/stubs/kafka-python/METADATA.toml index 2f06c1b96010..33686d2d9baf 100644 --- a/stubs/kafka-python/METADATA.toml +++ b/stubs/kafka-python/METADATA.toml @@ -1,2 +1,2 @@ -version = "2.3.*" +version = "2.3.2" upstream-repository = "https://github.com/dpkp/kafka-python" diff --git a/stubs/kafka-python/kafka/errors.pyi b/stubs/kafka-python/kafka/errors.pyi index 9c7913a42520..079bf3ebd5e4 100644 --- a/stubs/kafka-python/kafka/errors.pyi +++ b/stubs/kafka-python/kafka/errors.pyi @@ -26,6 +26,8 @@ class KafkaProtocolError(KafkaError): class CorrelationIdError(KafkaProtocolError): retriable: bool +class InvalidReceiveError(KafkaProtocolError): ... + class KafkaTimeoutError(KafkaError): retriable: bool diff --git a/stubs/kafka-python/kafka/protocol/parser.pyi b/stubs/kafka-python/kafka/protocol/parser.pyi index 7d0cf2946e66..63c350d3ee55 100644 --- a/stubs/kafka-python/kafka/protocol/parser.pyi +++ b/stubs/kafka-python/kafka/protocol/parser.pyi @@ -1,11 +1,15 @@ from _typeshed import Incomplete +from collections import deque +from logging import Logger -log: Incomplete +log: Logger 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): ... + in_flight_requests: deque[tuple[int, Incomplete]] + bytes_to_send: list[bytes] + def __init__( + self, client_id: str | None = None, api_version: tuple[int, int, int] | None = None, max_frame_size: int = 100000000 + ) -> None: ... + def send_request(self, request, correlation_id: int | None = None) -> int: ... + def send_bytes(self) -> bytes: ... + def receive_bytes(self, data: bytes) -> list[Incomplete]: ...