style fix

Author: rekby

ydb/_topic_reader/datatypes.py

@@ -179,7 +179,7 @@ def _commit_get_offsets_range(self) -> OffsetsRange:
             self.messages[-1]._commit_get_offsets_range().end,
         )
 
-    def is_empty(self) -> bool:
+    def empty(self) -> bool:
         return len(self.messages) == 0
 
     # ISessionAlive implementation
@@ -192,9 +192,4 @@ def is_alive(self) -> bool:
         )
 
     def pop_message(self) -> PublicMessage:
-        if len(self.messages) == 0:
-            raise IndexError()
-
-        res = self.messages[0]
-        self.messages = self.messages[1:]
-        return res
+        return self.messages.pop()

ydb/_topic_reader/topic_reader_asyncio.py

@@ -125,7 +125,7 @@ async def receive_batch(
         await self._reconnector.wait_message()
         return self._reconnector.receive_batch_nowait()
 
-    async def receive_message(self) -> typing.Union[datatypes.PublicMessage, None]:
+    async def receive_message(self) -> typing.Optional[datatypes.PublicMessage]:
         """
         Block until receive new message
 
@@ -414,7 +414,7 @@ def receive_message_nowait(self):
         except IndexError:
             return None
 
-        if batch.is_empty():
+        if batch.empty():
             self._message_batches.popleft()
 
         return message