New Features (nested_update, nested_delete) (#12)

* Fix for issue #10

* Renamed the test module

* New Features (nested_get, nested_update, nested_delete)

* Update Version to 0.3.0

* Updated Readme for new features

* Removed method (nested_get)

* Addressed review comments

* Removed nested_get references from README

* Rearanged the order of the functions

nested_lookup/__init__.py

@@ -1,2 +1,3 @@
 from .nested_lookup import nested_lookup, get_all_keys, get_occurrence_of_key,\
     get_occurrence_of_value
+from .lookup_api import nested_update, nested_delete

nested_lookup/lookup_api.py

@@ -0,0 +1,54 @@
+import copy
+from six import iteritems
+
+
+def nested_delete(document, key):
+    duplicate = copy.deepcopy(document)
+    return _nested_delete(document=duplicate, key=key)
+
+
+def _nested_delete(document, key):
+    """
+    Method to delete a key->value pair from a nested document
+    Args:
+        document: Might be List of Dicts (or) Dict of Lists (or)
+         Dict of List of Dicts etc...
+        key: Key to delete
+    Return:
+        Returns a document that includes everything but the given key
+    """
+    if isinstance(document, list):
+        for list_items in document:
+            _nested_delete(document=list_items, key=key)
+    elif isinstance(document, dict):
+        if document.get(key):
+            del document[key]
+        for dict_key, dict_value in iteritems(document):
+            _nested_delete(document=dict_value, key=key)
+    return document
+
+
+def nested_update(document, key, value):
+    duplicate = copy.deepcopy(document)
+    return _nested_update(document=duplicate, key=key, value=value)
+
+
+def _nested_update(document, key, value):
+    """
+    Method to update a key->value pair in a nested document
+    Args:
+        document: Might be List of Dicts (or) Dict of Lists (or)
+         Dict of List of Dicts etc...
+        key: Key to update the value
+    Return:
+        Returns a document that has updated key, value pair.
+    """
+    if isinstance(document, list):
+        for list_items in document:
+            _nested_update(document=list_items, key=key, value=value)
+    elif isinstance(document, dict):
+        if document.get(key):
+            document[key] = value
+        for dict_key, dict_value in iteritems(document):
+            _nested_update(document=dict_value, key=key, value=value)
+    return document

nested_lookup/nested_lookup.py

@@ -7,7 +7,9 @@ def nested_lookup(key, document, wild=False, with_keys=False):
     """Lookup a key in a nested document, return a list of values"""
     if with_keys:
         d = defaultdict(list)
-        for k, v in _nested_lookup(key, document, wild=wild, with_keys=with_keys):
+        for k, v in _nested_lookup(
+            key, document, wild=wild, with_keys=with_keys
+        ):
             d[k].append(v)
         return d
     return list(_nested_lookup(key, document, wild=wild, with_keys=with_keys))
@@ -17,7 +19,9 @@ def _nested_lookup(key, document, wild=False, with_keys=False):
     """Lookup a key in a nested document, yield a value"""
     if isinstance(document, list):
         for d in document:
-            for result in _nested_lookup(key, d, wild=wild, with_keys=with_keys):
+            for result in _nested_lookup(
+                key, d, wild=wild, with_keys=with_keys
+            ):
                 yield result
 
     if isinstance(document, dict):
@@ -28,7 +32,9 @@ def _nested_lookup(key, document, wild=False, with_keys=False):
                 else:
                     yield v
             if isinstance(v, dict):
-                for result in _nested_lookup(key, v, wild=wild, with_keys=with_keys):
+                for result in _nested_lookup(
+                    key, v, wild=wild, with_keys=with_keys
+                ):
                     yield result
             elif isinstance(v, list):
                 for d in v: