add str() method

Author: dcreemer

jsonpath_ng/ext/parser.py

@@ -100,6 +100,8 @@ def p_jsonpath_named_operator(self, p):
             p[0] = _string.Split(p[1])
         elif p[1].startswith("sub("):
             p[0] = _string.Sub(p[1])
+        elif p[1].startswith("str("):
+            p[0] = _string.Str(p[1])
         else:
             super(ExtentedJsonPathParser, self).p_jsonpath_named_operator(p)
 

jsonpath_ng/ext/string.py

@@ -17,6 +17,7 @@
 
 SUB = re.compile("sub\(/(.*)/,\s+(.*)\)")
 SPLIT = re.compile("split\((.),\s+(\d+),\s+(\d+|-1)\)")
+STR = re.compile("str\(\)")
 
 
 class DefintionInvalid(Exception):
@@ -87,3 +88,30 @@ def __repr__(self):
 
     def __str__(self):
         return '`%s`' % self.method
+
+
+class Str(This):
+    """String converter
+
+    Concrete syntax is '`str()`'
+    """
+
+    def __init__(self, method=None):
+        m = STR.match(method)
+        if m is None:
+            raise DefintionInvalid("%s is not valid" % method)
+        self.method = method
+
+    def find(self, datum):
+        datum = DatumInContext.wrap(datum)
+        value = str(datum.value)
+        return [DatumInContext.wrap(value)]
+
+    def __eq__(self, other):
+        return (isinstance(other, Str) and self.method == other.method)
+
+    def __repr__(self):
+        return '%s(%r)' % (self.__class__.__name__, self.method)
+
+    def __str__(self):
+        return '`str()`'

tests/test_jsonpath_rw_ext.py

@@ -276,7 +276,11 @@ class Testjsonpath_ng_ext(testscenarios.WithScenarios,
             data={'payload': "foo+bar"},
             target=["repl"]
         )),
-
+        ('str1', dict(
+            string='payload.`str()`',
+            data={'payload': 1},
+            target=["1"]
+        )),
         ('split1', dict(
             string='payload.`split(-, 2, -1)`',
             data={'payload': "foo-bar-cat-bow"},