Fix ID retrieval

Author: kzscisoft

simvue/api/objects/base.py

@@ -310,10 +310,10 @@ def ids(
         """Retrieve a list of all object identifiers"""
         _class_instance = cls(_read_only=True, _local=True)
         _count: int = 0
-        for _data in cls._get_all_objects(offset):
+        for response in cls._get_all_objects(offset):
             if count and _count > count:
                 return
-            if _data.get("data") is None:
+            if (_data := response.get("data")) is None:
                 raise RuntimeError(
                     f"Expected key 'data' for retrieval of {_class_instance.__class__.__name__.lower()}s"
                 )

simvue/client.py

@@ -333,7 +333,10 @@ def _get_folder_id_from_path(self, path: str) -> str | None:
         """
         _ids = Folder.ids(filters=json.dumps([f"path == {path}"]))
 
-        return _ids[0] if _ids else None
+        try:
+            return next(_ids)
+        except StopIteration:
+            return None
 
     @prettify_pydantic
     @pydantic.validate_call