From: Javier Sancho <jsf@jsancho.org>
Date: Wed, 30 May 2018 10:13:30 +0000 (+0200)
Subject: Show field names in create sql
X-Git-Url: https://git.jsancho.org/?p=datasette-pytables.git;a=commitdiff_plain;h=d3ebe73db6b25569b800ccd55ad8eb90dd88b773;ds=sidebyside

Show field names in create sql
---

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 []