X-Git-Url: http://git.onelab.eu/?a=blobdiff_plain;f=planetstack%2Fcore%2Fmodels%2Fplcorebase.py;h=4ac60b391575ce5338a1b7dc83954c424aa1a76e;hb=6bdf86d7b248c722134ab8fbb6e084cae4496fcd;hp=2f7fd82661c08bddffefc23910804bb87a6f0cab;hpb=3089d83e7699efa0d302a882e1afa0bdc892cc92;p=plstackapi.git diff --git a/planetstack/core/models/plcorebase.py b/planetstack/core/models/plcorebase.py index 2f7fd82..4ac60b3 100644 --- a/planetstack/core/models/plcorebase.py +++ b/planetstack/core/models/plcorebase.py @@ -5,6 +5,7 @@ from django.db import models from django.forms.models import model_to_dict from django.core.urlresolvers import reverse from django.forms.models import model_to_dict +import model_policies try: # This is a no-op if observer_disabled is set to 1 in the config file @@ -18,16 +19,24 @@ except: def notify_observer(*args, **kwargs): pass +# This manager will be inherited by all subclasses because +# the core model is abstract. +class PlCoreBaseDeletionManager(models.Manager): + def get_query_set(self): + return super(PlCoreBaseDeletionManager, self).get_query_set().filter(deleted=True) + # This manager will be inherited by all subclasses because # the core model is abstract. class PlCoreBaseManager(models.Manager): def get_query_set(self): return super(PlCoreBaseManager, self).get_query_set().filter(deleted=False) - # default values for created and updated are only there to keep evolution - # from failing. class PlCoreBase(models.Model): objects = PlCoreBaseManager() + deleted_objects = PlCoreBaseDeletionManager() + + # default values for created and updated are only there to keep evolution + # from failing. created = models.DateTimeField(auto_now_add=True, default=datetime.datetime.now()) updated = models.DateTimeField(auto_now=True, default=datetime.datetime.now()) enacted = models.DateTimeField(null=True, default=None) @@ -72,20 +81,24 @@ class PlCoreBase(models.Model): def delete(self, *args, **kwds): # so we have something to give the observer - pk = self.pk - model_dict = model_to_dict(self) - for (k,v) in model_dict.items(): - # things like datetime are not JSON serializable - model_dict[k] = str(v) + purge = kwds.get('purge',False) + try: + purge = purge or observer_disabled + except NameError: + pass + + if (purge): + del kwds['purge'] + super(PlCoreBase, self).delete(*args, **kwds) + else: + self.deleted = True + self.enacted=None + self.save(update_fields=['enacted','deleted']) - super(PlCoreBase, self).delete(*args, **kwds) - - # This is a no-op if observer_disabled is set - notify_observer(model=self, delete=True, pk=pk, model_dict=model_dict) def save(self, *args, **kwargs): super(PlCoreBase, self).save(*args, **kwargs) - + # This is a no-op if observer_disabled is set notify_observer()