forked from Patrowl/PatrowlManager
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathmodels.py
51 lines (39 loc) · 1.76 KB
/
models.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
from __future__ import unicode_literals
from django.db import models
from django.utils import timezone
from django.contrib.auth.models import User
from django.db.models.signals import post_save, post_delete
from django.dispatch import receiver
from events.models import Event
USER_STATUS = (
('ACTIVE', 'ACTIVE'),
('DISABLED', 'DISABLED'),
)
class Profile(models.Model):
user = models.OneToOneField(User, on_delete=models.CASCADE)
status = models.CharField(choices=USER_STATUS, default='ACTIVE', max_length=10)
bio = models.TextField(max_length=500, blank=True)
department = models.CharField(max_length=100, blank=True)
created_at = models.DateTimeField(default=timezone.now)
updated_at = models.DateTimeField(default=timezone.now)
def __str__(self):
return "profile_{}".format(self.user.username)
def save(self, *args, **kwargs):
if not self._state.adding:
self.updated_at = timezone.now()
super(Profile, self).save(*args, **kwargs)
def get_group(self):
return self
@receiver(post_save, sender=User)
def create_user_profile(sender, instance, created, **kwargs):
if created:
Profile.objects.create(user=instance, status='ACTIVE', bio="n/a", department="n/a")
Event.objects.create(message="[User] New user created (id={}): {}".format(instance.id, instance),
type="CREATE", severity="DEBUG")
@receiver(post_save, sender=User)
def save_user_profile(sender, instance, **kwargs):
instance.profile.save()
@receiver(post_delete, sender=User)
def delete_user_profile(sender, **kwargs):
Event.objects.create(message="[User] User '{}' deleted (id={})".format(kwargs['instance'], kwargs['instance'].id),
type="DELETE", severity="DEBUG")