fill_default ands tests

Author: Tariq Daouda

CHANGELOG.rst

@@ -1,3 +1,9 @@
+2.1.0
+=====
+* Added getitem for documents at the database level
+* Added fill_default() on documents to replace None values by schema defaults
+* fill_default() is automatically called on save
+
 2.0.2
 =====
 * Fixed contains functions

pyArango/document.py

@@ -108,12 +108,7 @@ def validate(self):
         return True
 
     def set(self, dct):
-        """Set the store using a dictionary"""
-        # if not self.mustValidate:
-        #     self.store = dct
-        #     self.patchStore = dct
-        #     return
-
+        """Set the values to a dict. Any missing value will be filled by it's default"""
         for field, value in dct.items():
             if field not in self.collection.arangoPrivates:
                 if isinstance(value, dict):
@@ -126,6 +121,14 @@ def set(self, dct):
                 else:
                     self[field] = value
 
+    def fill_default(self):
+        """replace all None values with defaults"""
+        for field, value in self.validators.items():
+            if isinstance(value, dict):
+                self[field].fill_default()
+            elif self[field] is None:
+                self[field] = value.default
+ 
     def __dir__(self):
         return dir(self.getStore())
 
@@ -234,6 +237,10 @@ def to_default(self):
         """reset the document to the default values"""
         self.reset(self.collection, self.collection.getDefaultDocument())
 
+    def fill_default(self):
+        """reset the document to the default values"""
+        self._store.fill_default()
+
     def validate(self):
         """validate the document"""
         self._store.validate()
@@ -264,7 +271,9 @@ def save(self, waitForSync = False, **docArgs):
         If you want to only update the modified fields use the .patch() function.
         Use docArgs to put things such as 'waitForSync = True' (for a full list cf ArangoDB's doc).
         It will only trigger a saving of the document if it has been modified since the last save. If you want to force the saving you can use forceSave()"""
+        self._store.fill_default()
         payload = self._store.getStore()
+        # print(payload)
         self._save(payload, waitForSync = False, **docArgs)
 
     def _save(self, payload, waitForSync = False, **docArgs):