Cleanup tasks on disconnect (#148)

* Clean up connections and tasks when disconnecting

* Fix lint

Author: puddly

zigpy_xbee/api.py

@@ -347,6 +347,10 @@ async def _reconnect_till_done(self) -> None:
         )
 
     def close(self):
+        if self._conn_lost_task:
+            self._conn_lost_task.cancel()
+            self._conn_lost_task = None
+
         if self._uart:
             self._uart.close()
             self._uart = None

zigpy_xbee/zigbee/application.py

@@ -49,6 +49,7 @@ async def disconnect(self):
         """Shutdown application."""
         if self._api:
             self._api.close()
+            self._api = None
 
     async def connect(self):
         self._api = await zigpy_xbee.api.XBee.new(self, self._config[CONF_DEVICE])
@@ -191,11 +192,11 @@ async def force_remove(self, dev):
     async def add_endpoint(self, descriptor: zdo_t.SimpleDescriptor) -> None:
         """Register a new endpoint on the device."""
         self._device.replacement["endpoints"][descriptor.endpoint] = {
-                "device_type": descriptor.device_type,
-                "profile_id": descriptor.profile,
-                "input_clusters": descriptor.input_clusters,
-                "output_clusters": descriptor.output_clusters,
-            }
+            "device_type": descriptor.device_type,
+            "profile_id": descriptor.profile,
+            "input_clusters": descriptor.input_clusters,
+            "output_clusters": descriptor.output_clusters,
+        }
         self._device.add_endpoint(descriptor.endpoint)
 
     async def _get_association_state(self):