Fix: linting errors

This commit is contained in:
Jinesh Jain
2025-07-08 18:34:30 +05:30
parent 75abc8b863
commit 8fb664bfdf

View File

@ -33,27 +33,27 @@ if TYPE_CHECKING:
class ReadWriteLock: class ReadWriteLock:
def __init__(self): def __init__(self) -> None:
self._readers = 0 self._readers = 0
self._lock = trio.Lock() # Protects _readers self._lock = trio.Lock() # Protects _readers
self._write_lock = trio.Lock() self._write_lock = trio.Lock()
async def acquire_read(self): async def acquire_read(self) -> None:
async with self._lock: async with self._lock:
self._readers += 1 self._readers += 1
if self._readers == 1: if self._readers == 1:
await self._write_lock.acquire() await self._write_lock.acquire()
async def release_read(self): async def release_read(self) -> None:
async with self._lock: async with self._lock:
self._readers -= 1 self._readers -= 1
if self._readers == 0: if self._readers == 0:
self._write_lock.release() self._write_lock.release()
async def acquire_write(self): async def acquire_write(self) -> None:
await self._write_lock.acquire() await self._write_lock.acquire()
def release_write(self): def release_write(self) -> None:
self._write_lock.release() self._write_lock.release()
@ -152,8 +152,8 @@ class MplexStream(IMuxedStream):
return await self._read_until_eof() return await self._read_until_eof()
if len(self._buf) == 0: if len(self._buf) == 0:
data: bytes data: bytes
# Peek whether there is data available. If yes, we just read until there is # Peek whether there is data available. If yes, we just read until
# no data, then return. # there is no data, then return.
try: try:
data = self.incoming_data_channel.receive_nowait() data = self.incoming_data_channel.receive_nowait()
self._buf.extend(data) self._buf.extend(data)
@ -185,6 +185,7 @@ class MplexStream(IMuxedStream):
return bytes(payload) return bytes(payload)
finally: finally:
await self.rw_lock.release_read() await self.rw_lock.release_read()
return b""
async def write(self, data: bytes) -> None: async def write(self, data: bytes) -> None:
""" """