X-Git-Url: http://git.onelab.eu/?a=blobdiff_plain;f=planetstack%2Fcore%2Fmodels%2Fplcorebase.py;h=4b63062d3c98a4c269745e3687f89c9cbc206a5d;hb=451c4fc73c5fbb698af6e67e9488c1e7d167d209;hp=51049a487dc249a92212f012ceeb383bf44517e7;hpb=e3ab5c3450f682b1315726fce8add6f5eddd872f;p=plstackapi.git diff --git a/planetstack/core/models/plcorebase.py b/planetstack/core/models/plcorebase.py index 51049a4..4b63062 100644 --- a/planetstack/core/models/plcorebase.py +++ b/planetstack/core/models/plcorebase.py @@ -7,6 +7,7 @@ from django.forms.models import model_to_dict from django.utils import timezone from django.core.exceptions import PermissionDenied import model_policy +from model_autodeletion import ephemeral_models try: # This is a no-op if observer_disabled is set to 1 in the config file @@ -76,25 +77,90 @@ class DiffModelMixIn: def changed_fields(self): return self.diff.keys() - @property def has_field_changed(self, field_name): return field_name in self.diff.keys() def get_field_diff(self, field_name): return self.diff.get(field_name, None) - -class PlCoreBase(models.Model, DiffModelMixIn): + #classmethod + def getValidators(cls): + """ primarily for REST API, return a dictionary of field names mapped + to lists of the type of validations that need to be applied to + those fields. + """ + validators = {} + for field in cls._meta.fields: + l = [] + if field.blank==False: + l.append("notBlank") + if field.__class__.__name__=="URLField": + l.append("url") + validators[field.name] = l + return validators + +class PlCoreBase(models.Model): # , DiffModelMixIn): objects = PlCoreBaseManager() deleted_objects = PlCoreBaseDeletionManager() + # ---- copy stuff from DiffModelMixin ---- + + # XXX Django fails miserably when trying to create initial migrations when + # DiffModelMixin is used. So, until we figure out what's wrong, + # just copied the guts of DiffModelMixIn here. + + @property + def _dict(self): + return model_to_dict(self, fields=[field.name for field in + self._meta.fields]) + + @property + def diff(self): + d1 = self._initial + d2 = self._dict + diffs = [(k, (v, d2[k])) for k, v in d1.items() if v != d2[k]] + return dict(diffs) + + @property + def has_changed(self): + return bool(self.diff) + + @property + def changed_fields(self): + return self.diff.keys() + + def has_field_changed(self, field_name): + return field_name in self.diff.keys() + + def get_field_diff(self, field_name): + return self.diff.get(field_name, None) + + #classmethod + def getValidators(cls): + """ primarily for REST API, return a dictionary of field names mapped + to lists of the type of validations that need to be applied to + those fields. + """ + validators = {} + for field in cls._meta.fields: + l = [] + if field.blank==False: + l.append("notBlank") + if field.__class__.__name__=="URLField": + l.append("url") + validators[field.name] = l + return validators + + # ---- end copy stuff from DiffModelMixin ---- + # default values for created and updated are only there to keep evolution # from failing. created = models.DateTimeField(auto_now_add=True, default=timezone.now) updated = models.DateTimeField(auto_now=True, default=timezone.now) - enacted = models.DateTimeField(null=True, default=None) + enacted = models.DateTimeField(null=True, blank=True, default=None) + policed = models.DateTimeField(null=True, blank=True, default=None) backend_status = models.CharField(max_length=140, - default="Provisioning in progress") + default="0 - Provisioning in progress") deleted = models.BooleanField(default=False) class Meta: @@ -113,24 +179,23 @@ class PlCoreBase(models.Model, DiffModelMixIn): return False if user.is_admin: return True - return False - def can_update_field(self, user, fieldName): - # Give us the opportunity to implement fine-grained permission checking. - # Default to True, and let can_update() permit or deny the whole object. - return True + return False def delete(self, *args, **kwds): # so we have something to give the observer purge = kwds.get('purge',False) + if purge: + del kwds['purge'] silent = kwds.get('silent',False) + if silent: + del kwds['silent'] try: purge = purge or observer_disabled except NameError: pass if (purge): - del kwds['purge'] super(PlCoreBase, self).delete(*args, **kwds) else: self.deleted = True @@ -149,15 +214,14 @@ class PlCoreBase(models.Model, DiffModelMixIn): if not silent: notify_observer() - self.__initial = self._dict + self._initial = self._dict def save_by_user(self, user, *args, **kwds): if not self.can_update(user): - raise PermissionDenied("You do not have permission to update %s objects" % self.__class__.__name__) - - for fieldName in self.changed_fields: - if not self.can_update_field(user, fieldName): - raise PermissionDenied("You do not have permission to update field %s in object %s" % (fieldName, self.__class__.__name__)) + if getattr(self, "_cant_update_fieldName", None) is not None: + raise PermissionDenied("You do not have permission to update field %s on object %s" % (self._cant_update_fieldName, self.__class__.__name__)) + else: + raise PermissionDenied("You do not have permission to update %s objects" % self.__class__.__name__) self.save(*args, **kwds) @@ -166,6 +230,18 @@ class PlCoreBase(models.Model, DiffModelMixIn): raise PermissionDenied("You do not have permission to delete %s objects" % self.__class__.__name__) self.delete(*args, **kwds) + @classmethod + def select_by_user(cls, user): + # This should be overridden by descendant classes that want to perform + # filtering of visible objects by user. + return cls.objects.all() + + @classmethod + def is_ephemeral(cls): + return cls in ephemeral_models + + +