Ticket #9633: 9633-adaptor-query-params-sqlite3.diff

django/db/backends/sqlite3/base.py

@@ -10 +10 @@
 from django.db.backends.sqlite3.client import DatabaseClient
 from django.db.backends.sqlite3.creation import DatabaseCreation
 from django.db.backends.sqlite3.introspection import DatabaseIntrospection
+from django.utils.encoding import force_unicode
 
 try:
     try:
@@ -165 +166 @@
     """
     def execute(self, query, params=()):
         query = self.convert_query(query, len(params))
-        return Database.Cursor.execute(self, query, params)
+        new_params = [self.convert_param(p) for p in params]
+        return Database.Cursor.execute(self, query, new_params)
 
     def executemany(self, query, param_list):
         try:
           query = self.convert_query(query, len(param_list[0]))
-          return Database.Cursor.executemany(self, query, param_list)
+          new_param_list = [[self.convert_param(p) for p in params] for params in param_list]
+          return Database.Cursor.executemany(self, query, new_param_list)
         except (IndexError,TypeError):
           # No parameter list provided
           return None
@@ -178 +181 @@
     def convert_query(self, query, num_params):
         return query % tuple("?" * num_params)
 
+    def convert_param(self, param):
+        # These are the types that sqlite3 can handle directly
+        if param is None or type(param) in (int,long,float,str,unicode,buffer,bool):
+            return param
+        # The rest we must convert to unicode first
+        return force_unicode(param)
+
 def _sqlite_extract(lookup_type, dt):
     if dt is None:
         return None

tests/regressiontests/backends/models.py

@@ -15 +15 @@
     def __unicode__(self):
         return u'%s %s' % (self.first_name, self.last_name)
 
+class Adaptor(object):
+    def __init__(self, value):
+        self.value = value
+
+    def __str__(self):
+        return self.value
+
 qn = connection.ops.quote_name
 
 __test__ = {'API_TESTS': """
@@ -56 +63 @@
 >>> list(cursor.fetchall())
 [(u'Mary', u'Agnelline'), (u'Peter', u'Parker')]
 
+#9633: Using adaptor objects as query parameters with sqlite3
+>>> query = 'SELECT %s'
+>>> param = Adaptor(u'Arthur')
+>>> foo = cursor.execute(query, (param,))
+>>> cursor.fetchone()
+(u'Arthur',)
 """}