Siobhan Tully | 5343728 | 2013-04-26 19:30:27 -0400 | [diff] [blame] | 1 | import os |
| 2 | import datetime |
Tony Mack | c14de8f | 2013-05-09 21:44:17 -0400 | [diff] [blame] | 3 | from collections import defaultdict |
Siobhan Tully | 5343728 | 2013-04-26 19:30:27 -0400 | [diff] [blame] | 4 | from django.db import models |
Tony Mack | 5b06147 | 2014-02-04 07:57:10 -0500 | [diff] [blame] | 5 | from django.db.models import F, Q |
Scott Baker | 2c3cb64 | 2014-05-19 17:55:56 -0700 | [diff] [blame] | 6 | from core.models import PlCoreBase,Site, DashboardView |
Sapan Bhatia | f7b29d2 | 2014-06-11 17:10:11 -0400 | [diff] [blame] | 7 | from core.models.site import Deployment |
Siobhan Tully | 30fd429 | 2013-05-10 08:59:56 -0400 | [diff] [blame] | 8 | from django.contrib.auth.models import AbstractBaseUser, BaseUserManager |
Scott Baker | 9266e6b | 2013-05-19 15:54:48 -0700 | [diff] [blame] | 9 | from timezones.fields import TimeZoneField |
Scott Baker | 2c3cb64 | 2014-05-19 17:55:56 -0700 | [diff] [blame] | 10 | from operator import itemgetter, attrgetter |
Scott Baker | a36d77e | 2014-08-29 11:43:23 -0700 | [diff] [blame] | 11 | from django.core.mail import EmailMultiAlternatives |
| 12 | from core.middleware import get_request |
Sapan Bhatia | 3a5ef2b | 2014-09-03 15:39:33 -0400 | [diff] [blame^] | 13 | import model_policies |
Siobhan Tully | 5343728 | 2013-04-26 19:30:27 -0400 | [diff] [blame] | 14 | |
| 15 | # Create your models here. |
Siobhan Tully | 30fd429 | 2013-05-10 08:59:56 -0400 | [diff] [blame] | 16 | class UserManager(BaseUserManager): |
Siobhan Tully | 5343728 | 2013-04-26 19:30:27 -0400 | [diff] [blame] | 17 | def create_user(self, email, firstname, lastname, password=None): |
| 18 | """ |
| 19 | Creates and saves a User with the given email, date of |
| 20 | birth and password. |
| 21 | """ |
| 22 | if not email: |
| 23 | raise ValueError('Users must have an email address') |
| 24 | |
| 25 | user = self.model( |
Siobhan Tully | 30fd429 | 2013-05-10 08:59:56 -0400 | [diff] [blame] | 26 | email=UserManager.normalize_email(email), |
Siobhan Tully | 5343728 | 2013-04-26 19:30:27 -0400 | [diff] [blame] | 27 | firstname=firstname, |
Siobhan Tully | 30fd429 | 2013-05-10 08:59:56 -0400 | [diff] [blame] | 28 | lastname=lastname, |
| 29 | password=password |
Siobhan Tully | 5343728 | 2013-04-26 19:30:27 -0400 | [diff] [blame] | 30 | ) |
Siobhan Tully | 30fd429 | 2013-05-10 08:59:56 -0400 | [diff] [blame] | 31 | #user.set_password(password) |
Siobhan Tully | 5343728 | 2013-04-26 19:30:27 -0400 | [diff] [blame] | 32 | user.is_admin = True |
| 33 | user.save(using=self._db) |
| 34 | return user |
| 35 | |
| 36 | def create_superuser(self, email, firstname, lastname, password): |
| 37 | """ |
| 38 | Creates and saves a superuser with the given email, date of |
| 39 | birth and password. |
| 40 | """ |
| 41 | user = self.create_user(email, |
| 42 | password=password, |
| 43 | firstname=firstname, |
| 44 | lastname=lastname |
| 45 | ) |
| 46 | user.is_admin = True |
| 47 | user.save(using=self._db) |
| 48 | return user |
| 49 | |
Sapan Bhatia | 5d605ff | 2014-07-21 20:08:04 -0400 | [diff] [blame] | 50 | class DeletedUserManager(UserManager): |
| 51 | def get_query_set(self): |
| 52 | return super(UserManager, self).get_query_set().filter(deleted=True) |
Siobhan Tully | 5343728 | 2013-04-26 19:30:27 -0400 | [diff] [blame] | 53 | |
Siobhan Tully | 30fd429 | 2013-05-10 08:59:56 -0400 | [diff] [blame] | 54 | class User(AbstractBaseUser): |
Siobhan Tully | 5343728 | 2013-04-26 19:30:27 -0400 | [diff] [blame] | 55 | |
| 56 | class Meta: |
| 57 | app_label = "core" |
| 58 | |
| 59 | email = models.EmailField( |
| 60 | verbose_name='email address', |
| 61 | max_length=255, |
| 62 | unique=True, |
| 63 | db_index=True, |
| 64 | ) |
Siobhan Tully | fece0d5 | 2013-09-06 12:57:05 -0400 | [diff] [blame] | 65 | |
| 66 | username = models.CharField(max_length=255, default="Something" ) |
| 67 | |
Siobhan Tully | 5343728 | 2013-04-26 19:30:27 -0400 | [diff] [blame] | 68 | firstname = models.CharField(help_text="person's given name", max_length=200) |
| 69 | lastname = models.CharField(help_text="person's surname", max_length=200) |
| 70 | |
| 71 | phone = models.CharField(null=True, blank=True, help_text="phone number contact", max_length=100) |
| 72 | user_url = models.URLField(null=True, blank=True) |
Siobhan Tully | bfd11dc | 2013-09-03 12:59:24 -0400 | [diff] [blame] | 73 | site = models.ForeignKey(Site, related_name='users', help_text="Site this user will be homed too", null=True) |
Tony Mack | 5cbadf8 | 2013-06-10 13:56:07 -0400 | [diff] [blame] | 74 | public_key = models.TextField(null=True, blank=True, max_length=1024, help_text="Public key string") |
Siobhan Tully | 5343728 | 2013-04-26 19:30:27 -0400 | [diff] [blame] | 75 | |
| 76 | is_active = models.BooleanField(default=True) |
| 77 | is_admin = models.BooleanField(default=True) |
| 78 | is_staff = models.BooleanField(default=True) |
Siobhan Tully | cf04fb6 | 2014-01-11 11:25:57 -0500 | [diff] [blame] | 79 | is_readonly = models.BooleanField(default=False) |
Siobhan Tully | 5343728 | 2013-04-26 19:30:27 -0400 | [diff] [blame] | 80 | |
Tony Mack | 0553f28 | 2013-06-10 22:54:50 -0400 | [diff] [blame] | 81 | created = models.DateTimeField(auto_now_add=True) |
| 82 | updated = models.DateTimeField(auto_now=True) |
| 83 | enacted = models.DateTimeField(null=True, default=None) |
Sapan Bhatia | 47b9bf2 | 2014-04-28 21:09:53 -0400 | [diff] [blame] | 84 | backend_status = models.CharField(max_length=140, |
Sapan Bhatia | d507f43 | 2014-04-29 00:41:39 -0400 | [diff] [blame] | 85 | default="Provisioning in progress") |
Sapan Bhatia | bcc1899 | 2014-04-29 10:32:14 -0400 | [diff] [blame] | 86 | deleted = models.BooleanField(default=False) |
Tony Mack | 0553f28 | 2013-06-10 22:54:50 -0400 | [diff] [blame] | 87 | |
Scott Baker | 9266e6b | 2013-05-19 15:54:48 -0700 | [diff] [blame] | 88 | timezone = TimeZoneField() |
| 89 | |
Scott Baker | 2c3cb64 | 2014-05-19 17:55:56 -0700 | [diff] [blame] | 90 | dashboards = models.ManyToManyField('DashboardView', through='UserDashboardView', blank=True) |
| 91 | |
Siobhan Tully | 30fd429 | 2013-05-10 08:59:56 -0400 | [diff] [blame] | 92 | objects = UserManager() |
Sapan Bhatia | 5d605ff | 2014-07-21 20:08:04 -0400 | [diff] [blame] | 93 | deleted_objects = DeletedUserManager() |
Siobhan Tully | 5343728 | 2013-04-26 19:30:27 -0400 | [diff] [blame] | 94 | |
| 95 | USERNAME_FIELD = 'email' |
| 96 | REQUIRED_FIELDS = ['firstname', 'lastname'] |
| 97 | |
Siobhan Tully | cf04fb6 | 2014-01-11 11:25:57 -0500 | [diff] [blame] | 98 | def isReadOnlyUser(self): |
| 99 | return self.is_readonly |
| 100 | |
Siobhan Tully | 5343728 | 2013-04-26 19:30:27 -0400 | [diff] [blame] | 101 | def get_full_name(self): |
| 102 | # The user is identified by their email address |
| 103 | return self.email |
| 104 | |
| 105 | def get_short_name(self): |
| 106 | # The user is identified by their email address |
| 107 | return self.email |
| 108 | |
Sapan Bhatia | 5d605ff | 2014-07-21 20:08:04 -0400 | [diff] [blame] | 109 | def delete(self, *args, **kwds): |
| 110 | # so we have something to give the observer |
| 111 | purge = kwds.get('purge',False) |
| 112 | try: |
| 113 | purge = purge or observer_disabled |
| 114 | except NameError: |
| 115 | pass |
| 116 | |
| 117 | if (purge): |
| 118 | super(User, self).delete(*args, **kwds) |
| 119 | else: |
| 120 | self.deleted = True |
| 121 | self.enacted=None |
| 122 | self.save(update_fields=['enacted','deleted']) |
| 123 | |
Tony Mack | b0d9742 | 2013-06-10 09:57:45 -0400 | [diff] [blame] | 124 | @property |
| 125 | def keyname(self): |
| 126 | return self.email[:self.email.find('@')] |
| 127 | |
Siobhan Tully | 5343728 | 2013-04-26 19:30:27 -0400 | [diff] [blame] | 128 | def __unicode__(self): |
| 129 | return self.email |
| 130 | |
| 131 | def has_perm(self, perm, obj=None): |
| 132 | "Does the user have a specific permission?" |
| 133 | # Simplest possible answer: Yes, always |
| 134 | return True |
| 135 | |
| 136 | def has_module_perms(self, app_label): |
| 137 | "Does the user have permissions to view the app `app_label`?" |
| 138 | # Simplest possible answer: Yes, always |
| 139 | return True |
| 140 | |
Siobhan Tully | bfd11dc | 2013-09-03 12:59:24 -0400 | [diff] [blame] | 141 | def is_superuser(self): |
| 142 | return False |
Siobhan Tully | 5343728 | 2013-04-26 19:30:27 -0400 | [diff] [blame] | 143 | |
Scott Baker | 2c3cb64 | 2014-05-19 17:55:56 -0700 | [diff] [blame] | 144 | def get_dashboards(self): |
| 145 | DEFAULT_DASHBOARDS=["Tenant"] |
| 146 | |
| 147 | dashboards = sorted(list(self.dashboardViews.all()), key=attrgetter('order')) |
| 148 | dashboards = [x.dashboardView for x in dashboards] |
| 149 | |
| 150 | if not dashboards: |
| 151 | for dashboardName in DEFAULT_DASHBOARDS: |
| 152 | dbv = DashboardView.objects.filter(name=dashboardName) |
| 153 | if dbv: |
| 154 | dashboards.append(dbv[0]) |
| 155 | |
| 156 | return dashboards |
| 157 | |
Siobhan Tully | bfd11dc | 2013-09-03 12:59:24 -0400 | [diff] [blame] | 158 | # def get_roles(self): |
| 159 | # from core.models.site import SitePrivilege |
| 160 | # from core.models.slice import SliceMembership |
| 161 | # |
| 162 | # site_privileges = SitePrivilege.objects.filter(user=self) |
| 163 | # slice_memberships = SliceMembership.objects.filter(user=self) |
| 164 | # roles = defaultdict(list) |
| 165 | # for site_privilege in site_privileges: |
| 166 | # roles[site_privilege.role.role_type].append(site_privilege.site.login_base) |
| 167 | # for slice_membership in slice_memberships: |
| 168 | # roles[slice_membership.role.role_type].append(slice_membership.slice.name) |
| 169 | # return roles |
Siobhan Tully | 5343728 | 2013-04-26 19:30:27 -0400 | [diff] [blame] | 170 | |
Tony Mack | 53106f3 | 2013-04-27 16:43:01 -0400 | [diff] [blame] | 171 | def save(self, *args, **kwds): |
Siobhan Tully | 30fd429 | 2013-05-10 08:59:56 -0400 | [diff] [blame] | 172 | if not self.id: |
Scott Baker | a36d77e | 2014-08-29 11:43:23 -0700 | [diff] [blame] | 173 | self.set_password(self.password) |
| 174 | if self.is_active: |
| 175 | if self.password=="!":
|
| 176 | self.send_temporary_password()
|
| 177 |
|
Siobhan Tully | fece0d5 | 2013-09-06 12:57:05 -0400 | [diff] [blame] | 178 | self.username = self.email |
Scott Baker | a36d77e | 2014-08-29 11:43:23 -0700 | [diff] [blame] | 179 | super(User, self).save(*args, **kwds) |
| 180 | |
| 181 | def send_temporary_password(self): |
| 182 | password = User.objects.make_random_password() |
| 183 | self.set_password(password)
|
| 184 | subject, from_email, to = 'OpenCloud Account Credentials', 'support@opencloud.us', str(self.email)
|
| 185 | text_content = 'This is an important message.'
|
Scott Baker | 51e7d40 | 2014-08-29 12:32:46 -0700 | [diff] [blame] | 186 | userUrl="http://%s/" % get_request().get_host()
|
Scott Baker | a36d77e | 2014-08-29 11:43:23 -0700 | [diff] [blame] | 187 | html_content = """<p>Your account has been created on OpenCloud. Please log in <a href="""+userUrl+""">here</a> to activate your account<br><br>Username: """+self.email+"""<br>Temporary Password: """+password+"""<br>Please change your password once you successully login into the site.</p>"""
|
| 188 | msg = EmailMultiAlternatives(subject,text_content, from_email, [to])
|
| 189 | msg.attach_alternative(html_content, "text/html")
|
| 190 | msg.send() |
Tony Mack | 5b06147 | 2014-02-04 07:57:10 -0500 | [diff] [blame] | 191 | |
| 192 | @staticmethod |
| 193 | def select_by_user(user): |
| 194 | if user.is_admin: |
| 195 | qs = User.objects.all() |
| 196 | else: |
| 197 | # can see all users at any site where this user has pi role |
| 198 | from core.models.site import SitePrivilege |
| 199 | site_privs = SitePrivilege.objects.filter(user=user) |
| 200 | sites = [sp.site for sp in site_privs if sp.role.role == 'pi'] |
| 201 | # get site privs of users at these sites |
| 202 | site_privs = SitePrivilege.objects.filter(site__in=sites) |
Scott Baker | a36d77e | 2014-08-29 11:43:23 -0700 | [diff] [blame] | 203 | user_ids = [sp.user.id for sp in site_privs] + [user.id] |
Tony Mack | 5b06147 | 2014-02-04 07:57:10 -0500 | [diff] [blame] | 204 | qs = User.objects.filter(Q(site__in=sites) | Q(id__in=user_ids)) |
Scott Baker | a36d77e | 2014-08-29 11:43:23 -0700 | [diff] [blame] | 205 | return qs |
Tony Mack | 5b06147 | 2014-02-04 07:57:10 -0500 | [diff] [blame] | 206 | |
Scott Baker | 2c3cb64 | 2014-05-19 17:55:56 -0700 | [diff] [blame] | 207 | class UserDashboardView(PlCoreBase): |
| 208 | user = models.ForeignKey(User, related_name="dashboardViews") |
| 209 | dashboardView = models.ForeignKey(DashboardView, related_name="dashboardViews") |
| 210 | order = models.IntegerField(default=0) |