Merge pull request #198 from siphu/main

Fix login issue with #196

Author: binarydev

custom_components/generac/__init__.py

@@ -9,9 +9,9 @@
 from homeassistant.config_entries import ConfigEntry
 from homeassistant.core import HomeAssistant
 from homeassistant.exceptions import ConfigEntryNotReady
-from homeassistant.helpers.aiohttp_client import async_get_clientsession
 
 from .api import GeneracApiClient
+from .utils import async_client_session
 from .const import CONF_PASSWORD
 from .const import CONF_USERNAME
 from .const import DOMAIN
@@ -32,7 +32,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry):
     username = entry.data.get(CONF_USERNAME, "")
     password = entry.data.get(CONF_PASSWORD, "")
 
-    session = async_get_clientsession(hass)
+    session = await async_client_session(hass)
     client = GeneracApiClient(username, password, session)
 
     coordinator = GeneracDataUpdateCoordinator(hass, client=client, config_entry=entry)

custom_components/generac/config_flow.py

@@ -4,9 +4,9 @@
 import voluptuous as vol
 from homeassistant import config_entries
 from homeassistant.core import callback
-from homeassistant.helpers.aiohttp_client import async_create_clientsession
 
 from .api import GeneracApiClient
+from .utils import async_client_session
 from .api import InvalidCredentialsException
 from .const import CONF_OPTIONS
 from .const import CONF_PASSWORD
@@ -68,7 +68,7 @@ async def _show_config_form(self, user_input):  # pylint: disable=unused-argumen
     async def _test_credentials(self, username, password):
         """Return true if credentials is valid."""
         try:
-            session = async_create_clientsession(self.hass)
+            session = await async_client_session(self.hass)
             client = GeneracApiClient(username, password, session)
             await client.async_get_data()
             return None

custom_components/generac/utils.py

@@ -0,0 +1,14 @@
+"""Helper to create an aiohttp client session for the Generac API."""
+
+from aiohttp import ClientSession, CookieJar
+
+from homeassistant.core import HomeAssistant
+from homeassistant.helpers import aiohttp_client
+
+
+async def async_client_session(hass: HomeAssistant) -> ClientSession:
+    """Return a new aiohttp session."""
+    return aiohttp_client.async_create_clientsession(
+        hass, cookie_jar=CookieJar(unsafe=True, quote_cookie=False)
+    )
+    

local_test.py

@@ -3,30 +3,28 @@
 import json
 import logging
 import os
-
 import aiohttp
 from custom_components.generac.api import GeneracApiClient
 
 logging.basicConfig(level=logging.DEBUG)
-
-jar = aiohttp.CookieJar(unsafe=True)
-
-
+# Your custom JSON encoder
 class EnhancedJSONEncoder(json.JSONEncoder):
     def default(self, o):
         if dataclasses.is_dataclass(o):
             return dataclasses.asdict(o)
         return super().default(o)
 
-
+# Async main logic
 async def main():
+    jar = aiohttp.CookieJar(unsafe=True,quote_cookie=False)
     async with aiohttp.ClientSession(cookie_jar=jar) as session:
         api = GeneracApiClient(
             os.environ["GENERAC_USER"], os.environ["GENERAC_PASS"], session
         )
         await api.login()
-        print(json.dumps(await api.get_device_data(), cls=EnhancedJSONEncoder))
-
+        device_data = await api.get_device_data()
+        print(json.dumps(device_data, cls=EnhancedJSONEncoder))
 
-loop = asyncio.get_event_loop()
-loop.run_until_complete(main())
+# Run it using asyncio.run (preferred method in Python 3.7+)
+if __name__ == "__main__":
+    asyncio.run(main())

requirements.txt

@@ -1,7 +1,7 @@
-aiohttp==3.11.18
 beautifulsoup4==4.13.4
 dacite==1.9.2
 homeassistant==2025.3.3
 voluptuous==0.15.2
 pre-commit==4.2.0
 reorder-python-imports==3.14.0
+Brotli==1.1.0