Replace "with await" with "async with"

delivrance · delivrance · commit 9bd9d7797b0e · 2019-06-05T11:58:29.000+02:00
diff --git a/pyrogram/client/client.py b/pyrogram/client/client.py
@@ -1555,7 +1555,7 @@ async def get_file(self, media_type: int,
                        is_big: bool,
                        progress: callable,
                        progress_args: tuple = ()) -> str:
-        with await self.media_sessions_lock:
+        async with self.media_sessions_lock:
             session = self.media_sessions.get(dc_id, None)
 
             if session is None:
@@ -1670,7 +1670,7 @@ async def get_file(self, media_type: int,
                         )
 
             elif isinstance(r, types.upload.FileCdnRedirect):
-                with await self.media_sessions_lock:
+                async with self.media_sessions_lock:
                     cdn_session = self.media_sessions.get(r.dc_id, None)
 
                     if cdn_session is None:
diff --git a/pyrogram/client/ext/syncer.py b/pyrogram/client/ext/syncer.py
@@ -44,7 +44,7 @@ async def add(cls, client):
         if cls.lock is None:
             cls.lock = asyncio.Lock()
 
-        with await cls.lock:
+        async with cls.lock:
             cls.sync(client)
 
             cls.clients[id(client)] = client
@@ -54,7 +54,7 @@ async def add(cls, client):
 
     @classmethod
     async def remove(cls, client):
-        with await cls.lock:
+        async with cls.lock:
             cls.sync(client)
 
             del cls.clients[id(client)]
@@ -77,7 +77,7 @@ async def worker(cls):
             try:
                 await asyncio.wait_for(cls.event.wait(), cls.INTERVAL)
             except asyncio.TimeoutError:
-                with await cls.lock:
+                async with cls.lock:
                     for client in cls.clients.values():
                         cls.sync(client)
             else:
diff --git a/pyrogram/connection/transport/tcp/tcp.py b/pyrogram/connection/transport/tcp/tcp.py
@@ -92,7 +92,7 @@ def close(self):
                 self.socket.close()
 
     async def send(self, data: bytes):
-        with await self.lock:
+        async with self.lock:
             self.writer.write(data)
             await self.writer.drain()
 
