Question

我有一个简单的实用程序脚本，用于下载给定URL的文件。它基本上只是Linux二进制文件“aria2c”的包装。

以下是名为getFile的脚本：

#!/usr/bin/python

#
# SCRIPT NAME: getFile
# PURPOSE: Download a file using up to 20 concurrent connections.
#

import os
import sys 
import re
import subprocess

try:
    fileToGet = sys.argv[1]

    if os.path.exists(fileToGet) and not os.path.exists(fileToGet+'.aria2'):
        print 'Skipping already-retrieved file: ' + fileToGet
    else:
        print 'Downloading file: ' + fileToGet
        subprocess.Popen(["aria2c-1.8.0", "-s", "20", str(fileToGet), "--check-certificate=false"]).wait() # SSL

except IndexError:
    print 'You must enter a URI.'

因此，例如，此命令将下载文件：

$ getFile http://upload.wikimedia.org/wikipedia/commons/8/8e/Self-portrait_with_Felt_Hat_by_Vincent_van_Gogh.jpg

我想要做的是允许一个可选的第二个参数（在URI之后）是一个带引号的字符串。该字符串将是下载文件的新文件名。因此，在下载完成后，将根据第二个参数重命名该文件。使用上面的例子，我希望能够输入：

$ getFile http://upload.wikimedia.org/wikipedia/commons/8/8e/Self-portrait_with_Felt_Hat_by_Vincent_van_Gogh.jpg "van-Gogh-painting.jpg"

但我不知道如何将带引号的字符串作为可选参数。我怎么能这样做？

Answer 1

只需测试sys.argv的长度;如果它超过2，你有一个额外的参数：

if len(sys.argv) > 2:
    filename = sys.argv[2]

Answer 2

如果你在它们之间提供空格，shell会将它作为第二个参数传递（通常）。

例如，这里是test.py：

import sys

for i in sys.argv:
    print(i)

结果如下：

$ python test.py url "folder_name"
test.py
url
folder_name

引号无关紧要，因为它是在shell中处理的，而不是python。要获得它，只需要sys.argv[2]。

希望这有帮助！

引用字符串作为Python脚本的可选参数

2 个答案: