iterator on objects

Author: Neradoc

adafruit_json_stream.py

@@ -239,6 +239,41 @@ def __getitem__(self, key):
             self.done = self.data.fast_forward(",")
         raise KeyError(key)
 
+    def __iter__(self):
+        return self
+
+    def _next_item(self):
+        """Return the next item as a (key, value) pair, regardless of key."""
+        if self.active_child:
+            self.active_child.finish()
+            self.done = self.data.fast_forward(",")
+            self.active_child = None
+        if self.done:
+            raise StopIteration()
+
+        current_key = self.data.next_value(":")
+        if current_key is None:
+            self.done = True
+            raise StopIteration()
+
+        next_value = self.data.next_value(",")
+        if self.data.last_char == ord("}"):
+            self.done = True
+        if isinstance(next_value, Transient):
+            self.active_child = next_value
+        return (current_key, next_value)
+
+    def __next__(self):
+        return self._next_item()[0]
+
+    def items(self):
+        """Return iterator ine the dictionary’s items ((key, value) pairs)."""
+        try:
+            while not self.done:
+                yield self._next_item()
+        except StopIteration:
+            return
+
 
 def load(data_iter):
     """Returns an object to represent the top level of the given JSON stream."""

tests/test_json_stream.py

@@ -685,3 +685,22 @@ def test_as_object_grabbing_multiple_subscriptable_levels_again_after_passed_rai
     assert next(dict_1["sub_list"]) == "a"
     with pytest.raises(KeyError, match="sub_dict"):
         dict_1["sub_dict"]["sub_dict_name"]
+
+
+def test_iterating_keys(dict_with_keys):
+    """Iterate through keys of a simple object"""
+
+    bytes_io_chunk = BytesChunkIO(dict_with_keys.encode())
+    stream = adafruit_json_stream.load(bytes_io_chunk)
+    output = list(stream)
+    assert output == ["field_1", "field_2", "field_3"]
+
+
+def test_iterating_items(dict_with_keys):
+    """Iterate through items of a simple object"""
+
+    bytes_io_chunk = BytesChunkIO(dict_with_keys.encode())
+    stream = adafruit_json_stream.load(bytes_io_chunk)
+    output = list(stream.items())
+    assert output == [("field_1", 1), ("field_2", 2), ("field_3", 3)]
+