Siobhan Tully | 4bc09f2 | 2013-04-10 21:15:21 -0400 | [diff] [blame] | 1 | import os |
| 2 | from django.db import models |
Scott Baker | 13acdd6 | 2013-05-08 17:42:56 -0700 | [diff] [blame] | 3 | from django.forms.models import model_to_dict |
Sapan Bhatia | bd02841 | 2013-09-03 11:53:18 -0400 | [diff] [blame] | 4 | from observer.event_manager import EventSender |
Sapan Bhatia | 66f4e61 | 2013-07-02 12:12:38 -0400 | [diff] [blame] | 5 | |
Siobhan Tully | 4bc09f2 | 2013-04-10 21:15:21 -0400 | [diff] [blame] | 6 | |
| 7 | class PlCoreBase(models.Model): |
| 8 | |
Sapan Bhatia | 322edfa | 2013-10-02 10:04:06 -0400 | [diff] [blame^] | 9 | created = models.DateTimeField(auto_now_add=True) |
| 10 | updated = models.DateTimeField(auto_now=True) |
| 11 | enacted = models.DateTimeField(null=True, default=None) |
Siobhan Tully | 4bc09f2 | 2013-04-10 21:15:21 -0400 | [diff] [blame] | 12 | |
Sapan Bhatia | 322edfa | 2013-10-02 10:04:06 -0400 | [diff] [blame^] | 13 | class Meta: |
| 14 | abstract = True |
| 15 | app_label = "core" |
Siobhan Tully | 4bc09f2 | 2013-04-10 21:15:21 -0400 | [diff] [blame] | 16 | |
Sapan Bhatia | 322edfa | 2013-10-02 10:04:06 -0400 | [diff] [blame^] | 17 | def __init__(self, *args, **kwargs): |
| 18 | super(PlCoreBase, self).__init__(*args, **kwargs) |
| 19 | self.__initial = self._dict |
Scott Baker | 13acdd6 | 2013-05-08 17:42:56 -0700 | [diff] [blame] | 20 | |
Sapan Bhatia | 322edfa | 2013-10-02 10:04:06 -0400 | [diff] [blame^] | 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) |
Siobhan Tully | 7329134 | 2013-05-10 10:50:08 -0400 | [diff] [blame] | 27 | |
Sapan Bhatia | 322edfa | 2013-10-02 10:04:06 -0400 | [diff] [blame^] | 28 | @property |
| 29 | def has_changed(self): |
| 30 | return bool(self.diff) |
Siobhan Tully | 7329134 | 2013-05-10 10:50:08 -0400 | [diff] [blame] | 31 | |
Sapan Bhatia | 322edfa | 2013-10-02 10:04:06 -0400 | [diff] [blame^] | 32 | @property |
| 33 | def changed_fields(self): |
| 34 | return self.diff.keys() |
Siobhan Tully | 7329134 | 2013-05-10 10:50:08 -0400 | [diff] [blame] | 35 | |
Sapan Bhatia | 322edfa | 2013-10-02 10:04:06 -0400 | [diff] [blame^] | 36 | def get_field_diff(self, field_name): |
| 37 | return self.diff.get(field_name, None) |
Siobhan Tully | 7329134 | 2013-05-10 10:50:08 -0400 | [diff] [blame] | 38 | |
Sapan Bhatia | 322edfa | 2013-10-02 10:04:06 -0400 | [diff] [blame^] | 39 | def delete(self, *args, **kwds): |
| 40 | super(PlCoreBase, self).delete(*args, **kwds) |
Sapan Bhatia | dbaf193 | 2013-09-03 11:28:52 -0400 | [diff] [blame] | 41 | |
Sapan Bhatia | 322edfa | 2013-10-02 10:04:06 -0400 | [diff] [blame^] | 42 | try: |
| 43 | EventSender().fire({'delete_flag':True,'model':self.__name__}) |
| 44 | except: |
| 45 | # Investigate later. |
| 46 | pass |
Sapan Bhatia | dbaf193 | 2013-09-03 11:28:52 -0400 | [diff] [blame] | 47 | |
Sapan Bhatia | 66f4e61 | 2013-07-02 12:12:38 -0400 | [diff] [blame] | 48 | |
Sapan Bhatia | 322edfa | 2013-10-02 10:04:06 -0400 | [diff] [blame^] | 49 | def save(self, *args, **kwargs): |
| 50 | super(PlCoreBase, self).save(*args, **kwargs) |
| 51 | |
| 52 | # Tell the observer that the source database has been updated |
| 53 | EventSender().fire() |
Scott Baker | 13acdd6 | 2013-05-08 17:42:56 -0700 | [diff] [blame] | 54 | |
Sapan Bhatia | 322edfa | 2013-10-02 10:04:06 -0400 | [diff] [blame^] | 55 | self.__initial = self._dict |
| 56 | |
| 57 | @property |
| 58 | def _dict(self): |
| 59 | return model_to_dict(self, fields=[field.name for field in |
| 60 | self._meta.fields]) |
Scott Baker | 13acdd6 | 2013-05-08 17:42:56 -0700 | [diff] [blame] | 61 | |
Siobhan Tully | 4bc09f2 | 2013-04-10 21:15:21 -0400 | [diff] [blame] | 62 | |
| 63 | |