| from django.conf import settings |
| from django.db.backends.postgresql.creation import DatabaseCreation |
| |
| class PostGISCreation(DatabaseCreation): |
| geom_index_type = 'GIST' |
| geom_index_opts = 'GIST_GEOMETRY_OPS' |
| |
| def sql_indexes_for_field(self, model, f, style): |
| "Return any spatial index creation SQL for the field." |
| from django.contrib.gis.db.models.fields import GeometryField |
| |
| output = super(PostGISCreation, self).sql_indexes_for_field(model, f, style) |
| |
| if isinstance(f, GeometryField): |
| gqn = self.connection.ops.geo_quote_name |
| qn = self.connection.ops.quote_name |
| db_table = model._meta.db_table |
| |
| if f.geography: |
| # Geogrophy columns are created normally. |
| pass |
| else: |
| # Geometry columns are created by `AddGeometryColumn` |
| # stored procedure. |
| output.append(style.SQL_KEYWORD('SELECT ') + |
| style.SQL_TABLE('AddGeometryColumn') + '(' + |
| style.SQL_TABLE(gqn(db_table)) + ', ' + |
| style.SQL_FIELD(gqn(f.column)) + ', ' + |
| style.SQL_FIELD(str(f.srid)) + ', ' + |
| style.SQL_COLTYPE(gqn(f.geom_type)) + ', ' + |
| style.SQL_KEYWORD(str(f.dim)) + ');') |
| |
| if not f.null: |
| # Add a NOT NULL constraint to the field |
| output.append(style.SQL_KEYWORD('ALTER TABLE ') + |
| style.SQL_TABLE(qn(db_table)) + |
| style.SQL_KEYWORD(' ALTER ') + |
| style.SQL_FIELD(qn(f.column)) + |
| style.SQL_KEYWORD(' SET NOT NULL') + ';') |
| |
| |
| if f.spatial_index: |
| # Spatial indexes created the same way for both Geometry and |
| # Geography columns |
| if f.geography: |
| index_opts = '' |
| else: |
| index_opts = ' ' + style.SQL_KEYWORD(self.geom_index_opts) |
| output.append(style.SQL_KEYWORD('CREATE INDEX ') + |
| style.SQL_TABLE(qn('%s_%s_id' % (db_table, f.column))) + |
| style.SQL_KEYWORD(' ON ') + |
| style.SQL_TABLE(qn(db_table)) + |
| style.SQL_KEYWORD(' USING ') + |
| style.SQL_COLTYPE(self.geom_index_type) + ' ( ' + |
| style.SQL_FIELD(qn(f.column)) + index_opts + ' );') |
| return output |
| |
| def sql_table_creation_suffix(self): |
| qn = self.connection.ops.quote_name |
| return ' TEMPLATE %s' % qn(getattr(settings, 'POSTGIS_TEMPLATE', 'template_postgis')) |