Ticket #6064: 6064-using-signals.diff

django/db/backends/ado_mssql/base.py

@@ -5 +5 @@
 """
 
 from django.db.backends import BaseDatabaseWrapper, BaseDatabaseFeatures, BaseDatabaseOperations, util
+from django.dispatch import dispatcher
+from django.db.backends import signals
 try:
     import adodbapi as Database
 except ImportError, e:
@@ -109 +111 @@
             # TODO: Handle DATABASE_PORT.
             conn_string = "PROVIDER=SQLOLEDB;DATA SOURCE=%s;UID=%s;PWD=%s;DATABASE=%s" % (settings.DATABASE_HOST, settings.DATABASE_USER, settings.DATABASE_PASSWORD, settings.DATABASE_NAME)
             self.connection = Database.connect(conn_string)
+        dispatcher.send(signal=signals.connection_created, sender=self.__class__)
         return self.connection.cursor()

django/db/backends/mysql_old/base.py

@@ -5 +5 @@
 """
 
 from django.db.backends import BaseDatabaseWrapper, BaseDatabaseFeatures, BaseDatabaseOperations, util
+from django.dispatch import dispatcher
+from django.db.backends import signals
 from django.utils.encoding import force_unicode
 try:
     import MySQLdb as Database
@@ -194 +196 @@
                     self.connection.set_character_set('utf8')
         else:
             cursor = self.connection.cursor()
+        dispatcher.send(signal=signals.connection_created, sender=self.__class__)
         return cursor
 
     def make_debug_cursor(self, cursor):

django/db/backends/postgresql/base.py

@@ -6 +6 @@
 
 from django.utils.encoding import smart_str, smart_unicode
 from django.db.backends import BaseDatabaseWrapper, BaseDatabaseFeatures, util
+from django.dispatch import dispatcher
+from django.db.backends import signals
 from django.db.backends.postgresql.operations import DatabaseOperations
 try:
     import psycopg as Database
@@ -102 +104 @@
             cursor.execute("SET TIME ZONE %s", [settings.TIME_ZONE])
         cursor.execute("SET client_encoding to 'UNICODE'")
         cursor = UnicodeCursorWrapper(cursor, 'utf-8')
+        dispatcher.send(signal=signals.connection_created, sender=self.__class__)
         return cursor
 
 def typecast_string(s):

django/db/backends/sqlite3/base.py

@@ -7 +7 @@
 """
 
 from django.db.backends import BaseDatabaseWrapper, BaseDatabaseFeatures, BaseDatabaseOperations, util
+from django.dispatch import dispatcher
+from django.db.backends import signals
 try:
     try:
         from sqlite3 import dbapi2 as Database
@@ -112 +114 @@
             self.connection.create_function("django_extract", 2, _sqlite_extract)
             self.connection.create_function("django_date_trunc", 2, _sqlite_date_trunc)
             self.connection.create_function("regexp", 2, _sqlite_regexp)
+        dispatcher.send(signal=signals.connection_created, sender=self.__class__)
         return self.connection.cursor(factory=SQLiteCursorWrapper)
 
     def close(self):

django/db/backends/mysql/base.py

@@ -5 +5 @@
 """
 
 from django.db.backends import BaseDatabaseWrapper, BaseDatabaseFeatures, BaseDatabaseOperations, util
+from django.dispatch import dispatcher
+from django.db.backends import signals
 try:
     import MySQLdb as Database
 except ImportError, e:
@@ -180 +182 @@
                 kwargs['port'] = int(settings.DATABASE_PORT)
             kwargs.update(self.options)
             self.connection = Database.connect(**kwargs)
+        dispatcher.send(signal=signals.connection_created, sender=self.__class__)
         cursor = self.connection.cursor()
         return cursor
 

django/db/backends/oracle/base.py

@@ -5 +5 @@
 """
 
 from django.db.backends import BaseDatabaseWrapper, BaseDatabaseFeatures, BaseDatabaseOperations, util
+from django.dispatch import dispatcher
+from django.db.backends import signals
 from django.utils.datastructures import SortedDict
 from django.utils.encoding import smart_str, force_unicode
 import datetime
@@ -432 +434 @@
         # Set oracle date to ansi date format.
         cursor.execute("ALTER SESSION SET NLS_DATE_FORMAT = 'YYYY-MM-DD'")
         cursor.execute("ALTER SESSION SET NLS_TIMESTAMP_FORMAT = 'YYYY-MM-DD HH24:MI:SS.FF'")
+        dispatcher.send(signal=signals.connection_created, sender=self.__class__)
         return cursor
 
 class FormatStylePlaceholderCursor(Database.Cursor):

django/db/backends/signals.py

@@ +1 @@
+connection_created = object()
+ No newline at end of file

django/db/backends/postgresql_psycopg2/base.py

@@ -6 +6 @@
 
 from django.db.backends import BaseDatabaseWrapper, BaseDatabaseFeatures
 from django.db.backends.postgresql.operations import DatabaseOperations as PostgresqlDatabaseOperations
+from django.dispatch import dispatcher
+from django.db.backends import signals
 from django.utils.safestring import SafeUnicode
 try:
     import psycopg2 as Database
@@ -73 +75 @@
         cursor.tzinfo_factory = None
         if set_tz:
             cursor.execute("SET TIME ZONE %s", [settings.TIME_ZONE])
+        dispatcher.send(signal=signals.connection_created, sender=self.__class__)
         return cursor

tests/modeltests/signals/models.py

@@ -2 +2 @@
 Testing signals before/after saving and deleting.
 """
 
+from django.dispatch import dispatcher
 from django.db import models
-from django.dispatch import dispatcher
+from django.db import connection
+from django.db.backends.signals import connection_created
 
 class Person(models.Model):
     first_name = models.CharField(max_length=20)
@@ -32 +34 @@
     print 'post_delete signal,', instance
     print 'instance.id is None: %s' % (instance.id == None)  
 
+def connection_created_test(sender, **kwargs):
+    print 'connection_created signal'
+
 __test__ = {'API_TESTS':"""
+>>> dispatcher.connect(connection_created_test, signal=connection_created)
 >>> dispatcher.connect(pre_save_test, signal=models.signals.pre_save)
 >>> dispatcher.connect(post_save_test, signal=models.signals.post_save)
 >>> dispatcher.connect(pre_delete_test, signal=models.signals.pre_delete)
 >>> dispatcher.connect(post_delete_test, signal=models.signals.post_delete)
 
+>>> cursor = connection.cursor()
+connection_created signal
+
+>>> dispatcher.disconnect(connection_created_test, signal=connection_created)
+>>> cursor = connection.cursor()
+
 >>> p1 = Person(first_name='John', last_name='Smith')
 >>> p1.save()
 pre_save signal, John Smith