Ticket #3702: adodbintrospection.patch

introspection.py

@@ +1 @@
+from django.db.backends import util
+import adodbapi.adodbapi
+
 def get_table_list(cursor):
-    raise NotImplementedError
+    "Returns a list of table names in the current database."
+    cursor.execute("sp_tables @table_type=\"'TABLE'\"")
+    rows = cursor.fetchall()
+    return [row[2] for row in rows]
+    
 
 def get_table_description(cursor, table_name):
-    raise NotImplementedError
+    "Returns a description of the table, with the DB-API cursor.description interface."
+    cursor.execute("SELECT TOP 1 * FROM %s" % table_name)
+    return cursor.description
+    
 
+def _name_to_index(cursor, table_name):
+    """
+    Returns a dictionary of {field_name: field_index} for the given table.
+    Indexes are 0-based.
+    """
+    return dict([(d[0], i) for i, d in enumerate(get_table_description(cursor, table_name))])
+
 def get_relations(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.
+    """
+    my_field_dict = _name_to_index(cursor, table_name)
+    cursor.execute("sp_fkeys @fktable_name='%s'" % u"ForeginTable1")
+    their_field_dicts = {}
+    result = {}
+    for row in util.dictfetchall(cursor):
+        if row[u'FKTABLE_NAME'] != table_name:
+            continue
+        if not their_field_dicts.get(row[u'PKTABLE_NAME']):
+            their_field_dicts[row[u'PKTABLE_NAME']
+                ] = _name_to_index(cursor, row[u'PKTABLE_NAME'])
+        their_fields = their_field_dicts[row[u'PKTABLE_NAME']]
+        result[my_field_dict[row[u'FKCOLUMN_NAME']]] = (
+            their_fields[row[u'PKCOLUMN_NAME']],
+            row[u'PKTABLE_NAME'])
+    return result
+    
 
 def get_indexes(cursor, table_name):
-    raise NotImplementedError
+    """
+    Returns a dictionary of fieldname -> infodict for the given table,
+    where each infodict is in the format:
+        {'primary_key': boolean representing whether it's the primary key,
+         'unique': boolean representing whether it's a unique index}
+    """
+    cursor.execute("sp_helpindex @objname='%s'" % table_name)
+    indexes = {}
+    try:
+        rows = cursor.fetchall()
+    except adodbapi.adodbapi.Error, e:
+        if e.args == (None,):
+            rows = [] # this is a workaround because I don't understand
+                      # why this error comes up. Will report to adodbapi
+        else:
+            raise
+        
+    for row in rows:
+        flags = row[1]
+        flagdict = {'primary_key': 'primary key' in flags,
+                    'unique': 'unique' in flags}
+        indexes[row[2]] = flagdict
+    return indexes
+    
 
 DATA_TYPES_REVERSE = {}