planetstack/core/models/plcorebase.py

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