fix: resolve issue #16 support quoted keys in filters and paths

Author: sean2077

jsonpath/jsonpath.py

@@ -118,7 +118,7 @@ def _get_quote(self, m):
         return f"#Q{n}"
 
     def _put_quote(self, m):
-        return self.subx["#Q"][int(m.group(1))]
+        return f"'{self.subx['#Q'][int(m.group(1))]}'"
 
     def _get_backquote(self, m):
         n = len(self.subx["#BQ"])
@@ -146,8 +146,11 @@ def _put_paren(self, m):
 
     @staticmethod
     def _gen_obj(m):
+        content = m.group(1) or m.group(2)  # group 2 is for len()
         ret = "__obj"
-        for e in m.group(1).split("."):
+        for e in content.split("."):
+            if len(e) >= 2 and ((e[0] == "'" and e[-1] == "'") or (e[0] == '"' and e[-1] == '"')):
+                e = e[1:-1]
             ret += f'["{e}"]'
         return ret
 
@@ -239,11 +242,15 @@ def _trace(self, obj, i: int, path):
             return
 
         # get value from dict
-        if isinstance(obj, dict) and step in obj:
-            if re.match(r"^\w+$", step):
-                self._trace(obj[step], i + 1, f"{path}.{step}")
+        step_key = step
+        if len(step) >= 2 and step[0] == "'" and step[-1] == "'":
+            step_key = step[1:-1]
+
+        if isinstance(obj, dict) and step_key in obj:
+            if re.match(r"^\w+$", step_key):
+                self._trace(obj[step_key], i + 1, f"{path}.{step_key}")
             else:
-                self._trace(obj[step], i + 1, f"{path}['{step}']")
+                self._trace(obj[step_key], i + 1, f"{path}['{step_key}']")
             return
 
         # slice

tests/test_issues.py

@@ -25,3 +25,22 @@ def test_issue_17_bracket_dot_normalization():
     data_recursive = {"a": {"store": "book"}, "b": {"store": "paper"}}
     # $..['store'] should find all 'store' keys
     assert sorted(JSONPath("$..['store']").parse(data_recursive)) == ["book", "paper"]
+
+
+def test_issue_16_quoted_keys():
+    data = {"user-list": [{"city-name": "Austin", "name": "John"}, {"city-name": "New York", "name": "Jane"}]}
+
+    # Case 1: Key with hyphen in the path
+    assert JSONPath("$.'user-list'").parse(data) == [
+        [{"city-name": "Austin", "name": "John"}, {"city-name": "New York", "name": "Jane"}]
+    ]
+
+    # Case 2: Key with hyphen in a filter (single quotes)
+    assert JSONPath("$.'user-list'[?(@.'city-name'=='Austin')]").parse(data) == [
+        {"city-name": "Austin", "name": "John"}
+    ]
+
+    # Case 3: Key with hyphen in a filter (double quotes)
+    assert JSONPath('$.\'user-list\'[?(@."city-name"=="Austin")]').parse(data) == [
+        {"city-name": "Austin", "name": "John"}
+    ]