Python获取远程文件大小的函数代码分享

2020-02-23 05:21:28

字体：大中小

来源：转载

供稿：网友

代码如下:
def getRemoteFileSize(url, proxy=None):
    """ 通过content-length头获取远程文件大小
        url - 目标文件URL
        proxy - 代理 """
    opener = urllib2.build_opener()
    if proxy:
        if url.lower().startswith('https://'):
            opener.add_handler(urllib2.ProxyHandler({'https' : proxy}))
        else:
            opener.add_handler(urllib2.ProxyHandler({'http' : proxy}))
    try:
        request = urllib2.Request(url)
        request.get_method = lambda: 'HEAD'
        response = opener.open(request)
        response.read()
    except Exception, e: # 远程文件不存在
        return 0
    else:
        fileSize = dict(response.headers).get('content-length', 0)
        return int(fileSize)

上一篇：python完成FizzBuzzWhizz问题(拉勾网面试题)示例

下一篇：python中的实例方法、静态方法、类方法、类变量和实例变量浅析