时间戳

首页 > 解决方案 > 如何利用 python 闭包编写回调函数

python - 如何利用 python 闭包编写回调函数

问题描述

我在亚马逊的手册页上找到了这段代码。它用于创建回调对象，可以作为传输模块的上传函数的参数传递：

class ProgressPercentage(object):

def __init__(self, filename):
    self._filename = filename
    self._size = float(os.path.getsize(filename))
    self._seen_so_far = 0
    self._lock = threading.Lock()

def __call__(self, bytes_amount):
    # To simplify we'll assume this is hooked up
    # to a single filename.
    with self._lock:
        self._seen_so_far += bytes_amount
        percentage = (self._seen_so_far / self._size) * 100
        sys.stdout.write(
            "\r%s  %s / %s  (%.2f%%)" % (
                self._filename, self._seen_so_far, self._size,
                percentage))
        sys.stdout.flush()

一个类用于保留后续调用所需的状态（self.seen_so_far ie）

有没有办法将其重新实现为利用 python 闭包确保有状态的函数？

标签： pythoncallbackclosures

解决方案

当然，一个简单的音译应该是这样的：

def ProgressPrecentage(filename):
    size = os.path.getsize(filename)
    seen_so_far = 0
    lock = threading.Lock()
    def worker(bytes_amount):
        nonlocal seen_so_far
        with lock:
            seen_so_far += bytes_amount
            percentage = (seen_so_far / size)*100
            msg = "\r%s  %s / %s  (%.2f%%)" % (
                filename, seen_so_far, size, percentage
            )
            sys.stdout.write(msg)
            sys.stdout.flush()
    return worker

推荐阅读