39 lines
1.8 KiB
Python
39 lines
1.8 KiB
Python
from django.conf import settings
|
|
from django.db.backends.creation import BaseDatabaseCreation
|
|
|
|
class DatabaseCreation(BaseDatabaseCreation):
|
|
# This dictionary maps Field objects to their associated PostgreSQL column
|
|
# types, as strings. Column-type strings can contain format strings; they'll
|
|
# be interpolated against the values of Field.__dict__ before being output.
|
|
# If a column type is set to None, it won't be included in the output.
|
|
data_types = {
|
|
'AutoField': 'serial',
|
|
'BooleanField': 'boolean',
|
|
'CharField': 'varchar(%(max_length)s)',
|
|
'CommaSeparatedIntegerField': 'varchar(%(max_length)s)',
|
|
'DateField': 'date',
|
|
'DateTimeField': 'timestamp with time zone',
|
|
'DecimalField': 'numeric(%(max_digits)s, %(decimal_places)s)',
|
|
'FileField': 'varchar(%(max_length)s)',
|
|
'FilePathField': 'varchar(%(max_length)s)',
|
|
'FloatField': 'double precision',
|
|
'IntegerField': 'integer',
|
|
'IPAddressField': 'inet',
|
|
'NullBooleanField': 'boolean',
|
|
'OneToOneField': 'integer',
|
|
'PhoneNumberField': 'varchar(20)',
|
|
'PositiveIntegerField': 'integer CHECK ("%(column)s" >= 0)',
|
|
'PositiveSmallIntegerField': 'smallint CHECK ("%(column)s" >= 0)',
|
|
'SlugField': 'varchar(%(max_length)s)',
|
|
'SmallIntegerField': 'smallint',
|
|
'TextField': 'text',
|
|
'TimeField': 'time',
|
|
'USStateField': 'varchar(2)',
|
|
}
|
|
|
|
def sql_table_creation_suffix(self):
|
|
assert settings.TEST_DATABASE_COLLATION is None, "PostgreSQL does not support collation setting at database creation time."
|
|
if settings.TEST_DATABASE_CHARSET:
|
|
return "WITH ENCODING '%s'" % settings.TEST_DATABASE_CHARSET
|
|
return ''
|