Browse Source

Added the management of allowed_classes in Reference

Roland Haroutiounian 8 years ago
parent
commit
1c1d353719
1 changed files with 4 additions and 1 deletions
  1. 4
    1
      lodel/leapi/datahandlers/reference.py

+ 4
- 1
lodel/leapi/datahandlers/reference.py View File

20
     ## @brief checks if the data value is valid
20
     ## @brief checks if the data value is valid
21
     # @param value
21
     # @param value
22
     # @return
22
     # @return
23
-    def check_data_value(self, value):
23
+    def _check_data_value(self, value):
24
 
24
 
25
         if not isinstance(value, self._refs_class):
25
         if not isinstance(value, self._refs_class):
26
             return value, "The reference should be an instance of %s, %s gotten" % (self._refs_class, value.__class__)
26
             return value, "The reference should be an instance of %s, %s gotten" % (self._refs_class, value.__class__)
35
             if not isinstance(related, EmClass):
35
             if not isinstance(related, EmClass):
36
                 return value, "The reference %s should be an instance of EmClass, %s gotten" % (related.display_name,
36
                 return value, "The reference %s should be an instance of EmClass, %s gotten" % (related.display_name,
37
                                                                                                 related.__class__)
37
                                                                                                 related.__class__)
38
+            if self.allowed_classes is not None and related.__class__.display_name not in self.allowed_classes:
39
+                return value, "The reference %s should be an instance of either one of those classes : %s, %s gotten" % \
40
+                              (related.display_name, self.allowed_classes, related.__class__)
38
         return value, None
41
         return value, None

Loading…
Cancel
Save