Merge pull request h2non#13 from dcreemer/master

add str() action to convert numeric values to string

Author: h2non

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,14 +17,15 @@
 
 SUB = re.compile("sub\(/(.*)/,\s+(.*)\)")
 SPLIT = re.compile("split\((.),\s+(\d+),\s+(\d+|-1)\)")
+STR = re.compile("str\(\)")
 
 
 class DefintionInvalid(Exception):
     pass
 
 
 class Sub(This):
-    """Regex subtituor
+    """Regex substituor
 
     Concrete syntax is '`sub(/regex/, repl)`'
     """
@@ -37,7 +38,6 @@ def __init__(self, method=None):
         self.repl = m.group(2).strip()
         self.regex = re.compile(self.expr)
         self.method = method
-        print("%r" % self)
 
     def find(self, datum):
         datum = DatumInContext.wrap(datum)
@@ -81,10 +81,37 @@ def find(self, datum):
         return [DatumInContext.wrap(value)]
 
     def __eq__(self, other):
-        return (isinstance(other, Sub) and self.method == other.method)
+        return (isinstance(other, Split) and self.method == other.method)
 
     def __repr__(self):
         return '%s(%r)' % (self.__class__.__name__, self.method)
 
     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"},