Merge pull request #16 from Poshy163/local-api-support

Author: CharlesGillanders

alphaess/alphaess.py

@@ -19,7 +19,8 @@ def __init__(
             appID,
             appSecret,
             session: aiohttp.ClientSession | None = None,
-            timeout: int = 30
+            timeout: int = 30,
+            ipaddress=None
     ) -> None:
         """Initialize."""
         self.appID = appID
@@ -31,6 +32,7 @@ def __init__(
         self.session = session or aiohttp.ClientSession()
         self._created_session = not session
         self.timeout = timeout
+        self.ipaddress = ipaddress
 
     async def close(self) -> None:
         """Close the AlphaESS API client."""
@@ -312,6 +314,31 @@ async def updateDisChargeConfigInfo(self, sysSn, batUseCap, ctrDis, timeDise1, t
         except Exception as e:
             logger.error(f"Error: {e} when calling {resource}")
 
+    async def getIPData(self) -> Optional(dict):
+        ENDPOINTS = {
+            "status": "/config?command=status",
+            "device_info": "/config?command=devinfo"
+        }
+
+        async with aiohttp.ClientSession() as session:
+            tasks = []
+            for name, path in ENDPOINTS.items():
+                url = f"http://{self.ipaddress}{path}"
+                tasks.append(self._fetch(session, name, url))
+            results = await asyncio.gather(*tasks)
+            return dict(results)
+
+    @staticmethod
+    async def _fetch(session, name, url):
+        try:
+            async with session.get(url) as response:
+                response.raise_for_status()
+                data = await response.json(content_type=None)
+                return name, data
+        except Exception as e:
+            print(f"Failed to fetch {name} from {url}: {e}")
+            return name, None
+
     async def api_get(self, path, json=None) -> Optional(list):
         """Retrieve ESS list by serial number from Alpha ESS"""
         if json is None:
@@ -379,6 +406,15 @@ async def getdata(self, get_power=False, get_ev=False, self_delay=0) -> Optional
         """Get All Data For All serial numbers from Alpha ESS"""
         try:
             alldata = []
+            if self.ipaddress:
+                ip_data = await self.getIPData()
+                if ip_data:
+                    # Wrap it like a unit and add to alldata
+                    alldata.append({
+                        "type": "local_ip_data",
+                        "ip": self.ipaddress,
+                        **ip_data  # merge status and device_info keys
+                    })
             units = await self.getESSList()
             for unit in units:
                 if "sysSn" in unit:
@@ -407,7 +443,6 @@ async def getdata(self, get_power=False, get_ev=False, self_delay=0) -> Optional
                             unit['EVCurrent'] = await self.getEvChargerCurrentsBySn(serial)
                         except Exception:
                             pass
-
                     alldata.append(unit)
                     logger.debug(alldata)
             return alldata

apitest.py

@@ -7,9 +7,11 @@
 if len(sys.argv) != 3:
     appID = input("AppID: ")
     appSecret = input("AppSecret: ")
+    IPAddress = input("IP Address (blank if not using): ") or None
 else:
     appID = sys.argv[1]
     appSecret = sys.argv[2]
+    IPAddress = sys.argv[3]
 
 logger = logging.getLogger('')
 logger.setLevel(logging.DEBUG)
@@ -19,11 +21,14 @@
 handler.setFormatter(formatter)
 logger.addHandler(handler)
 
+if sys.platform.startswith("win"):
+    asyncio.set_event_loop_policy(asyncio.WindowsSelectorEventLoopPolicy())
+
 
 async def main():
     logger.debug("instantiating Alpha ESS Client")
     try:
-        client: alphaess = alphaess(appID, appSecret)
+        client: alphaess = alphaess(appID, appSecret, ipaddress=IPAddress)
         ESSList = await client.getESSList()
         for unit in ESSList:
             if "sysSn" in unit:

requirements.txt

@@ -1,3 +1,3 @@
-aiohttp~=3.11.13
+aiohttp~=3.12.13
 voluptuous~=0.15.2
-setuptools~=75.8.2
+setuptools~=80.9.0

setup.py

@@ -5,7 +5,7 @@
 
 setup(
     name="alphaessopenapi",
-    version="0.0.13",
+    version="0.0.14",
     author="Charles Gillanders",
     author_email="charles@charlesgillanders.com",
     description="A python library to retrieve energy statistics from your Alpha ESS inverter by polling the Official Alpha ESS Open API.",