Siobhan Tully | 4bc09f2 | 2013-04-10 21:15:21 -0400 | [diff] [blame] | 1 | import os |
Scott Baker | 65d5a9a | 2014-05-26 15:58:09 -0700 | [diff] [blame] | 2 | import sys |
Siobhan Tully | 4bc09f2 | 2013-04-10 21:15:21 -0400 | [diff] [blame] | 3 | from django.db import models |
Scott Baker | 13acdd6 | 2013-05-08 17:42:56 -0700 | [diff] [blame] | 4 | from django.forms.models import model_to_dict |
Scott Baker | c1c45f8 | 2014-01-21 16:23:51 -0800 | [diff] [blame] | 5 | from django.core.urlresolvers import reverse |
Scott Baker | 6ecd426 | 2014-01-21 23:15:21 -0800 | [diff] [blame] | 6 | from django.forms.models import model_to_dict |
Scott Baker | b24f2c3 | 2014-09-17 22:18:46 -0700 | [diff] [blame] | 7 | from django.utils import timezone |
Scott Baker | 0bec56f | 2014-10-02 22:50:18 -0700 | [diff] [blame] | 8 | from django.core.exceptions import PermissionDenied |
Sapan Bhatia | bad6774 | 2014-09-04 00:39:19 -0400 | [diff] [blame] | 9 | import model_policy |
Scott Baker | 9e99074 | 2014-03-19 22:14:58 -0700 | [diff] [blame] | 10 | |
| 11 | try: |
| 12 | # This is a no-op if observer_disabled is set to 1 in the config file |
| 13 | from observer import * |
| 14 | except: |
Scott Baker | 65d5a9a | 2014-05-26 15:58:09 -0700 | [diff] [blame] | 15 | print >> sys.stderr, "import of observer failed! printing traceback and disabling observer:" |
Scott Baker | 9e99074 | 2014-03-19 22:14:58 -0700 | [diff] [blame] | 16 | import traceback |
| 17 | traceback.print_exc() |
| 18 | |
| 19 | # guard against something failing |
Scott Baker | fd44dfc | 2014-05-23 13:20:53 -0700 | [diff] [blame] | 20 | def notify_observer(*args, **kwargs): |
Scott Baker | 9e99074 | 2014-03-19 22:14:58 -0700 | [diff] [blame] | 21 | pass |
Siobhan Tully | 4bc09f2 | 2013-04-10 21:15:21 -0400 | [diff] [blame] | 22 | |
Sapan Bhatia | 3089d83 | 2014-04-29 14:36:51 -0400 | [diff] [blame] | 23 | # This manager will be inherited by all subclasses because |
| 24 | # the core model is abstract. |
Sapan Bhatia | 15bf5ac | 2014-07-21 20:06:59 -0400 | [diff] [blame] | 25 | class PlCoreBaseDeletionManager(models.Manager): |
Scott Baker | b08d656 | 2014-09-12 12:57:27 -0700 | [diff] [blame] | 26 | def get_queryset(self): |
Scott Baker | b24f2c3 | 2014-09-17 22:18:46 -0700 | [diff] [blame] | 27 | parent=super(PlCoreBaseDeletionManager, self) |
| 28 | if hasattr(parent, "get_queryset"): |
| 29 | return parent.get_queryset().filter(deleted=True) |
| 30 | else: |
| 31 | return parent.get_query_set().filter(deleted=True) |
Sapan Bhatia | 15bf5ac | 2014-07-21 20:06:59 -0400 | [diff] [blame] | 32 | |
Scott Baker | b24f2c3 | 2014-09-17 22:18:46 -0700 | [diff] [blame] | 33 | # deprecated in django 1.7 in favor of get_queryset(). |
Scott Baker | b08d656 | 2014-09-12 12:57:27 -0700 | [diff] [blame] | 34 | def get_query_set(self): |
| 35 | return self.get_queryset() |
| 36 | |
Sapan Bhatia | 15bf5ac | 2014-07-21 20:06:59 -0400 | [diff] [blame] | 37 | # This manager will be inherited by all subclasses because |
| 38 | # the core model is abstract. |
Sapan Bhatia | 4eb663a | 2014-04-29 14:26:10 -0400 | [diff] [blame] | 39 | class PlCoreBaseManager(models.Manager): |
Scott Baker | b08d656 | 2014-09-12 12:57:27 -0700 | [diff] [blame] | 40 | def get_queryset(self): |
Scott Baker | b24f2c3 | 2014-09-17 22:18:46 -0700 | [diff] [blame] | 41 | parent=super(PlCoreBaseManager, self) |
| 42 | if hasattr(parent, "get_queryset"): |
| 43 | return parent.get_queryset().filter(deleted=False) |
| 44 | else: |
| 45 | return parent.get_query_set().filter(deleted=False) |
Siobhan Tully | 4bc09f2 | 2013-04-10 21:15:21 -0400 | [diff] [blame] | 46 | |
Scott Baker | b24f2c3 | 2014-09-17 22:18:46 -0700 | [diff] [blame] | 47 | # deprecated in django 1.7 in favor of get_queryset(). |
Scott Baker | b08d656 | 2014-09-12 12:57:27 -0700 | [diff] [blame] | 48 | def get_query_set(self): |
| 49 | return self.get_queryset() |
| 50 | |
Scott Baker | cbfb600 | 2014-10-03 00:32:37 -0700 | [diff] [blame] | 51 | class DiffModelMixIn: |
| 52 | # Provides useful methods for computing which objects in a model have |
| 53 | # changed. Make sure to do self._initial = self._dict in the __init__ |
| 54 | # method. |
| 55 | |
| 56 | # This is broken out of PlCoreBase into a Mixin so the User model can |
| 57 | # also make use of it. |
| 58 | |
| 59 | @property |
| 60 | def _dict(self): |
| 61 | return model_to_dict(self, fields=[field.name for field in |
| 62 | self._meta.fields]) |
| 63 | |
| 64 | @property |
| 65 | def diff(self): |
| 66 | d1 = self._initial |
| 67 | d2 = self._dict |
| 68 | diffs = [(k, (v, d2[k])) for k, v in d1.items() if v != d2[k]] |
| 69 | return dict(diffs) |
| 70 | |
| 71 | @property |
| 72 | def has_changed(self): |
| 73 | return bool(self.diff) |
| 74 | |
| 75 | @property |
| 76 | def changed_fields(self): |
| 77 | return self.diff.keys() |
| 78 | |
Scott Baker | cbfb600 | 2014-10-03 00:32:37 -0700 | [diff] [blame] | 79 | def has_field_changed(self, field_name): |
| 80 | return field_name in self.diff.keys() |
| 81 | |
| 82 | def get_field_diff(self, field_name): |
| 83 | return self.diff.get(field_name, None) |
| 84 | |
Scott Baker | 46e50af | 2014-11-06 17:33:27 -0800 | [diff] [blame] | 85 | class PlCoreBase(models.Model): # , DiffModelMixIn): |
Sapan Bhatia | 4eb663a | 2014-04-29 14:26:10 -0400 | [diff] [blame] | 86 | objects = PlCoreBaseManager() |
Sapan Bhatia | 15bf5ac | 2014-07-21 20:06:59 -0400 | [diff] [blame] | 87 | deleted_objects = PlCoreBaseDeletionManager() |
| 88 | |
Scott Baker | 46e50af | 2014-11-06 17:33:27 -0800 | [diff] [blame] | 89 | # ---- copy stuff from DiffModelMixin ---- |
| 90 | |
| 91 | # XXX Django fails miserably when trying to create initial migrations when |
| 92 | # DiffModelMixin is used. So, until we figure out what's wrong, |
| 93 | # just copied the guts of DiffModelMixIn here. |
| 94 | |
| 95 | @property |
| 96 | def _dict(self): |
| 97 | return model_to_dict(self, fields=[field.name for field in |
| 98 | self._meta.fields]) |
| 99 | |
| 100 | @property |
| 101 | def diff(self): |
| 102 | d1 = self._initial |
| 103 | d2 = self._dict |
| 104 | diffs = [(k, (v, d2[k])) for k, v in d1.items() if v != d2[k]] |
| 105 | return dict(diffs) |
| 106 | |
| 107 | @property |
| 108 | def has_changed(self): |
| 109 | return bool(self.diff) |
| 110 | |
| 111 | @property |
| 112 | def changed_fields(self): |
| 113 | return self.diff.keys() |
| 114 | |
| 115 | def has_field_changed(self, field_name): |
| 116 | return field_name in self.diff.keys() |
| 117 | |
| 118 | def get_field_diff(self, field_name): |
| 119 | return self.diff.get(field_name, None) |
| 120 | # ---- end copy stuff from DiffModelMixin ---- |
| 121 | |
Sapan Bhatia | 15bf5ac | 2014-07-21 20:06:59 -0400 | [diff] [blame] | 122 | # default values for created and updated are only there to keep evolution |
| 123 | # from failing. |
Scott Baker | b24f2c3 | 2014-09-17 22:18:46 -0700 | [diff] [blame] | 124 | created = models.DateTimeField(auto_now_add=True, default=timezone.now) |
| 125 | updated = models.DateTimeField(auto_now=True, default=timezone.now) |
Sapan Bhatia | 9c2c8fa | 2013-10-16 13:26:05 -0400 | [diff] [blame] | 126 | enacted = models.DateTimeField(null=True, default=None) |
Sapan Bhatia | 34aee75 | 2014-04-28 21:06:39 -0400 | [diff] [blame] | 127 | backend_status = models.CharField(max_length=140, |
Sapan Bhatia | d507f43 | 2014-04-29 00:41:39 -0400 | [diff] [blame] | 128 | default="Provisioning in progress") |
Sapan Bhatia | bcc1899 | 2014-04-29 10:32:14 -0400 | [diff] [blame] | 129 | deleted = models.BooleanField(default=False) |
Siobhan Tully | 4bc09f2 | 2013-04-10 21:15:21 -0400 | [diff] [blame] | 130 | |
Sapan Bhatia | 9c2c8fa | 2013-10-16 13:26:05 -0400 | [diff] [blame] | 131 | class Meta: |
Sapan Bhatia | 3089d83 | 2014-04-29 14:36:51 -0400 | [diff] [blame] | 132 | # Changing abstract to False would require the managers of subclasses of |
| 133 | # PlCoreBase to be customized individually. |
Sapan Bhatia | 9c2c8fa | 2013-10-16 13:26:05 -0400 | [diff] [blame] | 134 | abstract = True |
| 135 | app_label = "core" |
Siobhan Tully | 4bc09f2 | 2013-04-10 21:15:21 -0400 | [diff] [blame] | 136 | |
Sapan Bhatia | 9c2c8fa | 2013-10-16 13:26:05 -0400 | [diff] [blame] | 137 | def __init__(self, *args, **kwargs): |
| 138 | super(PlCoreBase, self).__init__(*args, **kwargs) |
Scott Baker | cbfb600 | 2014-10-03 00:32:37 -0700 | [diff] [blame] | 139 | self._initial = self._dict # for DiffModelMixIn |
Scott Baker | 5dc87a6 | 2014-09-23 22:41:17 -0700 | [diff] [blame] | 140 | self.silent = False |
Scott Baker | 13acdd6 | 2013-05-08 17:42:56 -0700 | [diff] [blame] | 141 | |
Tony Mack | 5b06147 | 2014-02-04 07:57:10 -0500 | [diff] [blame] | 142 | def can_update(self, user): |
| 143 | if user.is_readonly: |
| 144 | return False |
| 145 | if user.is_admin: |
| 146 | return True |
Tony Mack | 5b06147 | 2014-02-04 07:57:10 -0500 | [diff] [blame] | 147 | |
Scott Baker | 0119c15 | 2014-10-06 22:58:48 -0700 | [diff] [blame] | 148 | return False |
Scott Baker | cbfb600 | 2014-10-03 00:32:37 -0700 | [diff] [blame] | 149 | |
Sapan Bhatia | 9c2c8fa | 2013-10-16 13:26:05 -0400 | [diff] [blame] | 150 | def delete(self, *args, **kwds): |
Scott Baker | 6ecd426 | 2014-01-21 23:15:21 -0800 | [diff] [blame] | 151 | # so we have something to give the observer |
Sapan Bhatia | 77d1d89 | 2014-07-21 20:07:23 -0400 | [diff] [blame] | 152 | purge = kwds.get('purge',False) |
Scott Baker | c5b5060 | 2014-10-09 16:22:00 -0700 | [diff] [blame] | 153 | if purge: |
| 154 | del kwds['purge'] |
Scott Baker | 0491f6f | 2014-09-23 16:04:36 -0700 | [diff] [blame] | 155 | silent = kwds.get('silent',False) |
Scott Baker | c5b5060 | 2014-10-09 16:22:00 -0700 | [diff] [blame] | 156 | if silent: |
| 157 | del kwds['silent'] |
Sapan Bhatia | 77d1d89 | 2014-07-21 20:07:23 -0400 | [diff] [blame] | 158 | try: |
| 159 | purge = purge or observer_disabled |
| 160 | except NameError: |
| 161 | pass |
Scott Baker | 0491f6f | 2014-09-23 16:04:36 -0700 | [diff] [blame] | 162 | |
Sapan Bhatia | 77d1d89 | 2014-07-21 20:07:23 -0400 | [diff] [blame] | 163 | if (purge): |
| 164 | super(PlCoreBase, self).delete(*args, **kwds) |
Sapan Bhatia | c860243 | 2014-04-29 20:33:51 -0400 | [diff] [blame] | 165 | else: |
| 166 | self.deleted = True |
| 167 | self.enacted=None |
Scott Baker | 0491f6f | 2014-09-23 16:04:36 -0700 | [diff] [blame] | 168 | self.save(update_fields=['enacted','deleted'], silent=silent) |
Sapan Bhatia | dbaf193 | 2013-09-03 11:28:52 -0400 | [diff] [blame] | 169 | |
Sapan Bhatia | 9c2c8fa | 2013-10-16 13:26:05 -0400 | [diff] [blame] | 170 | def save(self, *args, **kwargs): |
Scott Baker | 5dc87a6 | 2014-09-23 22:41:17 -0700 | [diff] [blame] | 171 | # let the user specify silence as either a kwarg or an instance varible |
| 172 | silent = self.silent |
Scott Baker | 0491f6f | 2014-09-23 16:04:36 -0700 | [diff] [blame] | 173 | if "silent" in kwargs: |
Scott Baker | 5dc87a6 | 2014-09-23 22:41:17 -0700 | [diff] [blame] | 174 | silent=silent or kwargs.pop("silent") |
Scott Baker | 0491f6f | 2014-09-23 16:04:36 -0700 | [diff] [blame] | 175 | |
Sapan Bhatia | 9c2c8fa | 2013-10-16 13:26:05 -0400 | [diff] [blame] | 176 | super(PlCoreBase, self).save(*args, **kwargs) |
Sapan Bhatia | c860243 | 2014-04-29 20:33:51 -0400 | [diff] [blame] | 177 | |
Sapan Bhatia | 9c2c8fa | 2013-10-16 13:26:05 -0400 | [diff] [blame] | 178 | # This is a no-op if observer_disabled is set |
Scott Baker | 0491f6f | 2014-09-23 16:04:36 -0700 | [diff] [blame] | 179 | if not silent: |
| 180 | notify_observer() |
Sapan Bhatia | 66f4e61 | 2013-07-02 12:12:38 -0400 | [diff] [blame] | 181 | |
Scott Baker | 165f70c | 2014-10-03 14:48:06 -0700 | [diff] [blame] | 182 | self._initial = self._dict |
Scott Baker | 13acdd6 | 2013-05-08 17:42:56 -0700 | [diff] [blame] | 183 | |
Tony Mack | 5b06147 | 2014-02-04 07:57:10 -0500 | [diff] [blame] | 184 | def save_by_user(self, user, *args, **kwds): |
Scott Baker | 0bec56f | 2014-10-02 22:50:18 -0700 | [diff] [blame] | 185 | if not self.can_update(user): |
Scott Baker | 0119c15 | 2014-10-06 22:58:48 -0700 | [diff] [blame] | 186 | if getattr(self, "_cant_update_fieldName", None) is not None: |
| 187 | raise PermissionDenied("You do not have permission to update field %s on object %s" % (self._cant_update_fieldName, self.__class__.__name__)) |
| 188 | else: |
| 189 | raise PermissionDenied("You do not have permission to update %s objects" % self.__class__.__name__) |
Scott Baker | cbfb600 | 2014-10-03 00:32:37 -0700 | [diff] [blame] | 190 | |
Scott Baker | 0bec56f | 2014-10-02 22:50:18 -0700 | [diff] [blame] | 191 | self.save(*args, **kwds) |
Tony Mack | 5b06147 | 2014-02-04 07:57:10 -0500 | [diff] [blame] | 192 | |
Tony Mack | 332ee1d | 2014-02-04 15:33:45 -0500 | [diff] [blame] | 193 | def delete_by_user(self, user, *args, **kwds): |
Scott Baker | 0bec56f | 2014-10-02 22:50:18 -0700 | [diff] [blame] | 194 | if not self.can_update(user): |
| 195 | raise PermissionDenied("You do not have permission to delete %s objects" % self.__class__.__name__) |
| 196 | self.delete(*args, **kwds) |
Tony Mack | 332ee1d | 2014-02-04 15:33:45 -0500 | [diff] [blame] | 197 | |
Scott Baker | 165f70c | 2014-10-03 14:48:06 -0700 | [diff] [blame] | 198 | @classmethod |
| 199 | def select_by_user(cls, user): |
| 200 | # This should be overridden by descendant classes that want to perform |
| 201 | # filtering of visible objects by user. |
| 202 | return cls.objects.all() |
| 203 | |
Sapan Bhatia | 998b32e | 2014-11-12 10:06:23 -0500 | [diff] [blame^] | 204 | @classmethod |
| 205 | def is_ephemeral(cls): |
| 206 | return False |
| 207 | |
Scott Baker | 13acdd6 | 2013-05-08 17:42:56 -0700 | [diff] [blame] | 208 | |
Siobhan Tully | 4bc09f2 | 2013-04-10 21:15:21 -0400 | [diff] [blame] | 209 | |
| 210 | |