Catch error when querying kernel with invalid token

qutech · Jun 23, 2020 · 9981fe2 · 9981fe2
1 parent b7ae604
commit 9981fe2
Showing 1 changed file with 10 additions and 6 deletions.
diff --git a/filter_functions/util.py b/filter_functions/util.py
@@ -149,12 +149,16 @@ def _get_notebook_name() -> str:
             response = requests.get(urljoin(ss['url'], 'api/sessions'),
                                     params={'token': ss.get('token', '')})
             for nn in json.loads(response.text):
-                if nn['kernel']['id'] == kernel_id:
-                    try:
-                        relative_path = nn['notebook']['path']
-                        return os.path.join(ss['notebook_dir'], relative_path)
-                    except KeyError:
-                        return ''
+                try:
+                    if nn['kernel']['id'] == kernel_id:
+                        try:
+                            relative_path = nn['notebook']['path']
+                            return os.path.join(ss['notebook_dir'],
+                                                relative_path)
+                        except KeyError:
+                            return ''
+                except TypeError:
+                    return ''
 
         return ''