Fix remaining type issues

This commit is contained in:
Teknique
2023-06-17 12:09:40 -07:00
parent f1aad14099
commit e0efb2853b
3 changed files with 22 additions and 9 deletions

View File

@@ -49,8 +49,8 @@ _VALIDATOR_RECV_MESSAGE = _get_schema_validator(
class _JsonVeilidAPI(VeilidAPI):
reader: asyncio.StreamReader
writer: asyncio.StreamWriter
reader: Optional[asyncio.StreamReader]
writer: Optional[asyncio.StreamWriter]
update_callback: Callable[[VeilidUpdate], Awaitable]
handle_recv_messages_task: Optional[asyncio.Task]
validate_schemas: bool
@@ -85,6 +85,7 @@ class _JsonVeilidAPI(VeilidAPI):
await self.lock.acquire()
try:
self.reader = None
assert self.writer is not None
self.writer.close()
await self.writer.wait_closed()
self.writer = None
@@ -138,6 +139,7 @@ class _JsonVeilidAPI(VeilidAPI):
async def handle_recv_messages(self):
# Read lines until we're done
try:
assert self.reader is not None
while True:
linebytes = await self.reader.readline()
if not linebytes.endswith(b"\n"):
@@ -225,6 +227,7 @@ class _JsonVeilidAPI(VeilidAPI):
# Send to socket
try:
assert writer is not None
writer.write(reqbytes)
await writer.drain()
except Exception:

View File

@@ -82,7 +82,7 @@ class ByteCount(int):
pass
class OperationId(int):
class OperationId(str):
pass