planetstack/core/models/plcorebase.py

   1 import os
   2 from django.db import models
   3 from django.forms.models import model_to_dict
   4 from django.core.urlresolvers import reverse
   5 from django.forms.models import model_to_dict
   6 # This is a no-op if observer_disabled is set to 1 in the config file
   7 from observer import *
   8
   9 class PlCoreBase(models.Model):
  10
  11     created = models.DateTimeField(auto_now_add=True)
  12     updated = models.DateTimeField(auto_now=True)
  13     enacted = models.DateTimeField(null=True, default=None)
  14
  15     class Meta:
  16         abstract = True
  17         app_label = "core"
  18
  19     def __init__(self, *args, **kwargs):
  20         super(PlCoreBase, self).__init__(*args, **kwargs)
  21         self.__initial = self._dict
  22
  23     @property
  24     def diff(self):
  25         d1 = self.__initial
  26         d2 = self._dict
  27         diffs = [(k, (v, d2[k])) for k, v in d1.items() if v != d2[k]]
  28         return dict(diffs)
  29
  30     @property
  31     def has_changed(self):
  32         return bool(self.diff)
  33
  34     @property
  35     def changed_fields(self):
  36         return self.diff.keys()
  37
  38     def get_field_diff(self, field_name):
  39         return self.diff.get(field_name, None)
  40
  41     def can_update(self, user):
  42         if user.is_readonly:
  43             return False
  44         if user.is_admin:
  45             return True
  46         return False
  47
  48     def delete(self, *args, **kwds):
  49         # so we have something to give the observer
  50         pk = self.pk
  51         model_dict = model_to_dict(self)
  52         for (k,v) in model_dict.items():
  53             # things like datetime are not JSON serializable
  54             model_dict[k] = str(v)
  55
  56         super(PlCoreBase, self).delete(*args, **kwds)
  57
  58         # This is a no-op if observer_disabled is set
  59         notify_observer(model=self, delete=True, pk=pk, model_dict=model_dict)
  60
  61     def save(self, *args, **kwargs):
  62         super(PlCoreBase, self).save(*args, **kwargs)
  63
  64         # This is a no-op if observer_disabled is set
  65         notify_observer()
  66
  67         self.__initial = self._dict
  68
  69     def save_by_user(self, user, *args, **kwds):
  70         if self.can_update(user):
  71             self.save(*args, **kwds)
  72
  73     @property
  74     def _dict(self):
  75         return model_to_dict(self, fields=[field.name for field in
  76                              self._meta.fields])
  77
  78
  79