#coding=utf-8

import os

import sys

import re

import urllib

URL_REG = re.compile(r'(http://[^///]+)', re.I)

IMG_REG = re.compile(r']*?src=([/'"])([^/1]*?)/1', re.I)

def download(dir, url):

'''下載網頁中的圖片

@dir 保存到本地的路徑

@url 網頁url

'''

global URL_REG, IMG_REG

m = URL_REG.match(url)

if not m:

print '[Error]Invalid URL: ', url

return

host = m.group(1)

if not os.path.isdir(dir):

os.mkdir(dir)

# 獲取html,提取圖片url

html = urllib.urlopen(url).read()

imgs = [item[1].lower() for item in IMG_REG.findall(html)]

f = lambda path: path if path.startswith('http://') else /

host + path if path.startswith('/') else url + '/' + path

imgs = list(set(map(f, imgs)))

print '[Info]Find %d images.' % len(imgs)

# 下載圖片

for idx, img in enumerate(imgs):

name = img.split('/')[-1]

path = os.path.join(dir, name)

try:

print '[Info]Download(%d): %s'% (idx + 1, img)

urllib.urlretrieve(img, path)

except:

print "[Error]Cant't download(%d): %s" % (idx + 1, img)

def main():

if len(sys.argv) != 3:

print 'Invalid argument count.'

return

dir, url = sys.argv[1:]

download(dir, url)

if __name__ == '__main__':

# download('D://Imgs', 'http://www.163.com')

main()

總結

以上是生活随笔為你收集整理的python如何绘制蜘蛛图_用Python写的图片蜘蛛人代码的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。