Detect disconnection and exit the script

game.py

@@ -21,7 +21,7 @@ from protocol.packets import (
     ClientWindowConfirmationPacket, EntityMetadataPacket,
     SpawnLivingEntityPacket, EntityPositionRotationPacket, DestroyEntitiesPacket,
     EntityActionPacket, EntityTeleport, InteractEntityPacket, TradeListPacket,
-    SelectTradePacket,
+    SelectTradePacket, DisconnectPacket,
 )
 
 from protocol.types import Slot
@@ -374,6 +374,7 @@ class Game:
         register(self.handle_update_health, clientbound.play.UpdateHealthPacket)
         #register(self.handle_entity_velocity, clientbound.play.EntityVelocityPacket)
         register(self.handle_trade_list, TradeListPacket)
+        register(self.handle_disconnect, DisconnectPacket)
 
         #register(self.handle_packet, Packet, early=True)
 
@@ -1230,6 +1231,12 @@ class Game:
         packet.selected_slot = num
         self.g.connection.write_packet(packet)
 
+    def handle_disconnect(self, packet):
+        print(packet)
+        print('Client disconnected!')
+        import os
+        os._exit(1)
+
     def tick(self):
         if self.g.breaking:
             self.animate()