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

在python中下载文件并取消

2 个答案: