python - 在python中下载文件并取消

Question

我正在尝试制作一个简单的函数来在 python 中下载文件代码类似于

def download(url , dest):
    urllib.urlretrieve(url, dest)

我的问题是，如果我想在下载过程中取消下载过程，我该怎么办？？？

该功能在app后台运行，由按钮触发。现在我试图用另一个按钮来触发它。

该平台是XBMC。

Answer 1

download一个简单的类与您的函数执行相同的操作：

import urllib
import threading

class Downloader:

    def __init__(self):
        self.stop_down = False
        self.thread = None

    def download(self, url, destination):
        self.thread = threading.Thread(target=self.__down, args=(url, destination))
        self.thread.start()

    def __down(self, url, dest):
        _continue = True
        handler = urllib.urlopen(url)
        self.fp = open(dest, "w")
        while not self.stop_down and _continue:
            data = handler.read(4096)
            self.fp.write(data)
            _continue = data
        handler.close()
        self.fp.close()

    def cancel(self):
        self.stop_down = True

因此，当有人单击“取消”按钮时，您必须调用该cancel()方法。

请注意，如果您取消它，这不会删除部分下载的文件，但使用os.unlink().

以下示例脚本显示了如何使用它，开始下载一个 ~20Mb 文件并在 5 秒后取消它：

import time
if __name__ == "__main__":
    url = "http://ftp.postgresql.org/pub/source/v9.2.3/postgresql-9.2.3.tar.gz"
    down = Downloader()
    down.download(url, "file")
    print "Download started..."
    time.sleep(5)
    down.cancel()
    print "Download canceled"

Answer 2

如果您通过按 CTRL+C 取消，那么您可以使用此内置异常并继续您认为最好的操作。

在这种情况下，如果我在下载过程中取消，我只想删除该部分文件：

def download(url , dest):
    try:
        urllib.urlretrieve(url, dest)
    except KeyboardInterrupt:
        if os.path.exists(dest):
            os.remove(dest)
    except Exception, e:
        raise