python-aidot 0.3.1__tar.gz → 0.3.2__tar.gz

{python_aidot-0.3.1 → python_aidot-0.3.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: python-aidot
-Version: 0.3.1
+Version: 0.3.2
 Summary: aidot control wifi lights
 Home-page: https://github.com/Aidot-Development-Team/python-aidot
 Author: aidotdev2024

{python_aidot-0.3.1 → python_aidot-0.3.2}/aidot/discover.py

@@ -12,7 +12,7 @@ from .exceptions import AidotOSError
 _LOGGER = logging.getLogger(__name__)
 
 class BroadcastProtocol:
-    _is_running: bool = True
+    _is_closed = False
     def __init__(self,callback,user_id):
         self.aes_key = bytearray(32)
         key_string = "T54uednca587"
@@ -28,6 +28,9 @@ class BroadcastProtocol:
         sock.setsockopt(socket.SOL_SOCKET, socket.SO_BROADCAST, 1)
 
     def send_broadcast(self) -> None:
+        if self._is_closed is True:
+            _LOGGER.error(f"{self.user_id}:Connection is closed")
+            return
         current_timestamp_milliseconds = int(time.time() * 1000)
         seq = str(current_timestamp_milliseconds + 1)[-9:]
         message = {
@@ -44,7 +47,10 @@ class BroadcastProtocol:
             }
         }
         send_data = aes_encrypt(json.dumps(message).encode(),self.aes_key)
-        self.transport.sendto(send_data, ('255.255.255.255', 6666))
+        try:
+            self.transport.sendto(send_data, ('255.255.255.255', 6666))
+        except Exception as error:
+            _LOGGER.error(f"{self.user_id}:Connection lost due to error: {error}")
 
     def datagram_received(self, data, addr):
         data_str = aes_decrypt(data,self.aes_key)
@@ -56,16 +62,21 @@ class BroadcastProtocol:
                         self._discover_cb(devId,{CONF_IPADDRESS : addr[0]})
 
     def error_received(self, exc):
-        _LOGGER.error(f"Error occurred: {exc}")
+        _LOGGER.error(f"{self.user_id}:Error occurred: {exc}")
     
     def close(self) -> None:
-        self.transport.close()
+        try:
+            self.transport.close()
+        except Exception as error:
+            _LOGGER.error(f"Connection lost due to error: {error}")
+        
     
     def connection_lost(self, exc):
+        self._is_closed = True
         if exc:
-            _LOGGER.error(f"Connection lost due to error: {exc}")
+            _LOGGER.error(f"{self.user_id}:Connection lost due to error: {exc}")
         else:
-            _LOGGER.info("Connection closed.")
+            _LOGGER.info("{self.user_id}:Connection closed.")
 
 class Discover:
     _login_info: dict[str: Any] = None
@@ -96,5 +107,4 @@ class Discover:
     def close(self) -> None:
         if self._broadcast_protocol is not None:
             self._broadcast_protocol.close()
-
-
+        self._broadcast_protocol.send_broadcast()

{python_aidot-0.3.1 → python_aidot-0.3.2}/python_aidot.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: python-aidot
-Version: 0.3.1
+Version: 0.3.2
 Summary: aidot control wifi lights
 Home-page: https://github.com/Aidot-Development-Team/python-aidot
 Author: aidotdev2024

{python_aidot-0.3.1 → python_aidot-0.3.2}/setup.py

@@ -5,7 +5,7 @@ with open("README.md", "r") as fh:
 
 setuptools.setup(
     name="python-aidot",
-    version="0.3.1",
+    version="0.3.2",
     author="aidotdev2024",
     url='https://github.com/Aidot-Development-Team/python-aidot',
     description="aidot control wifi lights",