TISControlProtocol 1.0.69__tar.gz → 1.0.71__tar.gz

{tiscontrolprotocol-1.0.69 → tiscontrolprotocol-1.0.71}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: TISControlProtocol
-Version: 1.0.69
+Version: 1.0.71
 Summary: A library for TIS Control Protocol
 Project-URL: Homepage, https://github.com/TISControlHass
 Project-URL: Issues, https://github.com/TISControlHass/issues

{tiscontrolprotocol-1.0.69 → tiscontrolprotocol-1.0.71}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "TISControlProtocol"
-version = "1.0.69"
+version = "1.0.71"
 description = "A library for TIS Control Protocol"
 readme = "README.md"
 requires-python = ">=3.11"

{tiscontrolprotocol-1.0.69 → tiscontrolprotocol-1.0.71}/src/TISControlProtocol/api.py

@@ -19,6 +19,7 @@ from TISControlProtocol.shared import get_real_mac
 from .apis import (
     BillConfigEndpoint,
     ChangeSecurityPassEndpoint,
+    ToggleConnectionEndpoint,
     GetBillConfigEndpoint,
     GetKeyEndpoint,
     PasswordsEndpoint,
@@ -47,8 +48,10 @@ class TISApi:
         """Initialize the API class."""
         self.host = host
         self.port = port
+        self.loop = None
         self.protocol = None
         self.transport = None
+        self.sock = None
         self.hass = hass
         self.config_entries = {}
         self.bill_configs = {}
@@ -60,20 +63,39 @@ class TISApi:
         self.cms_url = "https://cms-tis.com"
         self._cms_task_unsub = None
 
+    async def setup(self):
+        """Setup the TIS API."""
+        try:
+            await self.connect()
+            await self._initialize_hass_data()
+            await self._register_http_views()
+            self.hass.async_add_executor_job(self.run_display)
+        except Exception as e:
+            logging.error("Error during setup: %s", e)
+
     async def connect(self):
         """Connect to the TIS API."""
         self.loop = self.hass.loop
         self.sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
+
         try:
             await self._setup_udp_protocol()
-            await self._initialize_hass_data()
-            await self._register_http_views()
-            self.hass.async_add_executor_job(self.run_display)
-            # self._register_services()
-            # self._schedule_cms_data_task()
-        except Exception as e:
-            logging.error("Error during connection setup: %s", e)
-            raise ConnectionError
+        except Exception:
+            self.sock.close()
+            self.sock = None
+            self.loop = None
+            raise
+
+    async def disconnect(self):
+        """Disconnect from the TIS API."""
+        if self.transport:
+            self.transport.close()
+            self.transport = None
+        self.protocol = None
+        self.loop = None
+        if hasattr(self, "sock") and self.sock:
+            self.sock.close()
+            self.sock = None
 
     async def _setup_udp_protocol(self):
         """Setup the UDP protocol."""
@@ -107,6 +129,7 @@ class TISApi:
             self.hass.http.register_view(BillConfigEndpoint(self))
             self.hass.http.register_view(GetBillConfigEndpoint(self))
             self.hass.http.register_view(PasswordsEndpoint(self))
+            self.hass.http.register_view(ToggleConnectionEndpoint(self))
         except Exception as e:
             logging.error("Error registering views %s", e)
             raise ConnectionError
@@ -209,7 +232,7 @@ class TISApi:
 
         if self.display is None:
             self.display = TISDisplay(self.display_logo, self.version)
-            
+
         self.display.run_display()
 
     async def parse_device_manager_request(self, data: dict) -> None:

{tiscontrolprotocol-1.0.69 → tiscontrolprotocol-1.0.71}/src/TISControlProtocol/apis/__init__.py

@@ -16,6 +16,7 @@ from .password_form_endpoint import PasswordFormEndpoint
 from .submit_password import SubmitPasswordEndpoint
 from .password_dashboard import PasswordDashboardEndpoint
 from .passwords_endpoint import PasswordsEndpoint
+from .toggle_connection_endpoint import ToggleConnectionEndpoint
 
 
 __all__ = [
@@ -29,6 +30,7 @@ __all__ = [
     "GetBillConfigEndpoint",
     "SubmitPasswordEndpoint",
     "PasswordsEndpoint",
+    "ToggleConnectionEndpoint",
 ]
 
 

tiscontrolprotocol-1.0.71/src/TISControlProtocol/apis/toggle_connection_endpoint.py

@@ -0,0 +1,52 @@
+from homeassistant.components.http import HomeAssistantView
+from aiohttp import web
+from TISControlProtocol.shared import get_real_mac
+import logging
+
+
+class ToggleConnectionEndpoint(HomeAssistantView):
+    """Connect or Disconnect the TIS API"""
+
+    url = "/api/toggle-connection"
+    name = "api:toggle-connection"
+    requires_auth = False
+
+    def __init__(self, tis_api):
+        self.tis_api = tis_api
+
+    async def post(self, request):
+        mac_address = request.query.get("mac_address")
+
+        if mac_address is None:
+            logging.info("Required parameters not found in query, parsing request body")
+            try:
+                data = await request.json()
+                mac_address = data.get("mac_address")
+            except Exception:
+                pass
+
+        mac = await get_real_mac("end0")
+
+        if mac_address is None:
+            return web.json_response(
+                {"error": "required parameters are missing"}, status=400
+            )
+        elif mac_address != mac:
+            return web.json_response({"error": "Unauthorized"}, status=403)
+
+        if self.tis_api.transport:
+            logging.info("Disconnecting from TIS API")
+            try:
+                await self.tis_api.disconnect()
+                return web.json_response({"message": "Disconnected"}, status=200)
+            except Exception as e:
+                logging.error(f"Error disconnecting: {e}")
+                return web.json_response({"error": "Failed to disconnect"}, status=500)
+        else:
+            logging.info("Connecting to TIS API")
+            try:
+                await self.tis_api.connect()
+                return web.json_response({"message": "Connected"}, status=200)
+            except Exception as e:
+                logging.error(f"Error connecting: {e}")
+                return web.json_response({"error": "Failed to connect"}, status=500)