将变量从命令行解析为 url

Question

我有一个名为dlimage 的python 脚本。我想像$ python dlimage.py 1 2 这样在终端中键入一个变量，并让1 and 2 对应于download_web_image 中的url 成为http://www.example.com/1/2.jpg 并下载图像。我该怎么做？

import urllib.request
import argparse

def download_web_image(url):
    urllib.request.urlretrieve(url)

parser = argparse.ArgumentParser()
parser.add_argument("num1", "num2")
args = parser.parse_args()

download_web_image("http://www.example.com/"num1"/"num2".jpg")

编辑 2： 我终于让它工作了。感谢大家的帮助！ 有效的代码：

import urllib
import argparse

def download_web_image(url):
    IMAGE = url.rsplit('/',1)[1]
    urllib.urlretrieve(url, IMAGE)

parser = argparse.ArgumentParser()
parser.add_argument("num1")
parser.add_argument("num2")
args = parser.parse_args()

download_web_image("https://www.example.com/{num1}/{num2}.jpg".format(num1=args.num1, num2=args.num2))

Answer 1

当我尝试你的部分代码时，我得到一个错误：

In [1663]: parser = argparse.ArgumentParser()
In [1664]: parser.add_argument("num1", "num2")
....
ValueError: invalid option string 'num1': must start with a character '-'

add_argument 方法的参数错误。

你应该使用的是：

parser = argparse.ArgumentParser()
parser.add_argument("num1")
parser.add_argument("num2")

在这种情况下，help 将如下所示：

In [1668]: parser.print_help()
usage: ipython3 [-h] num1 num2

positional arguments:
  num1
  num2

optional arguments:
  -h, --help  show this help message and exit

并测试等效于myprog 1 2的输入

In [1669]: args = parser.parse_args(['1','2'])
In [1670]: args
Out[1670]: Namespace(num1='1', num2='2')
In [1671]: args.num1
Out[1671]: '1'
In [1672]: args.num2
Out[1672]: '2'

现在我可以使用以下格式设置 URL：

In [1675]: "https://www.example.com/{}/{}.jpg".format(args.num1, args.num2)
Out[1675]: 'https://www.example.com/1/2.jpg'

所以你的代码有两个问题：

每个参数 num1 和 num2 必须在单独的 add_argument 语句中定义。阅读文档以了解您还可以在该语句中添加什么内容，例如 help。您试图在一个语句中定义 2 个参数，但出现错误。

其次，您需要使用正确的格式。我添加了{}（{0} 和{num1} 样式也可以）。或者在旧的 Py2 样式中：

"https://www.example.com/%s/%s.jpg"%(args.num1, args.num2)

Answer 2

改一下

parser.add_argument("num1, "num2")

到

parser.add_argument("num1", "num2")

Answer 3

为此我会使用格式：

download_web_image("http://www.example.com/{num1}/{num2}.jpg".format(num1=args.num1, num2 = args.num2))

这是一个例子：

num1 = 5
num2 = 6
"http://www.example.com/{num1}/{num2}.jpg".format(num1 = num1, num2 = num2)

output:
'http://www.example.com/5/6.jpg'

format 可以很容易地将定义的参数插入到字符串中。

Answer 4

假设您使用的是 python3，我尝试做一些简单的事情，并且尽可能接近您最初尝试的方法。希望这会有所帮助。

import urllib.request
import sys
def download_web_image(url):
    urllib.request.urlretrieve(url)

download_web_image("http://www.example.com/{0}/{1}.jpg".format(sys.argv[1], sys.argv[2]))

Answer 5

我想你想传递 N 个参数以便生成一个 URL，我会这样做：

import sys
args = sys.argv[1:]
baseURL = "http://www.example.com"
url = baseURL + "/" + "/".join(args)
print url

输入：

$ python dlimage.py 1 2 3.jpg

输出：

http://www.example.com/1/2/3.jpg

至于下载，你可以试试this answer，或者在url中读取数据并将其写入一个名为最后一个参数的文件：

import urllib2
urlResponse = urllib2.urlopen(url)
urlData = urlResponse.read()

outFile = open(args[-1], 'w')
outFile.write(urlData)
outFile.close()

我使用工作代理，下载文件时出现错误，可能会在今天晚些时候在家更新我的答案。