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 # This is a no-op if observer_disabled is set to 1 in the config file
   6 from observer import *
   7
   8 class PlCoreBase(models.Model):
   9
  10     created = models.DateTimeField(auto_now_add=True)
  11     updated = models.DateTimeField(auto_now=True)
  12     enacted = models.DateTimeField(null=True, default=None)
  13
  14     class Meta:
  15         abstract = True
  16         app_label = "core"
  17
  18     def __init__(self, *args, **kwargs):
  19         super(PlCoreBase, self).__init__(*args, **kwargs)
  20         self.__initial = self._dict
  21
  22     @property
  23     def diff(self):
  24         d1 = self.__initial
  25         d2 = self._dict
  26         diffs = [(k, (v, d2[k])) for k, v in d1.items() if v != d2[k]]
  27         return dict(diffs)
  28
  29     @property
  30     def has_changed(self):
  31         return bool(self.diff)
  32
  33     @property
  34     def changed_fields(self):
  35         return self.diff.keys()
  36
  37     def get_field_diff(self, field_name):
  38         return self.diff.get(field_name, None)
  39
  40     def delete(self, *args, **kwds):
  41         super(PlCoreBase, self).delete(*args, **kwds)
  42
  43         # This is a no-op if observer_disabled is set
  44         notify_observer(model=self, delete=True, pk=self.pk)
  45
  46     def save(self, *args, **kwargs):
  47         super(PlCoreBase, self).save(*args, **kwargs)
  48
  49         # This is a no-op if observer_disabled is set
  50         notify_observer()
  51
  52         self.__initial = self._dict
  53
  54     @property
  55     def _dict(self):
  56         return model_to_dict(self, fields=[field.name for field in
  57                              self._meta.fields])
  58
  59
  60