mirror of
				https://github.com/django/django.git
				synced 2025-10-31 09:41:08 +00:00 
			
		
		
		
	[1.7.x] Fixed removal of GIS column in PostGIS 1.x migration
Refs #22481. Backport of2f9d1576efrom master, squashed with2f9d1576e8.
This commit is contained in:
		| @@ -7,6 +7,7 @@ class PostGISSchemaEditor(DatabaseSchemaEditor): | |||||||
|     geom_index_ops_nd = 'GIST_GEOMETRY_OPS_ND' |     geom_index_ops_nd = 'GIST_GEOMETRY_OPS_ND' | ||||||
|  |  | ||||||
|     sql_add_geometry_column = "SELECT AddGeometryColumn(%(table)s, %(column)s, %(srid)s, %(geom_type)s, %(dim)s)" |     sql_add_geometry_column = "SELECT AddGeometryColumn(%(table)s, %(column)s, %(srid)s, %(geom_type)s, %(dim)s)" | ||||||
|  |     sql_drop_geometry_column = "SELECT DropGeometryColumn(%(table)s, %(column)s)" | ||||||
|     sql_alter_geometry_column_not_null = "ALTER TABLE %(table)s ALTER COLUMN %(column)s SET NOT NULL" |     sql_alter_geometry_column_not_null = "ALTER TABLE %(table)s ALTER COLUMN %(column)s SET NOT NULL" | ||||||
|     sql_add_spatial_index = "CREATE INDEX %(index)s ON %(table)s USING %(index_type)s (%(column)s %(ops)s)" |     sql_add_spatial_index = "CREATE INDEX %(index)s ON %(table)s USING %(index_type)s (%(column)s %(ops)s)" | ||||||
|  |  | ||||||
| @@ -88,3 +89,15 @@ class PostGISSchemaEditor(DatabaseSchemaEditor): | |||||||
|         for sql in self.geometry_sql: |         for sql in self.geometry_sql: | ||||||
|             self.execute(sql) |             self.execute(sql) | ||||||
|         self.geometry_sql = [] |         self.geometry_sql = [] | ||||||
|  |  | ||||||
|  |     def remove_field(self, model, field): | ||||||
|  |         from django.contrib.gis.db.models.fields import GeometryField | ||||||
|  |         if not isinstance(field, GeometryField) or self.connection.ops.spatial_version > (2, 0): | ||||||
|  |             super(PostGISSchemaEditor, self).remove_field(model, field) | ||||||
|  |         else: | ||||||
|  |             self.execute( | ||||||
|  |                 self.sql_drop_geometry_column % { | ||||||
|  |                     "table": self.geo_quote_name(model._meta.db_table), | ||||||
|  |                     "column": self.geo_quote_name(field.column), | ||||||
|  |                 } | ||||||
|  |             ) | ||||||
|   | |||||||
| @@ -10,6 +10,11 @@ from django.test import TransactionTestCase | |||||||
|  |  | ||||||
| if HAS_SPATIAL_DB: | if HAS_SPATIAL_DB: | ||||||
|     from django.contrib.gis.db.models import fields |     from django.contrib.gis.db.models import fields | ||||||
|  |     try: | ||||||
|  |         from django.contrib.gis.models import GeometryColumns | ||||||
|  |         HAS_GEOMETRY_COLUMNS = True | ||||||
|  |     except ImportError: | ||||||
|  |         HAS_GEOMETRY_COLUMNS = False | ||||||
|  |  | ||||||
|  |  | ||||||
| @skipUnless(HAS_SPATIAL_DB, "Spatial db is required.") | @skipUnless(HAS_SPATIAL_DB, "Spatial db is required.") | ||||||
| @@ -23,6 +28,9 @@ class OperationTests(TransactionTestCase): | |||||||
|     def assertColumnExists(self, table, column): |     def assertColumnExists(self, table, column): | ||||||
|         self.assertIn(column, [c.name for c in self.get_table_description(table)]) |         self.assertIn(column, [c.name for c in self.get_table_description(table)]) | ||||||
|  |  | ||||||
|  |     def assertColumnNotExists(self, table, column): | ||||||
|  |         self.assertNotIn(column, [c.name for c in self.get_table_description(table)]) | ||||||
|  |  | ||||||
|     def apply_operations(self, app_label, project_state, operations): |     def apply_operations(self, app_label, project_state, operations): | ||||||
|         migration = Migration('name', app_label) |         migration = Migration('name', app_label) | ||||||
|         migration.operations = operations |         migration.operations = operations | ||||||
| @@ -30,6 +38,17 @@ class OperationTests(TransactionTestCase): | |||||||
|             return migration.apply(project_state, editor) |             return migration.apply(project_state, editor) | ||||||
|  |  | ||||||
|     def set_up_test_model(self): |     def set_up_test_model(self): | ||||||
|  |         # Delete the tables if they already exist | ||||||
|  |         with connection.cursor() as cursor: | ||||||
|  |             try: | ||||||
|  |                 cursor.execute("DROP TABLE %s" % connection.ops.quote_name("gis_neighborhood")) | ||||||
|  |             except: | ||||||
|  |                 pass | ||||||
|  |             else: | ||||||
|  |                 if HAS_GEOMETRY_COLUMNS: | ||||||
|  |                     cursor.execute("DELETE FROM geometry_columns WHERE %s = %%s" % ( | ||||||
|  |                         GeometryColumns.table_name_col(),), ["gis_neighborhood"]) | ||||||
|  |  | ||||||
|         operations = [migrations.CreateModel( |         operations = [migrations.CreateModel( | ||||||
|             "Neighborhood", |             "Neighborhood", | ||||||
|             [ |             [ | ||||||
| @@ -57,11 +76,27 @@ class OperationTests(TransactionTestCase): | |||||||
|         self.assertColumnExists("gis_neighborhood", "path") |         self.assertColumnExists("gis_neighborhood", "path") | ||||||
|  |  | ||||||
|         # Test GeometryColumns when available |         # Test GeometryColumns when available | ||||||
|         try: |         if HAS_GEOMETRY_COLUMNS: | ||||||
|             from django.contrib.gis.models import GeometryColumns |             self.assertEqual( | ||||||
|         except ImportError: |                 GeometryColumns.objects.filter(**{GeometryColumns.table_name_col(): "gis_neighborhood"}).count(), | ||||||
|             return |                 2 | ||||||
|         self.assertEqual( |             ) | ||||||
|             GeometryColumns.objects.filter(**{GeometryColumns.table_name_col(): "gis_neighborhood"}).count(), |  | ||||||
|             2 |     def test_remove_gis_field(self): | ||||||
|         ) |         """ | ||||||
|  |         Tests the RemoveField operation with a GIS-enabled column. | ||||||
|  |         """ | ||||||
|  |         project_state = self.set_up_test_model() | ||||||
|  |         operation = migrations.RemoveField("Neighborhood", "geom") | ||||||
|  |         new_state = project_state.clone() | ||||||
|  |         operation.state_forwards("gis", new_state) | ||||||
|  |         with connection.schema_editor() as editor: | ||||||
|  |             operation.database_forwards("gis", editor, project_state, new_state) | ||||||
|  |         self.assertColumnNotExists("gis_neighborhood", "geom") | ||||||
|  |  | ||||||
|  |         # Test GeometryColumns when available | ||||||
|  |         if HAS_GEOMETRY_COLUMNS: | ||||||
|  |             self.assertEqual( | ||||||
|  |                 GeometryColumns.objects.filter(**{GeometryColumns.table_name_col(): "gis_neighborhood"}).count(), | ||||||
|  |                 0 | ||||||
|  |             ) | ||||||
|   | |||||||
		Reference in New Issue
	
	Block a user