Python：在回调函数中定义变量……不确定在哪里

Question

摘抄：

file = open("D:\\DownloadFolder\\test.mp3", "wb")

def callback(data):
    file.write(data)
    sizeWritten += len(data)
    print(sizeWritten)

connect.retrbinary('RETR test.mp3', callback)
print("completed")

Python显然抱怨我没有定义sizeWritten ，但是我不确定应该在哪里定义它。 如果我在函数前放置sizeWritten = 0 ，它仍然会给出一个错误的local variable 'sizeWritten referenced before assignment 。 我应该怎么做？

Answer 1

如果sizeWritten可以是全局的（例如，一次只能有一个回调活动），则可以在函数中将其标记为：

file = open("D:\\DownloadFolder\\test.mp3", "wb")
sizeWritten = 0

def callback(data):
    global sizeWritten
    file.write(data)
    sizeWritten += len(data)
    print(sizeWritten)

callback对该名称的任何分配都会更改全局名称。

在Python 3中，您还可以使用闭包和nonlocal关键字：

def download(remote, local):
    file = open(local, "wb")
    sizeWritten = 0

    def callback(data):
        nonlocal sizeWritten
        file.write(data)
        sizeWritten += len(data)
        print(sizeWritten)

    connect.retrbinary('RETR ' + remote, callback)
    print("completed")

至少将sizeWritten和file对象封装在本地名称空间中。

但是，您可以直接从打开file file对象获得相同的信息：

def callback(data):
    file.write(data)
    print(file.tell())