Python 下载网页 Jpg 图片以及 Gif 图片实现

#!/usr/bin/python3

import os
import re
import urllib.request
import time
import random

ISOTIMEFORMAT="%Y%m%d%H%M%S"

def getHtml(address):
""" Get the html source """
response = urllib.request.urlopen(address)
page = response.read()
pageStr = str(page)
return pageStr

def getFile(page,filetype="jpg"):
"""" Store the File to device """
reg = re.compile(r"http[s]?://[\S]*\." + filetype)
imagesUrl = re.findall(reg,page)
imagesUrl.sort()
i = 1
print("Start to download " + filetype + " file")
for index in range(len(imagesUrl)):
filename = filetype.capitalize() + str(time.strftime(ISOTIMEFORMAT)) + "_" + str(random.randint(0,100000)) + "." + filetype
if index != 0:
if imagesUrl[index] != imagesUrl[index-1]:
print("Downloading " + filename + ":" + imagesUrl[index])
urllib.request.urlretrieve(imagesUrl[index],filename)
i+=1
else:
print("Downloading " + filename + ":" + imagesUrl[index])
urllib.request.urlretrieve(imagesUrl[index],filename)
i+=1
else:
print("Downloading " + filetype + " file end")

def getJpg(page):
getFile(page,"jpg")

def getGif(page):
getFile(page,"gif");

url = input("Please input the url:\n")

filetype = input("Please input the file type you want to download('jpg, gif, all'): \n If enter noting or else while download jpg and gif\n")

storeDir = input("Please input the dir name to store image:\n")

try:
if not os.path.isdir(storeDir):
os.mkdir(storeDir)
finally:
os.chdir(storeDir)
print("Store Dir: " + storeDir)

page = getHtml(url)

if filetype=="jpg":
getJpg(page)
elif filetype == "gif":
getGif(page)
else:
getJpg(page)
getGif(page)