Bläddra i källkod

[notebook] Have new snippets ask the assist for the selected database

Johan Ahlen 10 år sedan
förälder
incheckning
fd29185

+ 13 - 0
desktop/core/src/desktop/templates/assist.mako

@@ -342,6 +342,19 @@ from desktop.views import _ko
           self.selectedSource(sourceIndex[databaseDef.source]);
           setDatabaseWhenLoaded(databaseDef.name);
         });
+
+        huePubSub.subscribe("assist.get.database", function (source) {
+          if (sourceIndex[source] && sourceIndex[source].selectedDatabase()) {
+            huePubSub.publish("assist.database.set", {
+              source: source,
+              name: sourceIndex[source].selectedDatabase().databaseName
+            });
+          } else {
+            huePubSub.publish("assist.database.set", {
+              source: source,
+              name: null
+            });
+          }
         });
 
         huePubSub.publish("assist.ready");

+ 4 - 0
desktop/libs/notebook/src/notebook/static/notebook/js/notebook.ko.js

@@ -180,6 +180,10 @@
       }
     });
 
+    if (! self.database()) {
+      huePubSub.publish("assist.get.database", self.type());
+    }
+
     self.statement_raw = ko.observable(typeof snippet.statement_raw != "undefined" && snippet.statement_raw != null ? snippet.statement_raw : '');
     self.selectedStatement = ko.observable('');
     self.codemirrorSize = ko.observable(typeof snippet.codemirrorSize != "undefined" && snippet.codemirrorSize != null ? snippet.codemirrorSize : 100);