如何将块长度从上传方法传递到流媒体内容?

时间:2012-05-11 11:24:10

标签: python flask ftputil

在我的Flask项目中,我使用ftputil库。在其中一个应用程序部分中,我使用内容流式传输,如Flask documentation

中所述
@app.route('/section')
def section():
    def generate():
        ftp.upload(source, target, "b", callback)
        yield 'completed'
    return Response(generate())

示例中的函数generate使文件上传到FTP服务器,如ftputil documentation中所述。

callback(chunk)方法中使用的回调函数[upload]对每个上传的文件块执行。

是否有可能从回调输出len(chunk)到流?任何肮脏的黑客也非常受欢迎。

感谢您的帮助!

1 个答案:

答案 0 :(得分:1)

我假设ftp.upload()同步运行,这是有道理的。我没有测试下面的代码,所以它可能充满了错误,但这个想法应该有效。

import threading, Queue

@app.route('/section')
def section():
    q = Queue.Queue()
    def callback(chunk):
        q.put(len(chunk))
    t = threading.Thread(target=lambda: ftp.upload(source, target, "b", callback) or q.put(None))
    t.setDaemon(True)
    t.start()
    def generate():
        while 1:
            l = q.get()
            if l is None:
                return
            yield unicode(l) + '\n'
    return Response(generate())