using raw sql to update trac profile info - Updates #104

Luan
1 parent e7d91a5e
Showing 1 changed file with 9 additions and 21 deletions Show diff stats
src/proxy/models.py
@@ -4,7 +4,7 @@ import os
 import urllib2
  
 from django.conf import settings
-from django.db import models
+from django.db import models, connections
 from django.db.models.signals import post_save
 from django.dispatch import receiver
  
@@ -152,25 +152,13 @@ class TicketCollabCount(models.Model):
         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()
+    cursor = connections['trac'].cursor()
+
+    cursor.execute(("UPDATE session_attribute SET value = '%s' "
+                    "WHERE name='email' AND sid='%s'"),
+                    [instance.email, instance.username])
+    cursor.execute(("UPDATE session_attribute SET value = '%s' "
+                    "WHERE name='name' AND sid='%s'"),
+                    [instance.get_full_name(), instance.username])
...	...	@@ -4,7 +4,7 @@ import os
4	4	import urllib2
5	5
6	6	from django.conf import settings
7		-from django.db import models
	7	+from django.db import models, connections
8	8	from django.db.models.signals import post_save
9	9	from django.dispatch import receiver
10	10
...	...	@@ -152,25 +152,13 @@ class TicketCollabCount(models.Model):
152	152	db_table = 'ticket_collab_count_view'
153	153
154	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	155	@receiver(post_save, sender=User)
167	156	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()
	157	+ cursor = connections['trac'].cursor()
	158	+
	159	+ cursor.execute(("UPDATE session_attribute SET value = '%s' "
	160	+ "WHERE name='email' AND sid='%s'"),
	161	+ [instance.email, instance.username])
	162	+ cursor.execute(("UPDATE session_attribute SET value = '%s' "
	163	+ "WHERE name='name' AND sid='%s'"),
	164	+ [instance.get_full_name(), instance.username])
...	...