add jmesh query

Author: patrick

src/qpystructs/operations.py

@@ -112,13 +112,14 @@ def parse_file_as(file_path: Path | str, to_type: T, data_format=DataFormatType.
 
 
 def get_value(target_object: str | dict | BaseModel | GenericDataModel, path_exp: str,
-              ) -> Any:
+              **kwargs) -> Any:
     if isinstance(target_object, Dict):
         return jmespath.search(expression=path_exp, data=target_object)
     if isinstance(target_object, str):
         return jmespath.search(data=json.loads(target_object), expression=path_exp)
     if isinstance(target_object, BaseModel):
         return jmespath.search(data=target_object.dict(), expression=path_exp)
+
     raise NotImplementedError("not support type " + type(target_object))
 
 

tests/structured-data/query-data.json

@@ -0,0 +1,8 @@
+{
+  "locations": [
+    {"name": "Seattle", "state": "WA"},
+    {"name": "New York", "state": "NY"},
+    {"name": "Bellevue", "state": "WA"},
+    {"name": "Olympia", "state": "WA"}
+  ]
+}

tests/test_operations.py

@@ -27,3 +27,12 @@ def test_differ():
     result = qpystructs.differ(more_dict, more_json_dict)
     print(result)
     assert len(result) > 1
+
+
+def test_query_expression():
+    query = "locations[?state == 'WA'].name | sort(@) | {WashingtonCities: join(', ', @)}"
+    result = qpystructs.load_from_file("./structured-data/query-data.json")
+    query_result = qpystructs.get_value(result, query)
+    assert query_result == {
+        "WashingtonCities": "Bellevue, Olympia, Seattle"
+    }