Allow raw numeric values to be used as keys

Fixes #123
h2non · Oct 10, 2024 · 1b83307 · 1b83307
1 parent 58378a6
commit 1b83307
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 1 deletion.
diff --git a/jsonpath_ng/parser.py b/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]
 

diff --git a/tests/test_jsonpath.py b/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"]),
 )
 
 

diff --git a/tests/test_parser.py b/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)),