Allow raw numeric values to be used as keys

Fixes #123

Author: michaelmior

jsonpath_ng/parser.py

@@ -149,9 +149,12 @@ def p_jsonpath_parens(self, p):
     # Because fields in brackets cannot be '*' - that is reserved for array indices
     def p_fields_or_any(self, p):
         """fields_or_any : fields
-                         | '*'    """
+                         | '*'
+                         | NUMBER"""
         if p[1] == '*':
             p[0] = ['*']
+        elif isinstance(p[1], int):
+            p[0] = str(p[1])
         else:
             p[0] = p[1]
 

tests/test_jsonpath.py

@@ -276,6 +276,11 @@ def test_update(parse, expression, data, update_value, expected_value):
     # --------
     #
     ("A.'a.c'", {"A": {"a.c": "d"}}, ["d"], ["A.'a.c'"]),
+    #
+    # Numeric keys
+    # --------
+    #
+    ("1", {"1": "foo"}, ["foo"], ["1"]),
 )
 
 

tests/test_parser.py

@@ -12,6 +12,7 @@
     #
     ("foo", Fields("foo")),
     ("*", Fields("*")),
+    ("1", Fields("1")),
     ("baz,bizzle", Fields("baz", "bizzle")),
     ("[1]", Index(1)),
     ("[1:]", Slice(start=1)),