refactor code and show metadata of each table

Author: linychuo

app.py

@@ -18,8 +18,11 @@ def read_conf_file(fp):
     """
     Read configuration from file
     """
-    with open(fp) as f:
-        return f.readlines()
+    try:
+        with open(fp) as f:
+            return f.readlines()
+    except IOError as e:
+        return []
 
 
 def write_conf_file(fp, content):
@@ -48,7 +51,7 @@ def on_forward_main(self, evt):
         selected_file = evt.selected_file
         main_frame = MainFrame(self, selected_file)
         main_frame.Show()
-        main_frame.Maximize()
+        # main_frame.Maximize()
 
     def write_file_history(self, fh):
         fp = os.path.join(self.app_config_dir, 'fh')

db.py

@@ -0,0 +1,44 @@
+import sqlite3
+
+
+class TableMeta:
+
+    def __init__(self, idx, column_name, column_type, is_null, default_value, is_pk):
+        self.idx = idx + 1
+        self.column_name = column_name
+        self.column_type = column_type
+        self.is_null = is_null
+        self.allow_null = is_null == 1 and 'Y' or 'N'
+        self.default_value = default_value
+        self.is_pk = is_pk
+        self.pk_field = is_pk == 1 and 'Y' or 'N'
+
+
+class DBUtils:
+    def __init__(self, db_file):
+        self.db_file = db_file
+
+    def get_object_type(self):
+        result = []
+        with sqlite3.connect(self.db_file) as conn:
+            c = conn.cursor()
+            for row in c.execute("select type from sqlite_master group by type"):
+                result.append(row[0].capitalize())
+        return result
+
+    def get_object_list(self, object_type):
+        object_list = []
+        with sqlite3.connect(self.db_file) as conn:
+            c = conn.cursor()
+            for _row in c.execute("select * from sqlite_master where type = ?", (object_type,)):
+                object_list.append(_row[1])
+        return object_list
+
+    def get_meta_of_table(self, table_name):
+        meta_data = []
+        with sqlite3.connect(self.db_file) as conn:
+            cur = conn.cursor()
+            for row in cur.execute("PRAGMA table_info(%s)" % table_name):
+                print(row)
+                meta_data.append(TableMeta(row[0], row[1], row[2], row[3], row[4], row[5]))
+        return meta_data

events.py

@@ -1,3 +1,4 @@
 import wx.lib.newevent
 
 ForwardMainEvent, EVT_FORWARD_MAIN_EVENT = wx.lib.newevent.NewEvent()
+CreateTabEvent, EVT_CREATE_MAIN_EVENT = wx.lib.newevent.NewEvent()

main.py

@@ -2,18 +2,47 @@
 """
 Main frame
 """
+
 import wx
-import sqlite3
 from wx import aui
 
+from db import DBUtils
+from events import EVT_CREATE_MAIN_EVENT, CreateTabEvent
 from sql_editor import SQLEditor
 
 
+class TableMetaPanel(wx.Panel):
+    def __init__(self, parent, data_list):
+        wx.Panel.__init__(self, parent=parent)
+
+        meta_data_list_ctrl = wx.ListCtrl(self, style=wx.LC_REPORT)
+        meta_data_list_ctrl.InsertColumn(0, "")
+        meta_data_list_ctrl.InsertColumn(1, "PK")
+        meta_data_list_ctrl.InsertColumn(2, "Name")
+        meta_data_list_ctrl.InsertColumn(3, "Type")
+        meta_data_list_ctrl.InsertColumn(4, "Null?")
+        meta_data_list_ctrl.InsertColumn(5, "Default Value")
+
+        for item in data_list:
+            meta_data_list_ctrl.Append(
+                (item.idx, item.pk_field, item.column_name, item.column_type, item.allow_null, item.default_value))
+        meta_data_list_ctrl.SetColumnWidth(0, 20)
+        meta_data_list_ctrl.SetColumnWidth(1, 40)
+        meta_data_list_ctrl.SetColumnWidth(2, 150)
+        meta_data_list_ctrl.SetColumnWidth(5, 150)
+        root_sizer = wx.BoxSizer(wx.VERTICAL)
+        root_sizer.Add(meta_data_list_ctrl, 1, wx.EXPAND)
+
+        self.SetSizer(root_sizer)
+
+
 class SideBar(wx.Panel):
     def __init__(self, parent, db_fp):
         super(SideBar, self).__init__(
             parent, style=wx.TAB_TRAVERSAL | wx.CLIP_CHILDREN)
         self.db_fp = db_fp
+        self.db_utils = DBUtils(self.db_fp)
+        self.parent = parent
 
         self.tree = wx.TreeCtrl(self, -1, wx.Point(0, 0), wx.Size(160, 250),
                                 wx.TR_DEFAULT_STYLE | wx.NO_BORDER)
