added on load

Author: tariqdaouda

pyArango/collection.py

@@ -132,6 +132,7 @@ class Collection_metaclass(type) :
     _validationDefault = {
             'on_save' : False,
             'on_set' : False,
+            'on_load' : False,
             'allow_foreign_fields' : True
         }
 
@@ -217,6 +218,7 @@ class Collection(with_metaclass(Collection_metaclass, object)) :
     _validation = {
         'on_save' : False,
         'on_set' : False,
+        'on_load' : False,
         'allow_foreign_fields' : True
     }
 

pyArango/document.py

@@ -23,7 +23,11 @@ def reset(self, collection, jsonFieldInit = {}) :
         self._id, self._rev, self._key = None, None, None
         self.URL = None
 
-        self.set(jsonFieldInit)
+        if not self.collection._validation['on_load']:
+            self.set(jsonFieldInit, validate=False)
+        else :
+            self.set(jsonFieldInit, validate=True)
+
         self.modified = True
 
     def setPrivates(self, fieldDict) :
@@ -43,12 +47,16 @@ def setPrivates(self, fieldDict) :
             self._id, self._rev, self._key = None, None, None
             self.URL = None
 
-    def set(self, fieldDict = None) :
+    def set(self, fieldDict = None, validate = True) :
         """Sets the document according to values contained in the dictinnary fieldDict. This will also set self._id/_rev/_key"""
 
         if fieldDict and self._id is None :
             self.setPrivates(fieldDict)
 
+        if not validate :
+            self._store.update(fieldDict)
+            return
+
         if self.collection._validation['on_set']:
             for k in list(fieldDict.keys()) :
                 self[k] = fieldDict[k]