如何通过命令行传递参数以在 Python 中设置环境

Question

我正在使用 python 3.7 和操作系统库。我必须在 STAGE 和 TEST 环境中运行一系列测试。当前代码将环境设置为 STAGE

环境 = os.getenv('ENV', 'STAGE')

但我希望通过命令行传递参数来设置它。任何人？

Answer 1

如果是 UNIX shell 的命令行，您可以将 env 变量设置为命令的一部分：

$ ENV=STAGE pytest ./tests/

Answer 2

在“python myfile.py”之后，只需键入以空格分隔的参数，在代码中，您可以执行以下操作：

import sys

nameOfScript = sys.argv[0]
commandLineArgs = sys.argv[1:]
commandLineArgsAsStr = str(sys.argv)
numArgs = len(sys.argv)