Fixes add button

Resolves #31

superset/static/blotout.js

@@ -1,22 +1,48 @@
-function parentPost() {
+const parentPost = () => {
   parent.postMessage({
     type: "iframeUrl",
     data: window.location.href,
     handler: 'load'
   }, "*")
 }
 
-const modifyElements = () => {
-  // Add + button to source database and table pages
-  if (window.location.pathname === '/databaseview/list/' ||
-      window.location.pathname === '/tablemodelview/list/') {
-    document.querySelectorAll('header nav .navbar-right .btn-primary')[0].style.display = 'inline-block'
+let modifyTimer
+const modifyCheck = (func) => {
+  const result = func()
+  if (!result) {
+    modifyTimer = setInterval(func, 500)
+  } else {
+    clearInterval(modifyTimer)
   }
+}
 
-  // remove reset password button from user page
-  if (window.location.pathname === '/users/userinfo/' ||
-      window.location.pathname.startsWith('/users/show/')) {
-    document.querySelector('.panel-body .well .btn-primary').style.display = 'none'
+const modifyElements = () => {
+  switch (window.location.pathname) {
+    case '/databaseview/list/':
+    case '/tablemodelview/list/': {
+      // Add + button to source database and table pages
+      modifyCheck(() => {
+        const elements = document.querySelectorAll('header nav .navbar-right .btn-primary')
+        if (elements.length === 0) {
+          return false
+        }
+        elements[0].style.display = 'inline-block'
+        return true
+      })
+      break
+    }
+    case '/users/userinfo/':
+    case '/users/show/': {
+      // remove reset password button from user page
+      modifyCheck(() => {
+        const elements = document.querySelectorAll('.panel-body .well .btn-primary')
+        if (elements.length === 0) {
+          return false
+        }
+        elements[0].style.display = 'none'
+        return true
+     })
+    }
   }
 }
 

superset/templates/superset/base.html

@@ -23,12 +23,12 @@
   {{ super() }}
   <link rel="icon" type="image/png" href="/static/assets/images/favicon.png">
   {{ css_bundle("theme") }}
-  <link media="all" rel="stylesheet" href="/static/blotout.css" />
+  <link media="all" rel="stylesheet" href="/static/blotout.css?v=1" />
 {% endblock %}
 
 {% block head_js %}
   {{ super() }}
-  <script src="/static/blotout.js"></script>
+  <script src="/static/blotout.js?v=1"></script>
   {{ js_bundle("theme") }}
 {% endblock %}
 

superset/templates/superset/basic.html

@@ -52,11 +52,11 @@
         {{ css_bundle(entry) }}
       {% endif %}
 
-    <link media="all" rel="stylesheet" href="/static/blotout.css" />
+    <link media="all" rel="stylesheet" href="/static/blotout.css?v=1" />
     {% endblock %}
 
     {{ js_bundle("theme") }}
-    <script src="/static/blotout.js"></script>
+    <script src="/static/blotout.js?v=1"></script>
 
     <input
       type="hidden"