Add CompoundJSONPath.query()

Author: jg-rp

jsonpath/path.py

@@ -432,6 +432,30 @@ async def finditer_async(
 
         return matches
 
+    def query(
+        self,
+        data: Union[str, IOBase, Sequence[Any], Mapping[str, Any]],
+        *,
+        filter_context: Optional[FilterContextVars] = None,
+    ) -> Query:
+        """Return a `Query` iterator over matches found by applying this path to _data_.
+
+        Arguments:
+            data: A JSON document or Python object implementing the `Sequence`
+                or `Mapping` interfaces.
+            filter_context: Arbitrary data made available to filters using
+                the _filter context_ selector.
+
+        Returns:
+            A query iterator.
+
+        Raises:
+            JSONPathSyntaxError: If the path is invalid.
+            JSONPathTypeError: If a filter expression attempts to use types in
+                an incompatible way.
+        """
+        return Query(self.finditer(data, filter_context=filter_context), self.env)
+
     def union(self, path: JSONPath) -> CompoundJSONPath:
         """Union of this path and another path."""
         return self.__class__(

tests/test_fluent_api.py

@@ -276,3 +276,10 @@ def test_query_from_compiled_path() -> None:
     path = compile("$.some.*")
     it = path.query({"some": [0, 1, 2, 3]}).values()
     assert list(it) == [0, 1, 2, 3]
+
+
+def test_query_from_compiled_compound_path() -> None:
+    """Test that we can get a query iterator from a compiled path."""
+    path = compile("$.some[0] | $.some[2]")
+    it = path.query({"some": [0, 1, 2, 3]}).values()
+    assert list(it) == [0, 2]