From b9bf7cce896a4886f77b2162042a6fb676463b86 Mon Sep 17 00:00:00 2001 From: Carl Meyer Date: Fri, 2 Mar 2012 22:32:22 +0000 Subject: [PATCH] Fixed #15169 -- Added conversion of 0/1 to False/True for MySQL GIS backend. Thanks zmsmith for report, and Ramiro for draft patch and review. git-svn-id: http://code.djangoproject.com/svn/django/trunk@17632 bcc190cf-cafb-0310-a4f2-bffc1f526a37 --- .../contrib/gis/db/backends/mysql/compiler.py | 32 +++++++++++++++++++ .../gis/db/backends/mysql/operations.py | 2 +- django/contrib/gis/tests/geoapp/models.py | 4 +++ .../contrib/gis/tests/geoapp/test_regress.py | 16 +++++++++- 4 files changed, 52 insertions(+), 2 deletions(-) create mode 100644 django/contrib/gis/db/backends/mysql/compiler.py diff --git a/django/contrib/gis/db/backends/mysql/compiler.py b/django/contrib/gis/db/backends/mysql/compiler.py new file mode 100644 index 0000000000..7079db9f6a --- /dev/null +++ b/django/contrib/gis/db/backends/mysql/compiler.py @@ -0,0 +1,32 @@ +from django.contrib.gis.db.models.sql.compiler import GeoSQLCompiler as BaseGeoSQLCompiler +from django.db.backends.mysql import compiler + +SQLCompiler = compiler.SQLCompiler + +class GeoSQLCompiler(BaseGeoSQLCompiler, SQLCompiler): + def resolve_columns(self, row, fields=()): + """ + Integrate the cases handled both by the base GeoSQLCompiler and the + main MySQL compiler (converting 0/1 to True/False for boolean fields). + + Refs #15169. + + """ + row = BaseGeoSQLCompiler.resolve_columns(self, row, fields) + return SQLCompiler.resolve_columns(self, row, fields) + + +class SQLInsertCompiler(compiler.SQLInsertCompiler, GeoSQLCompiler): + pass + +class SQLDeleteCompiler(compiler.SQLDeleteCompiler, GeoSQLCompiler): + pass + +class SQLUpdateCompiler(compiler.SQLUpdateCompiler, GeoSQLCompiler): + pass + +class SQLAggregateCompiler(compiler.SQLAggregateCompiler, GeoSQLCompiler): + pass + +class SQLDateCompiler(compiler.SQLDateCompiler, GeoSQLCompiler): + pass diff --git a/django/contrib/gis/db/backends/mysql/operations.py b/django/contrib/gis/db/backends/mysql/operations.py index 1653636dd4..c0e5aa6691 100644 --- a/django/contrib/gis/db/backends/mysql/operations.py +++ b/django/contrib/gis/db/backends/mysql/operations.py @@ -5,7 +5,7 @@ from django.contrib.gis.db.backends.base import BaseSpatialOperations class MySQLOperations(DatabaseOperations, BaseSpatialOperations): - compiler_module = 'django.contrib.gis.db.models.sql.compiler' + compiler_module = 'django.contrib.gis.db.backends.mysql.compiler' mysql = True name = 'mysql' select = 'AsText(%s)' diff --git a/django/contrib/gis/tests/geoapp/models.py b/django/contrib/gis/tests/geoapp/models.py index 1ac830953a..79061e1cfc 100644 --- a/django/contrib/gis/tests/geoapp/models.py +++ b/django/contrib/gis/tests/geoapp/models.py @@ -34,6 +34,10 @@ class Track(models.Model): objects = models.GeoManager() def __unicode__(self): return self.name +class Truth(models.Model): + val = models.BooleanField() + objects = models.GeoManager() + if not spatialite: class Feature(models.Model): name = models.CharField(max_length=20) diff --git a/django/contrib/gis/tests/geoapp/test_regress.py b/django/contrib/gis/tests/geoapp/test_regress.py index 19f31e66be..a9d802d8f1 100644 --- a/django/contrib/gis/tests/geoapp/test_regress.py +++ b/django/contrib/gis/tests/geoapp/test_regress.py @@ -7,7 +7,7 @@ from django.contrib.gis.shortcuts import render_to_kmz from django.db.models import Count from django.test import TestCase -from .models import City, PennsylvaniaCity, State +from .models import City, PennsylvaniaCity, State, Truth class GeoRegressionTests(TestCase): @@ -64,3 +64,17 @@ class GeoRegressionTests(TestCase): "Regression for #16409. Make sure defer() and only() work with annotate()" self.assertIsInstance(list(City.objects.annotate(Count('point')).defer('name')), list) self.assertIsInstance(list(City.objects.annotate(Count('point')).only('name')), list) + + def test07_boolean_conversion(self): + "Testing Boolean value conversion with the spatial backend, see #15169." + t1 = Truth.objects.create(val=True) + t2 = Truth.objects.create(val=False) + + val1 = Truth.objects.get(pk=1).val + val2 = Truth.objects.get(pk=2).val + # verify types -- should't be 0/1 + self.assertIsInstance(val1, bool) + self.assertIsInstance(val2, bool) + # verify values + self.assertEqual(val1, True) + self.assertEqual(val2, False)