Fix: typings.

Signed-off-by: dblock <[email protected]>
opensearch-project · Sep 7, 2023 · 34a23fb · 34a23fb
1 parent d07cfaf
commit 34a23fb
Show file tree

Hide file tree

Showing 6 changed files with 15 additions and 6 deletions.
diff --git a/src/opensearch_sdk_py/transport/network_message.py b/src/opensearch_sdk_py/transport/network_message.py
@@ -10,6 +10,8 @@
 # https:/opensearch-project/OpenSearch/blob/main/server/src/main/java/org/opensearch/transport/NetworkMessage.java
 
 
+from typing import Optional
+
 from opensearch_sdk_py.transport.tcp_header import TcpHeader
 from opensearch_sdk_py.transport.thread_context_struct import ThreadContextStruct
 from opensearch_sdk_py.transport.transport_status import TransportStatus
@@ -22,7 +24,7 @@ def __init__(
  thread_context: ThreadContextStruct = None,
  version: Version = None,
  status: int = TransportStatus.STATUS_REQRES,
- request_id: int = None,
+ request_id: Optional[int] = None,
  ) -> None:
  self.thread_context_struct = thread_context if thread_context else ThreadContextStruct()
  self.tcp_header = TcpHeader(version=version, status=status, request_id=request_id)

diff --git a/src/opensearch_sdk_py/transport/outbound_message.py b/src/opensearch_sdk_py/transport/outbound_message.py
@@ -27,7 +27,7 @@ def __init__(
  thread_context: ThreadContextStruct = None,
  version: Version = None,
  status: int = TransportStatus.STATUS_REQRES,
- request_id: int = None,
+ request_id: Optional[int] = None,
  message: TransportMessage = None,
  ):
  self._message_bytes: Optional[bytes]

diff --git a/src/opensearch_sdk_py/transport/outbound_message_request.py b/src/opensearch_sdk_py/transport/outbound_message_request.py
@@ -9,6 +9,8 @@
 
 # https:/opensearch-project/OpenSearch/blob/main/server/src/main/java/org/opensearch/transport/OutboundMessage.java#L122
 
+from typing import Optional
+
 from opensearch_sdk_py.transport.outbound_message import OutboundMessage
 from opensearch_sdk_py.transport.stream_input import StreamInput
 from opensearch_sdk_py.transport.stream_output import StreamOutput
@@ -26,7 +28,7 @@ def __init__(
  message: TransportMessage = None,
  version: Version = None,
  action: str = "",
- request_id: int = None,
+ request_id: Optional[int] = None,
  is_handshake: bool = False,
  is_compress: bool = False,
  ) -> None:

diff --git a/src/opensearch_sdk_py/transport/outbound_message_response.py b/src/opensearch_sdk_py/transport/outbound_message_response.py
@@ -9,6 +9,8 @@
 
 # https:/opensearch-project/OpenSearch/blob/main/server/src/main/java/org/opensearch/transport/OutboundMessage.java#L168
 
+from typing import Optional
+
 from opensearch_sdk_py.transport.outbound_message import OutboundMessage
 from opensearch_sdk_py.transport.thread_context_struct import ThreadContextStruct
 from opensearch_sdk_py.transport.transport_message import TransportMessage
@@ -23,7 +25,7 @@ def __init__(
  features: list[str] = [],
  message: TransportMessage = None,
  version: Version = None,
- request_id: int = None,
+ request_id: Optional[int] = None,
  is_handshake: bool = False,
  is_compress: bool = False,
  ):

diff --git a/src/opensearch_sdk_py/transport/task_id.py b/src/opensearch_sdk_py/transport/task_id.py
@@ -9,12 +9,14 @@
 
 # https:/opensearch-project/OpenSearch/blob/main/libs/core/src/main/java/org/opensearch/core/tasks/TaskId.java
 
+from typing import Optional
+
 from opensearch_sdk_py.transport.stream_input import StreamInput
 from opensearch_sdk_py.transport.stream_output import StreamOutput
 
 
 class TaskId:
- def __init__(self, node_id: str = "", id: int = None) -> None:
+ def __init__(self, node_id: str = "", id: Optional[int] = None) -> None:
  self.node_id = node_id
  self.id = id
 

diff --git a/src/opensearch_sdk_py/transport/tcp_header.py b/src/opensearch_sdk_py/transport/tcp_header.py
@@ -10,6 +10,7 @@
 # https:/opensearch-project/OpenSearch/blob/main/server/src/main/java/org/opensearch/transport/TcpHeader.java
 
 import itertools
+from typing import Optional
 
 from opensearch_sdk_py.transport.stream_input import StreamInput
 from opensearch_sdk_py.transport.stream_output import StreamOutput
@@ -37,7 +38,7 @@ class TcpHeader:
  def __init__(
  self,
  prefix: bytes = b"ES",
- request_id: int = None,
+ request_id: Optional[int] = None,
  status: int = 0,
  version: Version = Version.CURRENT,
  size: int = MESSAGE_SIZE,