Use connector registry to fetch the table column class. (#2419)

apache · Mar 16, 2017 · 20aec3c · 20aec3c
1 parent 4ded37e
commit 20aec3c
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/superset/views/core.py b/superset/views/core.py
@@ -1792,7 +1792,9 @@ def sqllab_viz(self):
         metrics = []
         for column_name, config in data.get('columns').items():
             is_dim = config.get('is_dim', False)
-            col = models.TableColumn(
+            SqlaTable = ConnectorRegistry.sources['table']
+            TableColumn = SqlaTable.column_cls
+            col = TableColumn(
                 column_name=column_name,
                 filterable=is_dim,
                 groupby=is_dim,