Fixed #971 -- inspectdb for SQLite now introspects field types.

git-svn-id: http://code.djangoproject.com/svn/django/trunk@1518 bcc190cf-cafb-0310-a4f2-bffc1f526a37
This commit is contained in:
Adrian Holovaty 2005-12-01 06:32:25 +00:00
parent 38b8d18091
commit 9ede371c85
3 changed files with 46 additions and 5 deletions

View File

@ -185,4 +185,36 @@ DATA_TYPES = {
'USStateField': 'varchar(2)',
}
DATA_TYPES_REVERSE = {}
# Maps SQL types to Django Field types. Some of the SQL types have multiple
# entries here because SQLite allows for anything and doesn't normalize the
# field type; it uses whatever was given.
BASE_DATA_TYPES_REVERSE = {
'bool': 'BooleanField',
'boolean': 'BooleanField',
'smallint': 'SmallIntegerField',
'smallinteger': 'SmallIntegerField',
'int': 'IntegerField',
'integer': 'IntegerField',
'text': 'TextField',
'char': 'CharField',
'date': 'DateField',
'datetime': 'DateTimeField',
'time': 'TimeField',
}
# This light wrapper "fakes" a dictionary interface, because some SQLite data
# types include variables in them -- e.g. "varchar(30)" -- and can't be matched
# as a simple dictionary lookup.
class FlexibleFieldLookupDict:
def __getitem__(self, key):
key = key.lower()
try:
return BASE_DATA_TYPES_REVERSE[key]
except KeyError:
import re
m = re.search(r'^\s*(?:var)?char\s*\(\s*(\d+)\s*\)\s*$', key)
if m:
return ('CharField', {'maxlength': m.group(1)})
raise KeyError
DATA_TYPES_REVERSE = FlexibleFieldLookupDict()

View File

@ -591,9 +591,19 @@ def inspectdb(db_name):
field_type_was_guessed = True
else:
field_type_was_guessed = False
# This is a hook for DATA_TYPES_REVERSE to return a tuple of
# (field_type, extra_params_dict).
if type(field_type) is tuple:
field_type, extra_params = field_type
else:
extra_params = {}
if field_type == 'CharField' and row[3]:
extra_params['maxlength'] = row[3]
field_desc = '%s = meta.%s(' % (column_name, field_type)
if field_type == 'CharField':
field_desc += 'maxlength=%s' % (row[3])
field_desc += ', '.join(['%s=%s' % (k, v) for k, v in extra_params.items()])
field_desc += ')'
if field_type_was_guessed:
field_desc += ' # This is a guess!'

View File

@ -87,8 +87,7 @@ customizations. In particular, you'll need to do this:
doesn't yet introspect primary keys.
``inspectdb`` works with PostgreSQL, MySQL and SQLite. Foreign-key detection
only works in PostgreSQL. In SQLite, it cannot detect column types; it'll
use ``TextField`` for each column.
only works in PostgreSQL.
install [modelmodule modelmodule ...]
-------------------------------------