diff options
author | Kenn Knowles <kenn@kennknowles.com> | 2016-10-04 20:55:55 -0700 |
---|---|---|
committer | Kenn Knowles <kenn@kennknowles.com> | 2016-10-04 20:55:55 -0700 |
commit | 3a194faf81e79f44997f28d76c554b3ca7bd3b64 (patch) | |
tree | de4ffc4b4131f1391a78fd137217a5a06d2ce507 | |
parent | eb23b1e18426af5d429be7a0985002416b13e72d (diff) | |
parent | 270fcbf99cc5770fe545f12fa93d4930d3929481 (diff) | |
download | jsonpath-rw-3a194faf81e79f44997f28d76c554b3ca7bd3b64.tar.gz |
Merge #28
-rw-r--r-- | jsonpath_rw/jsonpath.py | 59 | ||||
-rw-r--r-- | tests/test_jsonpath.py | 62 |
2 files changed, 120 insertions, 1 deletions
diff --git a/jsonpath_rw/jsonpath.py b/jsonpath_rw/jsonpath.py index 3c491d0..93d7b81 100644 --- a/jsonpath_rw/jsonpath.py +++ b/jsonpath_rw/jsonpath.py @@ -26,7 +26,11 @@ class JSONPath(object): raise NotImplementedError() def update(self, data, val): - "Returns `data` with the specified path replaced by `val`" + """ + Returns `data` with the specified path replaced by `val`. Only updates + if the specified path exists. + """ + raise NotImplementedError() def child(self, child): @@ -227,6 +231,11 @@ class Child(JSONPath): if not isinstance(subdata, AutoIdForDatum) for submatch in self.right.find(subdata)] + def update(self, data, val): + for datum in self.left.find(data): + self.right.update(datum.value, val) + return data + def __eq__(self, other): return isinstance(other, Child) and self.left == other.left and self.right == other.right @@ -274,6 +283,11 @@ class Where(JSONPath): def find(self, data): return [subdata for subdata in self.left.find(data) if self.right.find(subdata)] + def update(self, data, val): + for datum in self.find(data): + datum.path.update(data, val) + return data + def __str__(self): return '%s where %s' % (self.left, self.right) @@ -329,6 +343,33 @@ class Descendants(JSONPath): def is_singular(): return False + def update(self, data, val): + # Get all left matches into a list + left_matches = self.left.find(data) + if not isinstance(left_matches, list): + left_matches = [left_matches] + + def update_recursively(data): + # Update only mutable values corresponding to JSON types + if not (isinstance(data, list) or isinstance(data, dict)): + return + + self.right.update(data, val) + + # Manually do the * or [*] to avoid coercion and recurse just the right-hand pattern + if isinstance(data, list): + for i in range(0, len(data)): + update_recursively(data[i]) + + elif isinstance(data, dict): + for field in data.keys(): + update_recursively(data[field]) + + for submatch in left_matches: + update_recursively(submatch.value) + + return data + def __str__(self): return '%s..%s' % (self.left, self.right) @@ -415,6 +456,12 @@ class Fields(JSONPath): for field_datum in [self.get_field_datum(datum, field) for field in self.reified_fields(datum)] if field_datum is not None] + def update(self, data, val): + for field in self.reified_fields(DatumInContext.wrap(data)): + if field in data: + data[field] = val + return data + def __str__(self): return ','.join(map(str, self.fields)) @@ -445,6 +492,11 @@ class Index(JSONPath): else: return [] + def update(self, data, val): + if len(data) > self.index: + data[self.index] = val + return data + def __eq__(self, other): return isinstance(other, Index) and self.index == other.index @@ -495,6 +547,11 @@ class Slice(JSONPath): else: return [DatumInContext(datum.value[i], path=Index(i), context=datum) for i in range(0, len(datum.value))[self.start:self.end:self.step]] + def update(self, data, val): + for datum in self.find(data): + datum.path.update(data, val) + return data + def __str__(self): if self.start == None and self.end == None and self.step == None: return '[*]' diff --git a/tests/test_jsonpath.py b/tests/test_jsonpath.py index bf3d5c6..2c01992 100644 --- a/tests/test_jsonpath.py +++ b/tests/test_jsonpath.py @@ -290,3 +290,65 @@ class TestJsonPath(unittest.TestCase): } }, ['foo.baz', 'foo.bing.baz'] )]) + + def check_update_cases(self, test_cases): + for original, expr_str, value, expected in test_cases: + print('parse(%r).update(%r, %r) =?= %r' + % (expr_str, original, value, expected)) + expr = parse(expr_str) + actual = expr.update(original, value) + assert actual == expected + + def test_update_root(self): + self.check_update_cases([ + ('foo', '$', 'bar', 'bar') + ]) + + def test_update_this(self): + self.check_update_cases([ + ('foo', '`this`', 'bar', 'bar') + ]) + + def test_update_fields(self): + self.check_update_cases([ + ({'foo': 1}, 'foo', 5, {'foo': 5}), + ({'foo': 1, 'bar': 2}, '$.*', 3, {'foo': 3, 'bar': 3}) + ]) + + def test_update_child(self): + self.check_update_cases([ + ({'foo': 'bar'}, '$.foo', 'baz', {'foo': 'baz'}), + ({'foo': {'bar': 1}}, 'foo.bar', 'baz', {'foo': {'bar': 'baz'}}) + ]) + + def test_update_where(self): + self.check_update_cases([ + ({'foo': {'bar': {'baz': 1}}, 'bar': {'baz': 2}}, + '*.bar where baz', 5, {'foo': {'bar': 5}, 'bar': {'baz': 2}}) + ]) + + def test_update_descendants_where(self): + self.check_update_cases([ + ({'foo': {'bar': 1, 'flag': 1}, 'baz': {'bar': 2}}, + '(* where flag) .. bar', 3, + {'foo': {'bar': 3, 'flag': 1}, 'baz': {'bar': 2}}) + ]) + + def test_update_descendants(self): + self.check_update_cases([ + ({'somefield': 1}, '$..somefield', 42, {'somefield': 42}), + ({'outer': {'nestedfield': 1}}, '$..nestedfield', 42, {'outer': {'nestedfield': 42}}), + ({'outs': {'bar': 1, 'ins': {'bar': 9}}, 'outs2': {'bar': 2}}, + '$..bar', 42, + {'outs': {'bar': 42, 'ins': {'bar': 42}}, 'outs2': {'bar': 42}}) + ]) + + def test_update_index(self): + self.check_update_cases([ + (['foo', 'bar', 'baz'], '[0]', 'test', ['test', 'bar', 'baz']) + ]) + + def test_update_slice(self): + self.check_update_cases([ + (['foo', 'bar', 'baz'], '[0:2]', 'test', ['test', 'test', 'baz']) + ]) |