summaryrefslogtreecommitdiff
path: root/jsonpath_rw/jsonpath.py
diff options
context:
space:
mode:
Diffstat (limited to 'jsonpath_rw/jsonpath.py')
-rw-r--r--jsonpath_rw/jsonpath.py29
1 files changed, 29 insertions, 0 deletions
diff --git a/jsonpath_rw/jsonpath.py b/jsonpath_rw/jsonpath.py
index 3c491d0..395516e 100644
--- a/jsonpath_rw/jsonpath.py
+++ b/jsonpath_rw/jsonpath.py
@@ -227,6 +227,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 +279,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 +339,11 @@ class Descendants(JSONPath):
def is_singular():
return False
+ def update(self, data, val):
+ for datum in self.left.find(data):
+ self.right.update(datum.value, val)
+ return data
+
def __str__(self):
return '%s..%s' % (self.left, self.right)
@@ -415,6 +430,11 @@ 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)):
+ data[field] = val
+ return data
+
def __str__(self):
return ','.join(map(str, self.fields))
@@ -445,6 +465,10 @@ class Index(JSONPath):
else:
return []
+ def update(self, data, val):
+ data[self.index] = val
+ return data
+
def __eq__(self, other):
return isinstance(other, Index) and self.index == other.index
@@ -495,6 +519,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 '[*]'