检测 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": []
}

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