Add shell_context_processor to inject models to shell.

getredash · Dec 8, 2016 · 81fb139 · 81fb139
1 parent 1d18109
commit 81fb139
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/redash/cli/__init__.py b/redash/cli/__init__.py
@@ -13,6 +13,12 @@
 def create(group):
     app = current_app or create_app()
     group.app = app
+
+    @app.shell_context_processor
+    def shell_context():
+        from redash import models
+        return dict(models=models)
+
     return app