adding support for lambda/callables to the new update function

jonathanglima · Jul 5, 2016 · 8f66e60 · 8f66e60
1 parent 0f8957f
commit 8f66e60
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/jsonpath_rw/jsonpath.py b/jsonpath_rw/jsonpath.py
@@ -432,7 +432,10 @@ def find(self, datum):
 
     def update(self, data, val):
         for field in self.reified_fields(DatumInContext.wrap(data)):
-            data[field] = val
+            if hasattr(val, '__call__'):
+                data[field] = val.__call__(data[field])
+            else:
+                data[field] = val
         return data
 
     def __str__(self):
@@ -466,7 +469,10 @@ def find(self, datum):
             return []
 
     def update(self, data, val):
-        data[self.index] = val
+        if hasattr(val, '__call__'):
+            data[self.index] = val.__call__(data[self.index])
+        else:
+            data[self.index] = val
         return data
 
     def __eq__(self, other):