mirror of
https://github.com/django/django.git
synced 2025-10-25 22:56:12 +00:00
Dropped support for SpatiaLite < 2.4.
This commit is contained in:
@@ -1,76 +1,8 @@
|
||||
import os
|
||||
|
||||
from django.conf import settings
|
||||
from django.core.exceptions import ImproperlyConfigured
|
||||
from django.db.backends.sqlite3.creation import DatabaseCreation
|
||||
|
||||
|
||||
class SpatiaLiteCreation(DatabaseCreation):
|
||||
|
||||
def create_test_db(self, verbosity=1, autoclobber=False, serialize=True, keepdb=False):
|
||||
"""
|
||||
Creates a test database, prompting the user for confirmation if the
|
||||
database already exists. Returns the name of the test database created.
|
||||
|
||||
This method is overloaded to load up the SpatiaLite initialization
|
||||
SQL prior to calling the `migrate` command.
|
||||
"""
|
||||
# Don't import django.core.management if it isn't needed.
|
||||
from django.core.management import call_command
|
||||
|
||||
test_database_name = self._get_test_db_name()
|
||||
|
||||
if verbosity >= 1:
|
||||
test_db_repr = ''
|
||||
action = 'Creating'
|
||||
if verbosity >= 2:
|
||||
test_db_repr = " ('%s')" % test_database_name
|
||||
if keepdb:
|
||||
action = 'Using existing'
|
||||
print("%s test database for alias '%s'%s..." % (
|
||||
action, self.connection.alias, test_db_repr))
|
||||
|
||||
self._create_test_db(verbosity, autoclobber, keepdb)
|
||||
|
||||
self.connection.close()
|
||||
self.connection.settings_dict["NAME"] = test_database_name
|
||||
|
||||
# Need to load the SpatiaLite initialization SQL before running `migrate`.
|
||||
self.load_spatialite_sql()
|
||||
|
||||
# Report migrate messages at one level lower than that requested.
|
||||
# This ensures we don't get flooded with messages during testing
|
||||
# (unless you really ask to be flooded)
|
||||
call_command('migrate',
|
||||
verbosity=max(verbosity - 1, 0),
|
||||
interactive=False,
|
||||
database=self.connection.alias,
|
||||
load_initial_data=False)
|
||||
|
||||
# We then serialize the current state of the database into a string
|
||||
# and store it on the connection. This slightly horrific process is so people
|
||||
# who are testing on databases without transactions or who are using
|
||||
# a TransactionTestCase still get a clean database on every test run.
|
||||
if serialize:
|
||||
self.connection._test_serialized_contents = self.serialize_db_to_string()
|
||||
|
||||
# We need to then do a flush to ensure that any data installed by
|
||||
# custom SQL has been removed. The only test data should come from
|
||||
# test fixtures, or autogenerated from post_migrate triggers.
|
||||
# This has the side effect of loading initial data (which was
|
||||
# intentionally skipped in the migrate).
|
||||
call_command('flush',
|
||||
verbosity=max(verbosity - 1, 0),
|
||||
interactive=False,
|
||||
database=self.connection.alias)
|
||||
|
||||
call_command('createcachetable', database=self.connection.alias)
|
||||
|
||||
# Ensure a connection for the side effect of initializing the test database.
|
||||
self.connection.ensure_connection()
|
||||
|
||||
return test_database_name
|
||||
|
||||
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
|
||||
@@ -99,35 +31,9 @@ class SpatiaLiteCreation(DatabaseCreation):
|
||||
|
||||
return output
|
||||
|
||||
def load_spatialite_sql(self):
|
||||
def _create_test_db_pre_migrate_sql(self):
|
||||
"""
|
||||
This routine loads up the SpatiaLite SQL file.
|
||||
Creates the spatial metadata tables.
|
||||
"""
|
||||
if self.connection.ops.spatial_version[:2] >= (2, 4):
|
||||
# Spatialite >= 2.4 -- No need to load any SQL file, calling
|
||||
# InitSpatialMetaData() transparently creates the spatial metadata
|
||||
# tables
|
||||
cur = self.connection._cursor()
|
||||
cur.execute("SELECT InitSpatialMetaData()")
|
||||
else:
|
||||
# Spatialite < 2.4 -- Load the initial SQL
|
||||
|
||||
# Getting the location of the SpatiaLite SQL file, and confirming
|
||||
# it exists.
|
||||
spatialite_sql = self.spatialite_init_file()
|
||||
if not os.path.isfile(spatialite_sql):
|
||||
raise ImproperlyConfigured('Could not find the required SpatiaLite initialization '
|
||||
'SQL file (necessary for testing): %s' % spatialite_sql)
|
||||
|
||||
# Opening up the SpatiaLite SQL initialization file and executing
|
||||
# as a script.
|
||||
with open(spatialite_sql, 'r') as sql_fh:
|
||||
cur = self.connection._cursor()
|
||||
cur.executescript(sql_fh.read())
|
||||
|
||||
def spatialite_init_file(self):
|
||||
# SPATIALITE_SQL may be placed in settings to tell GeoDjango
|
||||
# to use a specific path to the SpatiaLite initialization SQL.
|
||||
return getattr(settings, 'SPATIALITE_SQL',
|
||||
'init_spatialite-%s.%s.sql' %
|
||||
self.connection.ops.spatial_version[:2])
|
||||
cur = self.connection._cursor()
|
||||
cur.execute("SELECT InitSpatialMetaData()")
|
||||
|
||||
Reference in New Issue
Block a user