@@ -28,7 +57,8 @@ def __init__(self, parent, db_fp):
         self.tree.Bind(wx.EVT_TREE_ITEM_ACTIVATED, self.on_node_activated)
 
         root = self.tree.AddRoot(self.db_fp, 0)
-        self.tree.AppendItem(root, u'Tables', 0, data='TABLES')
+        for item in self.db_utils.get_object_type():
+            self.tree.AppendItem(root, item, 0, data=item.upper())
         self.tree.Expand(root)
 
         main_sizer = wx.BoxSizer(wx.VERTICAL)
@@ -37,16 +67,15 @@ def __init__(self, parent, db_fp):
 
     def on_node_activated(self, evt):
         selected_item_data = self.tree.GetItemData(evt.GetItem())
-        if 'TABLES' == selected_item_data:
-            with sqlite3.connect(self.db_fp) as conn:
-                for row in conn.execute(
-                        "select name from sqlite_master where type = 'table'"):
-                    self.tree.AppendItem(
-                        evt.GetItem(), row[0], 1, data='TABLE')
-                self.tree.Expand(evt.GetItem())
-        elif 'TABLE' == selected_item_data:
+        if selected_item_data in ['TABLE', 'INDEX', 'TRIGGER', 'VIEW']:
+            for name in self.db_utils.get_object_list(selected_item_data.lower()):
+                self.tree.AppendItem(evt.GetItem(), name, 1, data='EACH-%s' % selected_item_data)
+            self.tree.Expand(evt.GetItem())
+        elif 'EACH-TABLE' == selected_item_data:
             selected_item_text = self.tree.GetItemText(evt.GetItem())
-            # open tab page
+            table_metadata = self.db_utils.get_meta_of_table(selected_item_text)
+            wx.PostEvent(self.parent, CreateTabEvent(table_metadata=table_metadata,
+                                                     tab_name="Table %s" % selected_item_text))
 
 
 class MainFrame(wx.Frame):
@@ -65,13 +94,9 @@ def __init__(self, app, db_fp):
 
         self.main_panel = wx.Panel(self)
         self.left = SideBar(self.main_panel, self.db_fp)
-        self.right = aui.AuiNotebook(
-            self.main_panel, -1, style=wx.CLIP_CHILDREN)
+        self.right = aui.AuiNotebook(self.main_panel, -1, style=wx.CLIP_CHILDREN)
         self.right.AddPage(SQLEditor(self.right, self), "Welcome")
-        self.logger = wx.TextCtrl(
-            self.main_panel,
-            -1,
-            style=wx.TE_MULTILINE | wx.TE_READONLY | wx.HSCROLL)
+        self.logger = wx.TextCtrl(self.main_panel, style=wx.TE_MULTILINE | wx.TE_READONLY | wx.HSCROLL)
 
         self.mgr = aui.AuiManager()
         self.mgr.SetManagedWindow(self.main_panel)
@@ -80,18 +105,29 @@ def __init__(self, app, db_fp):
         self.mgr.AddPane(self.left,
                          aui.AuiPaneInfo().Left().Layer(2).BestSize(
                              (240, -1)).MinSize((240, -1)).Floatable(
-                                 self.allowAuiFloating).FloatingSize(
-                                     (240, 700)).Caption("DB")
+                             self.allowAuiFloating).FloatingSize(
+                             (240, 700)).Caption("DB")
                          .CloseButton(False).Name("Tree"))
         self.mgr.AddPane(self.logger,
                          aui.AuiPaneInfo().Bottom().BestSize(
                              (-1, 150)).MinSize((-1, 140)).Floatable(
-                                 self.allowAuiFloating).FloatingSize(
-                                     (500, 160)).Caption("Log Messages")
+                             self.allowAuiFloating).FloatingSize(
+                             (500, 160)).Caption("Log Messages")
                          .CloseButton(False).Name("LogWindow"))
         self.Center(wx.BOTH)
         self.mgr.Update()
         self.Bind(wx.EVT_CLOSE, self.on_exit)
+        self.main_panel.Bind(EVT_CREATE_MAIN_EVENT, self.on_create_new_tab)
+
+    def on_create_new_tab(self, evt):
+        has_created = False
+        for i in xrange(0, self.right.GetPageCount()):
+            if self.right.GetPageText(i) == evt.tab_name:
+                self.right.ChangeSelection(i)
+                has_created = True
+                break
+        if not has_created:
+            self.right.AddPage(TableMetaPanel(self.right, evt.table_metadata), caption=evt.tab_name, select=True)
 
     def create_menu_bar(self):
         fileMenu = wx.Menu()

requirements.txt

@@ -1,3 +1,2 @@
 wxPython
 pyinstaller
-yapf