Catch ClientError and raise CannotConnect

tronikos · Oct 11, 2024 · f015bf3 · f015bf3
1 parent c5d50cd
commit f015bf3
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [project]
 name = "opower"
-version = "0.8.2"
+version = "0.8.3"
 license = {text = "Apache-2.0"}
 authors = [
     { name="tronikos", email="tronikos@gmail.com" },

diff --git a/src/opower/opower.py b/src/opower/opower.py
@@ -9,7 +9,7 @@
 from urllib.parse import urlencode
 
 import aiohttp
-from aiohttp.client_exceptions import ClientResponseError
+from aiohttp.client_exceptions import ClientError, ClientResponseError
 import aiozoneinfo
 import arrow
 
@@ -205,6 +205,8 @@ async def async_login(self) -> None:
                 raise InvalidAuth(err)
             else:
                 raise CannotConnect(err)
+        except ClientError as err:
+            raise CannotConnect(err)
 
     async def async_get_accounts(self) -> list[Account]:
         """Get a list of accounts for the signed in user.