adding model and signal to user post_save - closes #104

Luan
1 parent f21f3dd5
Showing 1 changed file with 26 additions and 0 deletions Show diff stats
src/proxy/models.py
@@ -5,6 +5,8 @@ import urllib2
  
 from django.conf import settings
 from django.db import models
+from django.db.models.signals import post_save
+from django.dispatch import receiver
  
 from accounts.models import User
 from hitcounter.models import HitCounterModelMixin
@@ -148,3 +150,27 @@ class TicketCollabCount(models.Model):
     class Meta:
         managed = False
         db_table = 'ticket_collab_count_view'
+
+
+class SessionAttribute(models.Model):
+    sid = models.TextField(primary_key=True)
+    authenticated = models.IntegerField()
+    name = models.TextField()
+    value = models.TextField(blank=True)
+
+    class Meta:
+        managed = False
+        db_table = 'session_attribute'
+
+
+@receiver(post_save, sender=User)
+def change_session_attribute_email(sender, instance, **kwargs):
+    try:
+        session_attr = SessionAttribute.objects.get(
+            sid=instance.username, name='email'
+        )
+    except SessionAttribute.DoesNotExist:
+        pass
+    else:
+        session_attr.value = instance.email
+        session_attr.save()
...	...	@@ -5,6 +5,8 @@ import urllib2
5	5
6	6	from django.conf import settings
7	7	from django.db import models
	8	+from django.db.models.signals import post_save
	9	+from django.dispatch import receiver
8	10
9	11	from accounts.models import User
10	12	from hitcounter.models import HitCounterModelMixin
...	...	@@ -148,3 +150,27 @@ class TicketCollabCount(models.Model):
148	150	class Meta:
149	151	managed = False
150	152	db_table = 'ticket_collab_count_view'
	153	+
	154	+
	155	+class SessionAttribute(models.Model):
	156	+ sid = models.TextField(primary_key=True)
	157	+ authenticated = models.IntegerField()
	158	+ name = models.TextField()
	159	+ value = models.TextField(blank=True)
	160	+
	161	+ class Meta:
	162	+ managed = False
	163	+ db_table = 'session_attribute'
	164	+
	165	+
	166	+@receiver(post_save, sender=User)
	167	+def change_session_attribute_email(sender, instance, **kwargs):
	168	+ try:
	169	+ session_attr = SessionAttribute.objects.get(
	170	+ sid=instance.username, name='email'
	171	+ )
	172	+ except SessionAttribute.DoesNotExist:
	173	+ pass
	174	+ else:
	175	+ session_attr.value = instance.email
	176	+ session_attr.save()
...	...