add style checking

Author: slampy97

tests/coordination/test_coordination_client.py

@@ -147,7 +147,6 @@ async def test_coordination_lock_full_lifecycle(self, aio_connection):
         assert list(sem2.owners) == []
         assert list(sem2.waiters) == []
 
-
         lock2_started = asyncio.Event()
         lock2_acquired = asyncio.Event()
 
@@ -191,10 +190,8 @@ async def second_lock_task():
             sem3 = resp3.semaphore_description
             assert sem3.count == 1
 
-
         delete_resp = await lock.delete()
         assert delete_resp.status == StatusCode.SUCCESS
 
         describe_after_delete = await lock.describe()
         assert describe_after_delete.status == StatusCode.NOT_FOUND
-

ydb/_grpc/grpcwrapper/ydb_coordination.py

@@ -67,14 +67,16 @@ class SessionStart(IToProto):
     protection_key: bytes = b""
 
     def to_proto(self) -> ydb_coordination_pb2.SessionRequest:
-        return ydb_coordination_pb2.SessionRequest(session_start=ydb_coordination_pb2.SessionRequest.SessionStart(
-            path=self.path,
-            session_id=self.session_id,
-            timeout_millis=self.timeout_millis,
-            description=self.description,
-            seq_no=self.seq_no,
-            protection_key=self.protection_key,
-        ))
+        return ydb_coordination_pb2.SessionRequest(
+            session_start=ydb_coordination_pb2.SessionRequest.SessionStart(
+                path=self.path,
+                session_id=self.session_id,
+                timeout_millis=self.timeout_millis,
+                description=self.description,
+                seq_no=self.seq_no,
+                protection_key=self.protection_key,
+            )
+        )
 
 
 @dataclass
@@ -89,7 +91,8 @@ class Ping(IToProto):
 
     def to_proto(self) -> ydb_coordination_pb2.SessionRequest:
         return ydb_coordination_pb2.SessionRequest(
-            ping=ydb_coordination_pb2.SessionRequest.PingPong(opaque=self.opaque))
+            ping=ydb_coordination_pb2.SessionRequest.PingPong(opaque=self.opaque)
+        )
 
 
 @dataclass
@@ -100,9 +103,11 @@ class CreateSemaphore(IToProto):
     data: bytes = b""
 
     def to_proto(self) -> ydb_coordination_pb2.SessionRequest:
-        return ydb_coordination_pb2.SessionRequest(create_semaphore=ydb_coordination_pb2.SessionRequest.CreateSemaphore(
-            req_id=self.req_id, name=self.name, limit=self.limit, data=self.data
-        ))
+        return ydb_coordination_pb2.SessionRequest(
+            create_semaphore=ydb_coordination_pb2.SessionRequest.CreateSemaphore(
+                req_id=self.req_id, name=self.name, limit=self.limit, data=self.data
+            )
+        )
 
 
 @dataclass
@@ -112,9 +117,11 @@ class UpdateSemaphore(IToProto):
     data: bytes
 
     def to_proto(self) -> ydb_coordination_pb2.SessionRequest:
-        return ydb_coordination_pb2.SessionRequest(update_semaphore=ydb_coordination_pb2.SessionRequest.UpdateSemaphore(
-            req_id=self.req_id, name=self.name, data=self.data
-        ))
+        return ydb_coordination_pb2.SessionRequest(
+            update_semaphore=ydb_coordination_pb2.SessionRequest.UpdateSemaphore(
+                req_id=self.req_id, name=self.name, data=self.data
+            )
+        )
 
 
 @dataclass
@@ -124,9 +131,11 @@ class DeleteSemaphore(IToProto):
     force: bool = False
 
     def to_proto(self) -> ydb_coordination_pb2.SessionRequest:
-        return ydb_coordination_pb2.SessionRequest(delete_semaphore=ydb_coordination_pb2.SessionRequest.DeleteSemaphore(
-            req_id=self.req_id, name=self.name, force=self.force
-        ))
+        return ydb_coordination_pb2.SessionRequest(
+            delete_semaphore=ydb_coordination_pb2.SessionRequest.DeleteSemaphore(
+                req_id=self.req_id, name=self.name, force=self.force
+            )
+        )
 
 
 @dataclass
@@ -147,7 +156,8 @@ def to_proto(self) -> ydb_coordination_pb2.SessionRequest:
                 count=self.count,
                 data=self.data,
                 ephemeral=self.ephemeral,
-            ))
+            )
+        )
 
 
 @dataclass
@@ -157,9 +167,9 @@ class ReleaseSemaphore(IToProto):
 
     def to_proto(self) -> ydb_coordination_pb2.SessionRequest:
         return ydb_coordination_pb2.SessionRequest(
-            release_semaphore=ydb_coordination_pb2.SessionRequest.ReleaseSemaphore(
-                req_id=self.req_id, name=self.name
-            ))
+            release_semaphore=ydb_coordination_pb2.SessionRequest.ReleaseSemaphore(req_id=self.req_id, name=self.name)
+        )
+
 
 @dataclass
 class DescribeSemaphore(IToProto):
@@ -178,7 +188,7 @@ def to_proto(self) -> ydb_coordination_pb2.SessionRequest:
                 name=self.name,
                 req_id=self.req_id,
                 watch_data=self.watch_data,
-                watch_owners=self.watch_owners
+                watch_owners=self.watch_owners,
             )
         )
 

ydb/aio/coordination/lock.py

@@ -46,7 +46,6 @@ def __init__(
 
         self._wait_timeout: float = self._timeout_millis / 1000.0
 
-
     def next_req_id(self) -> int:
         r = self._next_req_id
         self._next_req_id += 1
@@ -100,13 +99,10 @@ async def _wait_for_response(self, req_id: int, *, kind: str):
                 "describe": "describe",
                 "update": "update",
                 "delete": "delete",
-                "create": "create"
+                "create": "create",
             }.get(kind, "operation")
 
-            raise issues.Error(
-                f"Timeout waiting for lock {self._name} {action}"
-            )
-
+            raise issues.Error(f"Timeout waiting for lock {self._name} {action}")
 
     async def __aenter__(self):
         await self._ensure_session()
@@ -157,12 +153,7 @@ async def create(self, init_limit, init_data):
 
         req_id = self.next_req_id()
 
-        req = CreateSemaphore(
-            req_id=req_id,
-            name=self._name,
-            limit=init_limit,
-            data=init_data
-        ).to_proto()
+        req = CreateSemaphore(req_id=req_id, name=self._name, limit=init_limit, data=init_data).to_proto()
 
         await self.send(req)
 
@@ -184,7 +175,6 @@ async def delete(self):
         resp = await self._wait_for_response(req_id, kind="delete")
         return resp
 
-
     async def describe(self):
         await self._ensure_session()
 
@@ -208,11 +198,7 @@ async def update(self, new_data):
         await self._ensure_session()
 
         req_id = self.next_req_id()
-        req = UpdateSemaphore(
-            req_id=req_id,
-            name=self._name,
-            data=new_data
-        ).to_proto()
+        req = UpdateSemaphore(req_id=req_id, name=self._name, data=new_data).to_proto()
 
         await self.send(req)