1 files changed, 7 insertions, 4 deletions
diff --git a/megapixels/app/models/sql_factory.py b/megapixels/app/models/sql_factory.py
index 2a18d6af..e35c3e15 100644
--- a/megapixels/app/models/sql_factory.py
+++ b/megapixels/app/models/sql_factory.py
@@ -21,10 +21,7 @@ datasets = {}
 loaded = False
 
 def list_datasets():
-  return [{
-    'name': name,
-    'tables': list(datasets[name].tables.keys()),
-  } for name in datasets.keys()]
+  return [dataset.describe() for dataset in datasets.values()]
 
 def get_dataset(name):
   return datasets[name] if name in datasets else None
@@ -76,6 +73,12 @@ class SqlDataset:
       base_model = declarative_base(engine)
     self.base_model = base_model
 
+  def describe(self):
+    return {
+      'name': self.name,
+      'tables': list(self.tables.keys()),
+    }
+
   def get_table(self, type):
     if type in self.tables:
       return self.tables[type]