Scott Baker | fd44dfc | 2014-05-23 13:20:53 -0700 | [diff] [blame] | 1 | import datetime |
Siobhan Tully | 4bc09f2 | 2013-04-10 21:15:21 -0400 | [diff] [blame] | 2 | import os |
Scott Baker | 65d5a9a | 2014-05-26 15:58:09 -0700 | [diff] [blame] | 3 | import sys |
Siobhan Tully | 4bc09f2 | 2013-04-10 21:15:21 -0400 | [diff] [blame] | 4 | from django.db import models |
Scott Baker | 13acdd6 | 2013-05-08 17:42:56 -0700 | [diff] [blame] | 5 | from django.forms.models import model_to_dict |
Scott Baker | c1c45f8 | 2014-01-21 16:23:51 -0800 | [diff] [blame] | 6 | from django.core.urlresolvers import reverse |
Scott Baker | 6ecd426 | 2014-01-21 23:15:21 -0800 | [diff] [blame] | 7 | from django.forms.models import model_to_dict |
Sapan Bhatia | bad6774 | 2014-09-04 00:39:19 -0400 | [diff] [blame] | 8 | import model_policy |
Scott Baker | 9e99074 | 2014-03-19 22:14:58 -0700 | [diff] [blame] | 9 | |
| 10 | try: |
| 11 | # This is a no-op if observer_disabled is set to 1 in the config file |
| 12 | from observer import * |
| 13 | except: |
Scott Baker | 65d5a9a | 2014-05-26 15:58:09 -0700 | [diff] [blame] | 14 | print >> sys.stderr, "import of observer failed! printing traceback and disabling observer:" |
Scott Baker | 9e99074 | 2014-03-19 22:14:58 -0700 | [diff] [blame] | 15 | import traceback |
| 16 | traceback.print_exc() |
| 17 | |
| 18 | # guard against something failing |
Scott Baker | fd44dfc | 2014-05-23 13:20:53 -0700 | [diff] [blame] | 19 | def notify_observer(*args, **kwargs): |
Scott Baker | 9e99074 | 2014-03-19 22:14:58 -0700 | [diff] [blame] | 20 | pass |
Siobhan Tully | 4bc09f2 | 2013-04-10 21:15:21 -0400 | [diff] [blame] | 21 | |
Sapan Bhatia | 3089d83 | 2014-04-29 14:36:51 -0400 | [diff] [blame] | 22 | # This manager will be inherited by all subclasses because |
| 23 | # the core model is abstract. |
Sapan Bhatia | 15bf5ac | 2014-07-21 20:06:59 -0400 | [diff] [blame] | 24 | class PlCoreBaseDeletionManager(models.Manager): |
| 25 | def get_query_set(self): |
Sapan Bhatia | bbcc190 | 2014-07-22 01:11:36 -0400 | [diff] [blame] | 26 | return super(PlCoreBaseDeletionManager, self).get_query_set().filter(deleted=True) |
Sapan Bhatia | 15bf5ac | 2014-07-21 20:06:59 -0400 | [diff] [blame] | 27 | |
| 28 | # This manager will be inherited by all subclasses because |
| 29 | # the core model is abstract. |
Sapan Bhatia | 4eb663a | 2014-04-29 14:26:10 -0400 | [diff] [blame] | 30 | class PlCoreBaseManager(models.Manager): |
| 31 | def get_query_set(self): |
| 32 | return super(PlCoreBaseManager, self).get_query_set().filter(deleted=False) |
Siobhan Tully | 4bc09f2 | 2013-04-10 21:15:21 -0400 | [diff] [blame] | 33 | |
Sapan Bhatia | 4eb663a | 2014-04-29 14:26:10 -0400 | [diff] [blame] | 34 | class PlCoreBase(models.Model): |
| 35 | objects = PlCoreBaseManager() |
Sapan Bhatia | 15bf5ac | 2014-07-21 20:06:59 -0400 | [diff] [blame] | 36 | deleted_objects = PlCoreBaseDeletionManager() |
| 37 | |
| 38 | # default values for created and updated are only there to keep evolution |
| 39 | # from failing. |
Scott Baker | 70b7005 | 2014-09-12 12:05:51 -0700 | [diff] [blame^] | 40 | created = models.DateTimeField(auto_now_add=True, default=datetime.datetime.now) |
| 41 | updated = models.DateTimeField(auto_now=True, default=datetime.datetime.now) |
Sapan Bhatia | 9c2c8fa | 2013-10-16 13:26:05 -0400 | [diff] [blame] | 42 | enacted = models.DateTimeField(null=True, default=None) |
Sapan Bhatia | 34aee75 | 2014-04-28 21:06:39 -0400 | [diff] [blame] | 43 | backend_status = models.CharField(max_length=140, |
Sapan Bhatia | d507f43 | 2014-04-29 00:41:39 -0400 | [diff] [blame] | 44 | default="Provisioning in progress") |
Sapan Bhatia | bcc1899 | 2014-04-29 10:32:14 -0400 | [diff] [blame] | 45 | deleted = models.BooleanField(default=False) |
Siobhan Tully | 4bc09f2 | 2013-04-10 21:15:21 -0400 | [diff] [blame] | 46 | |
Sapan Bhatia | 9c2c8fa | 2013-10-16 13:26:05 -0400 | [diff] [blame] | 47 | class Meta: |
Sapan Bhatia | 3089d83 | 2014-04-29 14:36:51 -0400 | [diff] [blame] | 48 | # Changing abstract to False would require the managers of subclasses of |
| 49 | # PlCoreBase to be customized individually. |
Sapan Bhatia | 9c2c8fa | 2013-10-16 13:26:05 -0400 | [diff] [blame] | 50 | abstract = True |
| 51 | app_label = "core" |
Siobhan Tully | 4bc09f2 | 2013-04-10 21:15:21 -0400 | [diff] [blame] | 52 | |
Sapan Bhatia | 9c2c8fa | 2013-10-16 13:26:05 -0400 | [diff] [blame] | 53 | def __init__(self, *args, **kwargs): |
| 54 | super(PlCoreBase, self).__init__(*args, **kwargs) |
| 55 | self.__initial = self._dict |
Scott Baker | 13acdd6 | 2013-05-08 17:42:56 -0700 | [diff] [blame] | 56 | |
Sapan Bhatia | 9c2c8fa | 2013-10-16 13:26:05 -0400 | [diff] [blame] | 57 | @property |
| 58 | def diff(self): |
| 59 | d1 = self.__initial |
| 60 | d2 = self._dict |
| 61 | diffs = [(k, (v, d2[k])) for k, v in d1.items() if v != d2[k]] |
| 62 | return dict(diffs) |
Siobhan Tully | 7329134 | 2013-05-10 10:50:08 -0400 | [diff] [blame] | 63 | |
Sapan Bhatia | 9c2c8fa | 2013-10-16 13:26:05 -0400 | [diff] [blame] | 64 | @property |
| 65 | def has_changed(self): |
| 66 | return bool(self.diff) |
Siobhan Tully | 7329134 | 2013-05-10 10:50:08 -0400 | [diff] [blame] | 67 | |
Sapan Bhatia | 9c2c8fa | 2013-10-16 13:26:05 -0400 | [diff] [blame] | 68 | @property |
| 69 | def changed_fields(self): |
| 70 | return self.diff.keys() |
Siobhan Tully | 7329134 | 2013-05-10 10:50:08 -0400 | [diff] [blame] | 71 | |
Sapan Bhatia | 9c2c8fa | 2013-10-16 13:26:05 -0400 | [diff] [blame] | 72 | def get_field_diff(self, field_name): |
| 73 | return self.diff.get(field_name, None) |
Siobhan Tully | 7329134 | 2013-05-10 10:50:08 -0400 | [diff] [blame] | 74 | |
Tony Mack | 5b06147 | 2014-02-04 07:57:10 -0500 | [diff] [blame] | 75 | def can_update(self, user): |
| 76 | if user.is_readonly: |
| 77 | return False |
| 78 | if user.is_admin: |
| 79 | return True |
| 80 | return False |
| 81 | |
Sapan Bhatia | 9c2c8fa | 2013-10-16 13:26:05 -0400 | [diff] [blame] | 82 | def delete(self, *args, **kwds): |
Scott Baker | 6ecd426 | 2014-01-21 23:15:21 -0800 | [diff] [blame] | 83 | # so we have something to give the observer |
Sapan Bhatia | 77d1d89 | 2014-07-21 20:07:23 -0400 | [diff] [blame] | 84 | purge = kwds.get('purge',False) |
| 85 | try: |
| 86 | purge = purge or observer_disabled |
| 87 | except NameError: |
| 88 | pass |
| 89 | |
| 90 | if (purge): |
Sapan Bhatia | ad2c4a1 | 2014-07-22 01:21:05 -0400 | [diff] [blame] | 91 | del kwds['purge'] |
Sapan Bhatia | 77d1d89 | 2014-07-21 20:07:23 -0400 | [diff] [blame] | 92 | super(PlCoreBase, self).delete(*args, **kwds) |
Sapan Bhatia | c860243 | 2014-04-29 20:33:51 -0400 | [diff] [blame] | 93 | else: |
| 94 | self.deleted = True |
| 95 | self.enacted=None |
| 96 | self.save(update_fields=['enacted','deleted']) |
Scott Baker | 6ecd426 | 2014-01-21 23:15:21 -0800 | [diff] [blame] | 97 | |
Sapan Bhatia | dbaf193 | 2013-09-03 11:28:52 -0400 | [diff] [blame] | 98 | |
Sapan Bhatia | 9c2c8fa | 2013-10-16 13:26:05 -0400 | [diff] [blame] | 99 | def save(self, *args, **kwargs): |
| 100 | super(PlCoreBase, self).save(*args, **kwargs) |
Sapan Bhatia | c860243 | 2014-04-29 20:33:51 -0400 | [diff] [blame] | 101 | |
Sapan Bhatia | 9c2c8fa | 2013-10-16 13:26:05 -0400 | [diff] [blame] | 102 | # This is a no-op if observer_disabled is set |
| 103 | notify_observer() |
Sapan Bhatia | 66f4e61 | 2013-07-02 12:12:38 -0400 | [diff] [blame] | 104 | |
Sapan Bhatia | 9c2c8fa | 2013-10-16 13:26:05 -0400 | [diff] [blame] | 105 | self.__initial = self._dict |
Scott Baker | 13acdd6 | 2013-05-08 17:42:56 -0700 | [diff] [blame] | 106 | |
Tony Mack | 5b06147 | 2014-02-04 07:57:10 -0500 | [diff] [blame] | 107 | def save_by_user(self, user, *args, **kwds): |
| 108 | if self.can_update(user): |
| 109 | self.save(*args, **kwds) |
| 110 | |
Tony Mack | 332ee1d | 2014-02-04 15:33:45 -0500 | [diff] [blame] | 111 | def delete_by_user(self, user, *args, **kwds): |
| 112 | if self.can_update(user): |
| 113 | self.delete(*args, **kwds) |
| 114 | |
Sapan Bhatia | 9c2c8fa | 2013-10-16 13:26:05 -0400 | [diff] [blame] | 115 | @property |
| 116 | def _dict(self): |
| 117 | return model_to_dict(self, fields=[field.name for field in |
| 118 | self._meta.fields]) |
Scott Baker | 13acdd6 | 2013-05-08 17:42:56 -0700 | [diff] [blame] | 119 | |
Siobhan Tully | 4bc09f2 | 2013-04-10 21:15:21 -0400 | [diff] [blame] | 120 | |
| 121 | |