检测 Python 正在 Visual Studio Code 中运行

Question

在某些情况下，如果在 Visual Studio Code 中运行，代码需要采取不同的行为。

有人知道检测 python 代码正在 Visual Studio Code 调试器中运行的最有效方法吗？

到目前为止，我能找到的最好方法是使用：

import sys
if 'debugpy' in sys.modules:
    print("Running in VS Code")

Answer 1

我认为解决这个问题最优雅的方法就是在 vscode 中设置一个运行任务，该任务运行带有额外命令行标志的 Python 脚本。

例如：

import argparse

parser = argparse.ArgumentParser()
parser.add_argument('--vscode', action='store_true')

args = parser.parse_args()

if args.vscode:
    print("vscode")
else:
    print("not vscode")

那么如果你调用脚本python myscript.py

'不是vscode'

如果你调用python myscript.py --vscode

'vscode'

然后你可以在 vscode 中添加一个运行任务：

{
    "label": "run",
    "command": "python", // or python3
    "group": {
        "kind": "test",
        "isDefault": true
    },
    "args": [
        "${file}",
        "--vscode"
    ],
    "presentation": {
        "echo": true,
        "panel": "shared",
        "focus": true
    },
    "problemMatcher": []
}

要运行您的代码，只需使用运行任务的快捷方式