diff --git a/jupyter_vscode_proxy/__init__.py b/jupyter_vscode_proxy/__init__.py index 2ae6fd4..eb5ae43 100644 --- a/jupyter_vscode_proxy/__init__.py +++ b/jupyter_vscode_proxy/__init__.py @@ -61,6 +61,7 @@ def _get_cmd(port: int) -> List[str]: def setup_vscode() -> Dict[str, Any]: executable = os.environ.get("CODE_EXECUTABLE", "code-server") + icon = "code-server.svg" if executable == "code-server" else "vscode.svg" return { "command": _get_cmd_factory(executable), "timeout": 300, @@ -68,7 +69,7 @@ def setup_vscode() -> Dict[str, Any]: "launcher_entry": { "title": "VS Code", "icon_path": os.path.join( - os.path.dirname(os.path.abspath(__file__)), "icons", "vscode.svg" + os.path.dirname(os.path.abspath(__file__)), "icons", icon ), }, } diff --git a/jupyter_vscode_proxy/icons/code-server.svg b/jupyter_vscode_proxy/icons/code-server.svg new file mode 100644 index 0000000..85b701d --- /dev/null +++ b/jupyter_vscode_proxy/icons/code-server.svg @@ -0,0 +1,9 @@ + + + + + + + + + \ No newline at end of file