在 python 中下载文件并取消

Question

我正在尝试制作一个简单的函数来在 python 中下载文件代码是这样的

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

我的问题是，如果我想在下载过程中取消下载过程，我该如何处理？？？

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

该平台是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"