rename messagetype

Author: rekby

ydb/_topic_writer/topic_writer.py

@@ -13,7 +13,7 @@
 from .._grpc.grpcwrapper.common_utils import IToProto
 from .._grpc.grpcwrapper.ydb_topic_public_types import PublicCodec
 
-MessageType = typing.Union["PublicMessage", "PublicMessage.SimpleMessageSourceType"]
+Message = typing.Union["PublicMessage", "PublicMessage.SimpleMessageSourceType"]
 
 
 @dataclass
@@ -116,9 +116,7 @@ def __init__(
         self.data = data
 
     @staticmethod
-    def _create_message(
-        data: Union["PublicMessage", "PublicMessage.SimpleMessageSourceType"]
-    ) -> "PublicMessage":
+    def _create_message(data: Message) -> "PublicMessage":
         if isinstance(data, PublicMessage):
             return data
         return PublicMessage(data=data)

ydb/_topic_writer/topic_writer_asyncio.py

@@ -17,7 +17,7 @@
     TopicWriterError,
     messages_to_proto_requests,
     PublicWriteResultTypes,
-    MessageType,
+    Message,
 )
 from .. import (
     _apis,
@@ -79,7 +79,7 @@ async def close(self, *, flush: bool = True):
 
     async def write_with_ack(
         self,
-        messages: Union[MessageType, List[MessageType]],
+        messages: Union[Message, List[Message]],
     ) -> Union[PublicWriteResultTypes, List[PublicWriteResultTypes]]:
         """
         IT IS SLOWLY WAY. IT IS BAD CHOISE IN MOST CASES.
@@ -100,7 +100,7 @@ async def write_with_ack(
 
     async def write_with_ack_future(
         self,
-        messages: Union[MessageType, List[MessageType]],
+        messages: Union[Message, List[Message]],
     ) -> Union[asyncio.Future, List[asyncio.Future]]:
         """
         send one or number of messages to server.
@@ -126,7 +126,7 @@ async def write_with_ack_future(
 
     async def write(
         self,
-        messages: Union[MessageType, List[MessageType]],
+        messages: Union[Message, List[Message]],
     ):
         """
         send one or number of messages to server.

ydb/_topic_writer/topic_writer_sync.py

@@ -10,7 +10,7 @@
     TopicWriterError,
     PublicWriterInitInfo,
     PublicWriteResult,
-    MessageType,
+    Message,
 )
 
 from .topic_writer_asyncio import WriterAsyncIO
@@ -91,20 +91,20 @@ def wait_init(self, timeout: Optional[TimeoutType] = None) -> PublicWriterInitIn
 
     def write(
         self,
-        messages: Union[MessageType, List[MessageType]],
+        messages: Union[Message, List[Message]],
         timeout: Union[float, None] = None,
     ):
         self._call_sync(self._async_writer.write(messages), timeout=timeout)
 
     def async_write_with_ack(
         self,
-        messages: Union[MessageType, List[MessageType]],
+        messages: Union[Message, List[Message]],
     ) -> Future[Union[PublicWriteResult, List[PublicWriteResult]]]:
         return self._call(self._async_writer.write_with_ack(messages))
 
     def write_with_ack(
         self,
-        messages: Union[MessageType, List[MessageType]],
+        messages: Union[Message, List[Message]],
         timeout: Union[float, None] = None,
     ) -> Union[PublicWriteResult, List[PublicWriteResult]]:
         return self._call_sync(