From 5a2f20e489065b9fbd1725dc247bc64f35d65126 Mon Sep 17 00:00:00 2001
From: Lance Edgar <lance@edbob.org>
Date: Tue, 28 Mar 2017 00:54:46 -0500
Subject: [PATCH] Add basic table listing view, with rough estimate row counts

---
 tailbone/views/tables.py | 72 ++++++++++++++++++++++++++++++++++++++++
 1 file changed, 72 insertions(+)
 create mode 100644 tailbone/views/tables.py

diff --git a/tailbone/views/tables.py b/tailbone/views/tables.py
new file mode 100644
index 00000000..379df0b5
--- /dev/null
+++ b/tailbone/views/tables.py
@@ -0,0 +1,72 @@
+# -*- coding: utf-8; -*-
+################################################################################
+#
+#  Rattail -- Retail Software Framework
+#  Copyright © 2010-2017 Lance Edgar
+#
+#  This file is part of Rattail.
+#
+#  Rattail is free software: you can redistribute it and/or modify it under the
+#  terms of the GNU Affero General Public License as published by the Free
+#  Software Foundation, either version 3 of the License, or (at your option)
+#  any later version.
+#
+#  Rattail is distributed in the hope that it will be useful, but WITHOUT ANY
+#  WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
+#  FOR A PARTICULAR PURPOSE.  See the GNU Affero General Public License for
+#  more details.
+#
+#  You should have received a copy of the GNU Affero General Public License
+#  along with Rattail.  If not, see <http://www.gnu.org/licenses/>.
+#
+################################################################################
+"""
+Views with info about the underlying Rattail tables
+"""
+
+from __future__ import unicode_literals, absolute_import
+
+from tailbone.views import MasterView
+from tailbone.newgrids import Grid, GridColumn
+
+
+class TablesView(MasterView):
+    """
+    Master view for tables
+    """
+    normalized_model_name = 'table'
+    model_key = 'name'
+    model_title = "Table"
+    creatable = False
+    editable = False
+    deletable = False
+    viewable = False
+    grid_factory = Grid
+    filterable = False
+    pageable = False
+
+    def get_data(self, **kwargs):
+        """
+        Fetch existing table names and estimate row counts via PG SQL
+        """
+        sql = """
+        select schemaname, relname, n_live_tup
+        from pg_stat_user_tables
+        order by n_live_tup desc;
+        """
+        result = self.Session.execute(sql)
+        return [dict(name=row[1], row_count=row[2]) for row in result]
+
+    def configure_grid(self, g):
+        g.columns = [
+            GridColumn('name'),
+            GridColumn('row_count'),
+        ]
+
+        g.sorters['name'] = g.make_sorter('name', foldcase=True)
+        g.sorters['row_count'] = g.make_sorter('row_count')
+        g.default_sortkey = 'name'
+
+
+def includeme(config):
+    TablesView.defaults(config)