在Python中从命令行输入变量

Question

我想做的事情很简单，但是我无法为此找到一个好的且不太复杂的解决方案。 基本上，我想定义一些全局变量，例如，将其用作文件夹名称

global folder = "C:\\TEMP\\" + foldername

我想要的是在运行脚本时将foldername值设置为输入，例如：

python myscript.py --folder somebeautifulfoldername

因此，在运行脚本时，该文件夹将变为C：\\ TEMP \\ somebeautifulfoldername

Answer 1

您可以为此使用内置的argparse模块，并结合从sys.argv获取命令行参数：

import argparse
import sys

if __name__ == '__main__':
    parser = argparse.ArgumentParser(description = 'My script')
    parser.add_argument('--folder', help = "Subfolder of C:\TEMP\ to manipulate")
    args = parser.parse_args(sys.argv[1:])
    folder = "C:\\TEMP\\"+args.folder
    print folder

在这里，我只添加了一个非常简单的参数以及一些基本的帮助字符串，但是您可以做很多事情，例如提供默认值，允许使用文件列表而不是单个文件，指定类型...。 有关更多详细信息和示例，请参见手册。

用法：

>python myscript.py --folder somebeautifulfoldername
C:\TEMP\somebeautifulfoldername

>python myscript.py --help
usage: tmp.py [-h] [--folder FOLDER]

My script

optional arguments:
  -h, --help       show this help message and exit
  --folder FOLDER  Subfolder of C:\TEMP\ to manipulate

Answer 2

您可以将参数传递给Python脚本，如下所示：

python test.py arg1 arg2 arg3

这就是你得到的

Argument List: ['test.py', 'arg1', 'arg2', 'arg3']

在您的情况下：

python myscript.py somebeautifulfoldername

folder = "C:\\\\TEMP\\\\" + sys.argv[1]

Answer 3

import sys

folder = "none"
if("--folder" in  sys.argv):
    folder = sys.argv[sys.argv.index("--folder") + 1]        
print folder

如果以所需方式运行它：

python myscript.py --folder "HELLOFOLDER"

它将给出：HELLOFOLDER

Answer 4

有许多选项可将命令行参数解析为Python脚本。 例如，有标准库的optparse和argparse 。

一个非常不错的第三方工具docopt ，通过将脚本用法直接描述为脚本文档，您可以非常轻松地编写上述逻辑：

"""My script

Usage:
  myscript.py --folder=<folder>

Options:
  -h --help             Show this screen.
  --version             Show version.
  --folder=<folder>     Choose folder.
"""

from docopt import docopt

if __name__ == '__main__':
    arguments = docopt(__doc__, version='myscript 1.0')
    folder = "C:\\TEMP\\" + arguments["--folder"]
    print(folder)

也就是说，您可能还希望研究用于生成临时文件的tempfile ，以使脚本更具跨平台性。 硬编码Windows专用路径很少是个好主意。