mirrors/django
1
0
Fork 0
mirror of https://github.com/django/django.git synced 2025-07-05 10:19:20 +00:00
Code Issues 32 Releases Wiki Activity

Updated ansi sql schema builder to access db connection through model._default_manager.

Browse Source 
git-svn-id: http://code.djangoproject.com/svn/django/branches/multiple-db-support@3367 bcc190cf-cafb-0310-a4f2-bffc1f526a37
This commit is contained in:
Jason Pellerin 2006-07-19 02:38:31 +00:00
parent 8b77fe3195
commit ed4e198d11
1 changed files with 29 additions and 30 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

59
django/db/backends/ansi/sql.py
View File

@ -64,10 +64,10 @@ class SchemaBuilder(object):
self.models_already_seen.add(model) self.models_already_seen.add(model)
opts = model._meta opts = model._meta
info = opts.connection_info db = model._default_manager.db
backend = info.backend backend = db.backend
quote_name = backend.quote_name quote_name = backend.quote_name
data_types = info.get_creation_module().DATA_TYPES data_types = db.get_creation_module().DATA_TYPES
table_output = [] table_output = []
# pending field references, keyed by the model class # pending field references, keyed by the model class
@ -126,7 +126,7 @@ class SchemaBuilder(object):
full_statement.append(' %s%s' % full_statement.append(' %s%s' %
(line, i < len(table_output)-1 and ',' or '')) (line, i < len(table_output)-1 and ',' or ''))
full_statement.append(');') full_statement.append(');')
create = [BoundStatement('\n'.join(full_statement), opts.connection)] create = [BoundStatement('\n'.join(full_statement), db.connection)]
if (pending_references and if (pending_references and
backend.supports_constraints): backend.supports_constraints):
@ -142,7 +142,7 @@ class SchemaBuilder(object):
quote_name('%s_referencing_%s_%s' % (r_col, r_table, col)), quote_name('%s_referencing_%s_%s' % (r_col, r_table, col)),
quote_name(r_col), quote_name(r_table), quote_name(col)) quote_name(r_col), quote_name(r_table), quote_name(col))
pending.setdefault(rel_class, []).append( pending.setdefault(rel_class, []).append(
BoundStatement(sql, opts.connection)) BoundStatement(sql, db.connection))
return (create, pending) return (create, pending)
def get_create_indexes(self, model, style=None): def get_create_indexes(self, model, style=None):
@ -151,9 +151,9 @@ class SchemaBuilder(object):
""" """
if style is None: if style is None:
style = default_style style = default_style
info = model._meta.connection_info db = model._default_manager.db
backend = info.backend backend = db.backend
connection = info.connection connection = db.connection
output = [] output = []
for f in model._meta.fields: for f in model._meta.fields:
if f.db_index: if f.db_index:
@ -182,10 +182,10 @@ class SchemaBuilder(object):
""" """
if style is None: if style is None:
style = default_style style = default_style
info = model._meta.connection_info db = model._default_manager.db
quote_name = info.backend.quote_name quote_name = db.backend.quote_name
connection = info.connection connection = db.connection
data_types = info.get_creation_module().DATA_TYPES data_types = db.get_creation_module().DATA_TYPES
opts = model._meta opts = model._meta
# statements to execute, keyed by the other model # statements to execute, keyed by the other model
@ -229,15 +229,15 @@ class SchemaBuilder(object):
if style is None: if style is None:
style = default_style style = default_style
opts = model._meta opts = model._meta
info = opts.connection_info db = model._default_manager.db
db_table = opts.db_table db_table = opts.db_table
backend = info.backend backend = db.backend
qn = backend.quote_name qn = backend.quote_name
output = [] output = []
output.append(BoundStatement( output.append(BoundStatement(
'%s %s;' % (style.SQL_KEYWORD('DROP TABLE'), '%s %s;' % (style.SQL_KEYWORD('DROP TABLE'),
style.SQL_TABLE(qn(db_table))), style.SQL_TABLE(qn(db_table))),
info.connection)) db.connection))
if cascade: if cascade:
# deal with others that might have a foreign key TO me: alter # deal with others that might have a foreign key TO me: alter
@ -247,7 +247,7 @@ class SchemaBuilder(object):
if model in references_to_delete: if model in references_to_delete:
for rel_class, f in references_to_delete[model]: for rel_class, f in references_to_delete[model]:
table = rel_class._meta.db_table table = rel_class._meta.db_table
if not self.table_exists(info, table): if not self.table_exists(db, table):
continue continue
col = f.column col = f.column
r_table = opts.db_table r_table = opts.db_table
@ -260,7 +260,7 @@ class SchemaBuilder(object):
backend.get_drop_foreignkey_sql()), backend.get_drop_foreignkey_sql()),
style.SQL_FIELD(qn("%s_referencing_%s_%s" % style.SQL_FIELD(qn("%s_referencing_%s_%s" %
(col, r_table, r_col)))), (col, r_table, r_col)))),
info.connection)) db.connection))
del references_to_delete[model] del references_to_delete[model]
# many to many: drop any many-many tables that are my # many to many: drop any many-many tables that are my
# responsiblity # responsiblity
@ -270,16 +270,16 @@ class SchemaBuilder(object):
'%s %s;' % '%s %s;' %
(style.SQL_KEYWORD('DROP TABLE'), (style.SQL_KEYWORD('DROP TABLE'),
style.SQL_TABLE(qn(f.m2m_db_table()))), style.SQL_TABLE(qn(f.m2m_db_table()))),
info.connection)) db.connection))
# Reverse it, to deal with table dependencies. # Reverse it, to deal with table dependencies.
output.reverse() output.reverse()
return output return output
def get_initialdata(self, model): def get_initialdata(self, model):
opts = model._meta opts = model._meta
info = opts.connection_info db = model._default_manager.db
settings = info.connection.settings settings = db.connection.settings
backend = info.backend backend = db.backend
app_dir = self.get_initialdata_path(model) app_dir = self.get_initialdata_path(model)
output = [] output = []
@ -295,12 +295,12 @@ class SchemaBuilder(object):
if os.path.exists(sql_file): if os.path.exists(sql_file):
fp = open(sql_file) fp = open(sql_file)
if backend.supports_compound_statements: if backend.supports_compound_statements:
output.append(BoundStatement(fp.read(), info.connection)) output.append(BoundStatement(fp.read(), db.connection))
else: else:
for statement in statements.split(fp.read()): for statement in statements.split(fp.read()):
if statement.strip(): if statement.strip():
output.append(BoundStatement(statement + ";", output.append(BoundStatement(statement + ";",
info.connection)) db.connection))
fp.close() fp.close()
return output return output
@ -327,16 +327,15 @@ class SchemaBuilder(object):
self.references.setdefault(f.rel.to, []).append((klass, f)) self.references.setdefault(f.rel.to, []).append((klass, f))
return self.references return self.references
def get_table_list(self, connection_info): def get_table_list(self, db):
"""Get list of tables accessible via the connection described by """Get list of tables accessible via db.
connection_info.
""" """
if self.tables is not None: if self.tables is not None:
return self.tables return self.tables
cursor = info.connection.cursor() cursor = db.connection.cursor()
introspection = connection_info.get_introspection_module() introspection = db.get_introspection_module()
return introspection.get_table_list(cursor) return introspection.get_table_list(cursor)
def table_exists(self, connection_info, table): def table_exists(self, db, table):
tables = self.get_table_list(connection_info) tables = self.get_table_list(db)
return table in tables return table in tables