autocompletion sends busy/idle messages; fixes autocomplete in Jupyter lab

Author: craigbarratt

custom_components/pyscript/eval.py

@@ -1172,7 +1172,8 @@ def completions(self, root):
         for name, value in sym_table.items():
             if name.lower().startswith(root):
                 if callable(value) or isinstance(value, EvalFunc):
-                    words.add(f"{name}(")
+                    # used to be f"{name}(", but Jupyter doesn't always do the right thing with that
+                    words.add(name)
                 else:
                     words.add(name)
         return words

custom_components/pyscript/jupyter_kernel.py

@@ -296,6 +296,11 @@ async def shell_handler(self, msg):
             }
             await self.send(self.iopub_socket, 'status', content, parent_header=msg['header'])
         elif msg['header']["msg_type"] == "complete_request":
+            content = {
+                'execution_state': "busy",
+            }
+            await self.send(self.iopub_socket, 'status', content, parent_header=msg['header'])
+
             code = msg["content"]["code"]
             posn = msg["content"]["cursor_pos"]
             match = self.completion_re.match(code[0:posn].lower())
@@ -317,6 +322,11 @@ async def shell_handler(self, msg):
                 "metadata": {},
             }
             await self.send(self.shell_socket, 'complete_reply', content, parent_header=msg['header'], identities=identities)
+
+            content = {
+                'execution_state': "idle",
+            }
+            await self.send(self.iopub_socket, 'status', content, parent_header=msg['header'])
         elif msg['header']["msg_type"] == "is_complete_request":
             code = msg['content']["code"]
             self.ast_ctx.parse(code)