Function iter_file

tn-cli/commands.py:433–447

(filepath, size=1024*1024)

def fileUpload(id, cmd, args):
    def iter_file(filepath, size=1024*1024):
        _, name = os.path.split(filepath)
        mimeType = mimetypes.guess_type(filepath)[0]
        with open(filepath, mode='rb') as fd:
            try:
                yield pb.FileUpReq(id=str(id), auth=pb.Auth(scheme='token', secret=tn_globals.AuthToken),
                                   topic="", meta=pb.FileMeta(name=name, mime_type=mimeType, size=0))
                while True:
                    chunk = fd.read(size)
                    if chunk:
                        yield pb.FileUpReq(content=chunk)
                    else:  # Finished.
                        break
            except Exception as ex:
                stdoutln("Failed to read '{0}':".format(cmd.filename), ex)

    try:
        response = pbx.NodeStub(tn_globals.Connection).LargeFileReceive(iter_file(cmd.filename))