forked from hk029/Pickup
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathPicDownloader.py
34 lines (25 loc) · 1.01 KB
/
PicDownloader.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
#-*- coding:utf-8 -*-
import re
import requests
def dowmloadPic(html,keyword):
pic_url = re.findall('"objURL":"(.*?)",',html,re.S)
i = 0
print '找到关键词:'+keyword+'的图片,现在开始下载图片...'
for each in pic_url:
print '正在下载第'+str(i+1)+'张图片,图片地址:'+str(each)
try:
pic= requests.get(each, timeout=10)
except requests.exceptions.ConnectionError:
print '【错误】当前图片无法下载'
continue
string = 'pictures\\'+keyword+'_'+str(i) + '.jpg'
#resolve the problem of encode, make sure that chinese name could be store
fp = open(string.decode('utf-8').encode('cp936'),'wb')
fp.write(pic.content)
fp.close()
i += 1
if __name__ == '__main__':
word = raw_input("Input key word: ")
url = 'http://image.baidu.com/search/flip?tn=baiduimage&ie=utf-8&word='+word+'&ct=201326592&v=flip'
result = requests.get(url)
dowmloadPic(result.text,word)