style fix

Author: rekby

ydb/_topic_common/common.py

@@ -86,7 +86,8 @@ def unsafe_call_with_result(self, coro: typing.Coroutine, timeout: TimeoutType):
         except concurrent.futures.TimeoutError:
             raise TimeoutError()
         finally:
-            f.cancel()
+            if not f.done():
+                f.cancel()
 
     def safe_call_with_result(self, coro: typing.Coroutine, timeout: TimeoutType):
         """
@@ -126,11 +127,8 @@ async def call_coro():
             except asyncio.CancelledError:
                 res.set_exception(TimeoutError())
 
-        async def sleep0():
-            await asyncio.sleep(0)
-
         coro_future = asyncio.run_coroutine_threadsafe(call_coro(), self._loop)
-        asyncio.run_coroutine_threadsafe(sleep0(), self._loop).result()
+        asyncio.run_coroutine_threadsafe(asyncio.sleep(0), self._loop).result()
         coro_future.cancel()
         return res.result()
 

ydb/_topic_common/common_test.py

@@ -217,11 +217,8 @@ async def callback():
             caller.safe_call_with_result(callback(), timeout)
         finished = time.monotonic()
 
-        async def sleep0():
-            await asyncio.sleep(0)
-
         # wait one loop for handle task cancelation
-        asyncio.run_coroutine_threadsafe(sleep0(), separate_loop)
+        asyncio.run_coroutine_threadsafe(asyncio.sleep(0), separate_loop)
 
         assert callback_loop is separate_loop
         assert finished - start > timeout

ydb/_topic_writer/topic_writer.py

@@ -178,7 +178,7 @@ def __init__(self, message: str):
 
 class TopicWriterClosedError(ydb.Error):
     def __init__(self):
-        super(TopicWriterClosedError, self).__init__("Topic writer already closed")
+        super().__init__("Topic writer already closed")
 
 
 class TopicWriterRepeatableError(TopicWriterError):