opensource/galaxy
3
0
Fork 0
mirror of https://github.com/galaxyproject/galaxy.git synced 2026-09-24 16:30:27 +08:00
Code Issues Packages Projects Releases Wiki Activity

Use a SELECT statement to reliably retrieve table column names.

Browse Source
This commit is contained in:
James Johnson
2014-12-12 19:16:28 +00:00
parent 3afb6cb024
commit 64500b00aa
1 changed files with 12 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
lib/galaxy/datatypes/binary.py
+12 -3
View File
@@ -571,10 +571,19 @@ class SQlite ( Binary ):
rslt = c.execute(tables_query).fetchall()
for table,sql in rslt:
tables.append(table)
columns[table] = re.sub('^.*\((.*)\)$','\\1',sql).split(',')
try:
col_query = 'SELECT * FROM %s LIMIT 0' % table
cur = conn.cursor().execute(col_query)
cols = [col[0] for col in cur.description]
columns[table] = cols
except Exception, exc:
log.warn( '%s, set_meta Exception: %s', self, exc )
for table in tables:
row_query = "SELECT count(*) FROM %s" % table
rowcounts[table] = c.execute(row_query).fetchone()[0]
try:
row_query = "SELECT count(*) FROM %s" % table
rowcounts[table] = c.execute(row_query).fetchone()[0]
except Exception, exc:
log.warn( '%s, set_meta Exception: %s', self, exc )
dataset.metadata.tables = tables
dataset.metadata.table_columns = columns
dataset.metadata.table_row_count = rowcounts