Add optional API URL to connect functions (#156)

fustom · May 21, 2024 · 6f01404 · 6f01404
1 parent a21c6ae
commit 6f01404
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/ariston/__init__.py b/ariston/__init__.py
@@ -114,9 +114,9 @@ def _get_device(
             return None
 
 
-def _connect(username: str, password: str) -> AristonAPI:
+def _connect(username: str, password: str, api_url: str = ARISTON_API_URL) -> AristonAPI:
     """Connect to ariston api"""
-    api = AristonAPI(username, password)
+    api = AristonAPI(username, password, api_url)
     api.connect()
     return api
 
@@ -149,9 +149,9 @@ def hello(
     return _get_device(cloud_devices, api, gateway, is_metric, language_tag)
 
 
-async def _async_connect(username: str, password: str) -> AristonAPI:
+async def _async_connect(username: str, password: str, api_url: str = ARISTON_API_URL) -> AristonAPI:
     """Async connect to ariston api"""
-    api = AristonAPI(username, password)
+    api = AristonAPI(username, password, api_url)
     if not await api.async_connect():
         raise ConnectionException
     return api