X-Git-Url: https://git.jsancho.org/?p=datasette-pytables.git;a=blobdiff_plain;f=datasette_pytables%2F__init__.py;h=58fee10364d3fec5c83d5b011579555f8d492712;hp=7bbc7c14063b5a562cb080c72609ccdf062fc9b2;hb=439b459adfa2adadb613ad9ec3779d6a8ec060c4;hpb=b1fff4628f94a3f3eab70f5fcc1e5005679ef1d0

diff --git a/datasette_pytables/__init__.py b/datasette_pytables/__init__.py
index 7bbc7c1..58fee10 100644
--- a/datasette_pytables/__init__.py
+++ b/datasette_pytables/__init__.py
@@ -240,8 +240,11 @@ class Connection:
         elif sql == 'select sql from sqlite_master where name = :n and type="table"':
             try:
                 table = self.h5file.get_node(params['n'])
+                colnames = ['value']
+                if type(table) is tables.table.Table:
+                    colnames = table.colnames
                 row = Row()
-                row['sql'] = 'CREATE TABLE {} ()'.format(params['n'])
+                row['sql'] = 'CREATE TABLE {} ({})'.format(params['n'], ", ".join(colnames))
                 return [row]
             except:
                 return []