Fix vscode-jupyter hanging by executing vscode commands locally

CHANGELOG.md

@@ -2,6 +2,10 @@
 
 ## Next Release
 
+## Bug Fixes
+
+- Resolve indefinite hanging bug when connecting to livy via vscode-jupyter
+
 ## 0.23.0
 
 ### Bug Fixes

autovizwidget/autovizwidget/tests/test_encodingwidget.py

@@ -16,12 +16,12 @@ def setup_function():
     global df, encoding, ipywidget_factory, change_hook
 
     records = [
-        {"buildingID": 0, "date": "6/1/13", "temp_diff": 12, "\u263A": True},
-        {"buildingID": 1, "date": "6/1/13", "temp_diff": 0, "\u263A": True},
-        {"buildingID": 2, "date": "6/1/14", "temp_diff": 11, "\u263A": True},
-        {"buildingID": 0, "date": "6/1/15", "temp_diff": 5, "\u263A": True},
-        {"buildingID": 1, "date": "6/1/16", "temp_diff": 19, "\u263A": True},
-        {"buildingID": 2, "date": "6/1/17", "temp_diff": 32, "\u263A": True},
+        {"buildingID": 0, "date": "6/1/13", "temp_diff": 12, "\u263a": True},
+        {"buildingID": 1, "date": "6/1/13", "temp_diff": 0, "\u263a": True},
+        {"buildingID": 2, "date": "6/1/14", "temp_diff": 11, "\u263a": True},
+        {"buildingID": 0, "date": "6/1/15", "temp_diff": 5, "\u263a": True},
+        {"buildingID": 1, "date": "6/1/16", "temp_diff": 19, "\u263a": True},
+        {"buildingID": 2, "date": "6/1/17", "temp_diff": 32, "\u263a": True},
     ]
     df = pd.DataFrame(records)
 

sparkmagic/sparkmagic/kernels/wrapperkernel/usercodeparser.py

@@ -42,5 +42,8 @@ def get_code_to_run(self, code):
             return code
         elif not code:
             return code
+        elif "_VSCODE_" in code:
+            # Avoids sending VSCode-jupyter bootstrapping logic to livy
+            return code
         else:
             return "%%spark\n{}".format(code)

sparkmagic/sparkmagic/tests/test_usercodeparser.py

@@ -121,3 +121,15 @@ def test_unicode_in_double_magics():
     assert "%%{}\nmy content è🐙\nmore content\n ".format(
         magic_name
     ) == parser.get_code_to_run(cell)
+
+
+def test_vscode_bootstrapping():
+    parser = UserCodeParser()
+    cell = """
+from IPython import get_ipython
+
+# Variable that prevent wrapping from happing more than_once
+__VSCODE_wrapped_run_cell = False
+"""
+
+    assert cell == parser.get_code_to_run(cell)