Make fetching zones optional

pydrawise/base.py

@@ -10,9 +10,10 @@ class HydrawiseBase(ABC):
     """Base class for Hydrawise client APIs."""
 
     @abstractmethod
-    async def get_user(self) -> User:
+    async def get_user(self, fetch_zones: bool = True) -> User:
         """Retrieves the currently authenticated user.
 
+        :param fetch_zones: When True, also fetch zones.
         :rtype: User
         """
 

pydrawise/client.py

@@ -75,9 +75,10 @@ async def _mutation(self, selector: DSLField) -> None:
                 # Assume bool response
                 raise MutationError
 
-    async def get_user(self) -> User:
+    async def get_user(self, fetch_zones: bool = True) -> User:
         """Retrieves the currently authenticated user.
 
+        :param fetch_zones: Not used in this implementation.
         :rtype: User
         """
         selector = self._schema.Query.me.select(*get_selectors(self._schema, User))

pydrawise/legacy.py

@@ -47,9 +47,10 @@ async def _get(self, path: str, **kwargs) -> dict:
             async with session.get(url, params=params, timeout=_TIMEOUT) as resp:
                 return await resp.json()
 
-    async def get_user(self) -> User:
+    async def get_user(self, fetch_zones: bool = True) -> User:
         """Retrieves the currently authenticated user.
 
+        :param fetch_zones: When True, also fetch zones.
         :rtype: User
         """
         resp_json = await self._get("customerdetails.php")
@@ -60,8 +61,9 @@ async def get_user(self) -> User:
             email="",
             controllers=[_controller_from_json(c) for c in resp_json["controllers"]],
         )
-        for controller in user.controllers:
-            controller.zones = await self.get_zones(controller)
+        if fetch_zones:
+            for controller in user.controllers:
+                controller.zones = await self.get_zones(controller)
         return user
 
     async def get_controllers(self) -> list[Controller]: