Python 如何判断urllib.urlretrieve是否成功

在本文中，我们将介绍如何使用Python中的urllib库中的urlretrieve函数来判断下载文件是否成功。

阅读更多：Python 教程

urllib.urlretrieve函数简介

urllib.urlretrieve函数是Python标准库urllib中的一个函数，用于从指定的URL下载文件到本地。该函数的语法如下：

urllib.urlretrieve(url, filename=None, reporthook=None)

其中，url是要下载文件的URL地址，filename是要保存文件的路径（包括文件名），reporthook是一个可选的回调函数，用于显示下载进度。

判断urlretrieve是否成功下载文件的方法

使用urllib.urlretrieve下载文件后，可以通过以下方法来判断是否成功下载文件：

1. 返回值判断:
   urlretrieve函数的返回值是一个元组，包含两个元素，第一个元素是已经下载的文件名，第二个元素是包含一些关于下载的信息。因此，可以通过判断返回值的类型或者第二个元素的内容来确定是否成功下载文件。

   示例代码如下：

   import urllib
   
   try:
      result = urllib.urlretrieve(url, filename)
      if result[1].getcode() == 200:
          print("文件下载成功!")
      else:
          print("文件下载失败!")
   except Exception as e:
      print("发生异常：", str(e))
   

   Python

   Copy
2. 异常处理:
   在使用urllib.urlretrieve下载文件时，如果遇到网络断开、文件不存在等异常情况，都会抛出异常。因此，我们可以通过捕获异常来判断是否成功下载文件。

   示例代码如下：

   import urllib.error
   import urllib.request
   
   try:
      urllib.request.urlretrieve(url, filename)
      print("文件下载成功!")
   except urllib.error.HTTPError as e:
      print("HTTPError:", str(e))
   except urllib.error.URLError as e:
      print("URLError:", str(e))
   except Exception as e:
      print("发生异常：", str(e))
   

   Python

   Copy

   在上述代码中，我们通过捕获urllib.error模块中的HTTPError和URLError异常来判断是否成功下载文件，并打印出相应的错误信息。

3. 文件存在判断:
   在某些场景下，我们可能需要判断下载文件是否已经成功保存到本地。我们可以通过判断文件的存在与否来确定是否成功下载文件。

   示例代码如下：

   import os
   import urllib.request
   
   urllib.request.urlretrieve(url, filename)
   if os.path.exists(filename):
      print("文件下载成功!")
   else:
      print("文件下载失败!")
   

   Python

   Copy

   在上述代码中，我们使用os.path.exists函数来判断文件是否存在，进而判断文件是否成功下载。

总结

本文介绍了如何使用Python中的urllib库中的urlretrieve函数来判断文件是否成功下载。通过返回值判断、异常处理和文件存在判断三种方法，我们可以方便地判断是否成功下载文件。使用合适的方法，可以更好地处理文件下载中的异常情况，提高程序的健壮性。

注意，在实际使用过程中，还需要考虑网络环境、文件大小等因素对文件下载的影响，并做相应的处理。