Author: jacob
Date: 2009-04-02 00:02:39 -0500 (Thu, 02 Apr 2009)
New Revision: 10364
Modified:
django/trunk/django/db/backends/sqlite3/introspection.py
Log:
Fixed #9779: added support for relation detection to inspectdb under SQLite.
Thanks, gabbott and ramiro.
Modified: django/trunk/django/db/backends/sqlite3/introspection.py
===================================================================
--- django/trunk/django/db/backends/sqlite3/introspection.py 2009-04-02
04:34:42 UTC (rev 10363)
+++ django/trunk/django/db/backends/sqlite3/introspection.py 2009-04-02
05:02:39 UTC (rev 10364)
@@ -1,3 +1,4 @@
+import re
from django.db.backends import BaseDatabaseIntrospection
# This light wrapper "fakes" a dictionary interface, because some SQLite data
@@ -55,8 +56,54 @@
info['null_ok']) for info in self._table_info(cursor,
table_name)]
def get_relations(self, cursor, table_name):
- raise NotImplementedError
-
+ """
+ Returns a dictionary of {field_index: (field_index_other_table,
other_table)}
+ representing all relationships to the given table. Indexes are
0-based.
+ """
+
+ # Dictionary of relations to return
+ relations = {}
+
+ # Schema for this table
+ cursor.execute("SELECT sql FROM sqlite_master WHERE tbl_name = %s",
[table_name])
+ results = cursor.fetchone()[0].strip()
+ results = results.split('(', 1)[1]
+ results = results.rsplit(')', 1)[0]
+
+ # Walk through and look for references to other tables. SQLite doesn't
+ # really have enforced references, but since it echoes out the SQL used
+ # to create the table we can look for REFERENCES statements used there.
+ for field_index, field_desc in enumerate(results.split(',')):
+ field_desc = field_desc.strip()
+ if field_desc.startswith("UNIQUE"):
+ continue
+
+ m = re.search('references (.*) \(["|](.*)["|]\)', field_desc, re.I)
+ if not m:
+ continue
+
+ table, column = [s.strip('"') for s in m.groups()]
+
+ cursor.execute("SELECT sql FROM sqlite_master WHERE tbl_name =
%s", [table])
+ result = cursor.fetchone()
+ if not result:
+ continue
+ other_table_results = result[0].strip()
+ other_table_results = other_table_results.split('(', 1)[1]
+ other_table_results = other_table_results.rsplit(')', 1)[0]
+
+ for other_index, other_desc in
enumerate(other_table_results.split(',')):
+ other_desc = other_desc.strip()
+ if other_desc.startswith('UNIQUE'):
+ continue
+
+ name = other_desc.split(' ', 1)[0].strip('"')
+ if name == column:
+ relations[field_index] = (other_index, table)
+ break
+
+ return relations
+
def get_indexes(self, cursor, table_name):
"""
Returns a dictionary of fieldname -> infodict for the given table,
--~--~---------~--~----~------------~-------~--~----~
You received this message because you are subscribed to the Google Groups
"Django updates" group.
To post to this group, send email to [email protected]
To unsubscribe from this group, send email to
[email protected]
For more options, visit this group at
http://groups.google.com/group/django-updates?hl=en
-~----------~----~----~----~------~----~------~--~---
