don't stop if already stopped

This commit is contained in:
əlemi 2022-04-19 01:17:57 +02:00
parent ca45980bcd
commit 022ed83187
No known key found for this signature in database
GPG key ID: BBCBFE5D7244634E
2 changed files with 5 additions and 2 deletions

View file

@ -88,6 +88,7 @@ class MinecraftClient:
await self._handshake(ConnectionState.STATUS) await self._handshake(ConnectionState.STATUS)
return await self._status(ping) return await self._status(ping)
finally: finally:
if self.dispatcher.connected:
await self.dispatcher.disconnect() await self.dispatcher.disconnect()
async def join(self, host:str="", port:int=0, proto:int=0): async def join(self, host:str="", port:int=0, proto:int=0):
@ -99,6 +100,7 @@ class MinecraftClient:
if await self._login(): if await self._login():
await self._play() await self._play()
finally: finally:
if self.dispatcher.connected:
await self.dispatcher.disconnect() await self.dispatcher.disconnect()
async def _handshake(self, state:ConnectionState): async def _handshake(self, state:ConnectionState):

View file

@ -249,6 +249,7 @@ class Dispatcher:
await self._incoming.put(packet) await self._incoming.put(packet)
if self.state != ConnectionState.PLAY: if self.state != ConnectionState.PLAY:
await self._incoming.join() # During play we can pre-process packets await self._incoming.join() # During play we can pre-process packets
except (asyncio.TimeoutError, TimeoutError): except (asyncio.TimeoutError, TimeoutError):
self.logger.error("Connection timed out") self.logger.error("Connection timed out")
await self.disconnect(block=False) await self.disconnect(block=False)