Allow delattr and del operations

1 files changed, 12 insertions, 1 deletions

diff --git a/warlock/core.py b/warlock/core.py
index da662ec..38cb854 100644
--- a/warlock/core.py
+++ b/warlock/core.py
@@ -76,7 +76,18 @@ def model_factory(schema):
             raise InvalidOperation()
 
         def __delitem__(self, key):
-            raise InvalidOperation()
+            mutation = dict(self.items())
+            del mutation[key]
+            try:
+                self.validator(mutation)
+            except ValidationError:
+                msg = "Unable to delete attribute '%s'" % (key)
+                raise InvalidOperation(msg)
+
+            dict.__delitem__(self, key)
+
+        def __delattr__(self, key):
+            self.__delitem__(key)
 
         # NOTE(termie): This is kind of the opposite of what copy usually does
         def copy(self):