implemented items, to_object and zip functions

Author: Jonathan Stewmon

jmespath/functions.py

@@ -2,6 +2,9 @@
 import json
 
 from jmespath import exceptions
+from jmespath.compat import get_methods
+from jmespath.compat import iteritems
+from jmespath.compat import map
 from jmespath.compat import string_type as STRING_TYPE
 from jmespath.compat import get_methods
 
@@ -185,6 +188,10 @@ def _func_to_array(self, arg):
         else:
             return [arg]
 
+    @signature({'types': ['array']})
+    def _func_to_object(self, pairs):
+        return dict(pairs)
+
     @signature({'types': []})
     def _func_to_string(self, arg):
         if isinstance(arg, STRING_TYPE):
@@ -281,6 +288,10 @@ def _func_sort(self, arg):
     def _func_sum(self, arg):
         return sum(arg)
 
+    @signature({'types': ['object']})
+    def _func_items(self, arg):
+        return list(map(list, iteritems(arg)))
+
     @signature({"types": ['object']})
     def _func_keys(self, arg):
         # To be consistent with .values()
@@ -346,6 +357,10 @@ def _func_max_by(self, array, expref):
         else:
             return None
 
+    @signature({'types': ['array'], 'variadic': True})
+    def _func_zip(self, *arguments):
+        return list(map(list, zip(*arguments)))
+
     def _create_key_func(self, expref, allowed_types, function_name):
         def keyfunc(x):
             result = expref.visit(expref.expression, x)

tests/compliance/functions.json

@@ -12,6 +12,7 @@
     "empty_list": [],
     "empty_hash": {},
     "objects": {"foo": "bar", "bar": "baz"},
+    "pairs": [["a", "first"], ["b", "second"], ["c", "third"]],
     "null_key": null
   },
   "cases": [
@@ -175,6 +176,18 @@
       "expression": "floor(str)",
       "error": "invalid-type"
     },
+    {
+      "expression": "items(objects)",
+      "result": [["foo", "bar"], ["bar", "baz"]]
+    },
+    {
+      "expression": "items(empty_hash)",
+      "result": []
+    },
+    {
+      "expression": "items(numbers)",
+      "error": "invalid-type"
+    },
     {
       "expression": "length('abc')",
       "result": 3
@@ -189,7 +202,7 @@
     },
     {
       "expression": "length(@)",
-      "result": 12
+      "result": 13
     },
     {
       "expression": "length(strings[0])",
@@ -479,6 +492,10 @@
       "expression": "to_array(false)",
       "result": [false]
     },
+    {
+      "expression": "to_object(pairs)",
+      "result": {"a": "first", "b": "second", "c": "third"}
+    },
     {
       "expression": "to_string('foo')",
       "result": "foo"
@@ -588,6 +605,18 @@
       "comment": "function projection on single arg function",
       "expression": "array[].to_number(@)",
       "result": [-1, 3, 4, 5, 100]
+    },
+    {
+      "expression": "zip(strings, numbers)",
+      "result": [["a", -1], ["b", 3], ["c", 4]]
+    },
+    {
+      "expression": "zip(strings, numbers, decimals)",
+      "result": [["a", -1, 1.01], ["b", 3, 1.2], ["c", 4, -1.5]]
+    },
+    {
+      "expression": "zip(str)",
+      "error": "invalid-type"
     }
   ]
 }, {