您的位置:首页 > 理论基础 > 计算机网络

【Python】Python的urllib模、urllib2模块的网络下载文件

2015-12-02 14:40 591 查看
因为需要从一些下载一个页PDF文件。但是需要下载PDF有数百个文件,这是不可能用人工点击下载。只是Python有相关模块,所以写一个程序PDF文件下载,顺便熟悉Python的urllib模块和ulrllib2模块。

1、问题描写叙述

须要从http://www.cvpapers.com/cvpr2014.html上下载几百个论文的PDF文件,该网页例如以下图所看到的:



2、问题解决

通过结合Python的urllib模块和urllib2模块来实现自己主动下载。

代码例如以下:

test.py

#!/usr/bin/python
# -*- coding:utf-8 -*-

import urllib                                                     #导入urllib模块
import urllib2                                                    #导入urllib2模块
import re                                                         #导入正則表達式模块:re模块

def getPDFFromNet(inputURL):
req = urllib2.Request(inputURL)
f = urllib2.urlopen(req)                                  #打开网页
localDir = 'E:\downloadPDF\\'                             #下载PDF文件须要存储在本地的文件夹
urlList = []                                              #用来存储提取的PDF下载的url的列表
for eachLine in f:                                        #遍历网页的每一行
line = eachLine.strip()                           #去除行首位的空格。习惯性写法
if re.match('.*PDF.*', line):                     #去匹配含有“PDF”字符串的行。仅仅有这些行才有PDF下载地址
wordList = line.split('\"')               #以"为分界。将该行分开,这样就将url地址单独分开了
for word in wordList:                     #遍历每一个字符串
if re.match('.*\.pdf$', word):    #去匹配含有“.pdf”的字符串,仅仅有url中才有
urlList.append(word)      #将提取的url存入列表
for everyURL in urlList:                                  #遍历列表的每一项,即每一个PDF的url
wordItems = everyURL.split('/')                   #将url以/为界进行划分。为了提取该PDF文件名称
for item in wordItems:                            #遍历每一个字符串
if re.match('.*\.pdf$', item):            #查找PDF的文件名称
PDFName = item                    #查找到PDF文件名称
localPDF = localDir + PDFName                     #将本地存储文件夹和须要提取的PDF文件名称进行连接
try:
urllib.urlretrieve(everyURL, localPDF)    #依照url进行下载。并以其文件名称存储到本地文件夹
except Exception,e:
continue

getPDFFromNet('http://www.cvpapers.com/cvpr2014.html')

注意:

(1)第1、6、8、23行分别多谢了一个“\”来进行转义。

(2)第27行的urlretrieve函数有3个參数:第一个參数就是目标url;第二个參数是保存的文件绝对路径(含文件名称),该函数的返回值是一个tuple(filename,header),当中的filename就是第二个參数filename。

假设urlretrieve仅提供1个參数,返回值的filename就是产生的暂时文件名称,函数运行完成后该暂时文件会被删除參数。第3个參数是一个回调函数,当连接上server、以及对应的数据块传输完成的时候会触发该回调。当中回调函数名称可随意,可是參数必须为三个。一般直接使用reporthook(block_read,block_size,total_size)定义回调函数。block_size是每次读取的数据块的大小。block_read是每次读取的数据块个数,taotal_size是一一共读取的数据量,单位是byte。

能够使用reporthook函数来显示读取进度。

假设想显示读取进度。则能够讲第三个參数加上。将上述程序第27行改为例如以下:

urllib.urlretrieve(everyURL, localPDF, reporthook=reporthook)


而reporthook回调函数的代码例如以下:

def reporthook(block_read,block_size,total_size):
if not block_read:
print "connection opened";
return
if total_size<0:
#unknown size
print "read %d blocks (%dbytes)" %(block_read,block_read*block_size);
else:
amount_read=block_read*block_size;
print 'Read %d blocks,or %d/%d' %(block_read,block_read*block_size,total_size);


综上所述。这就是一个简单的从网页抓取数据、下载文件的小程序。希望对正在学习Python的同学有帮助。谢谢!

内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: