Class SourceMiddleInfo

client/task/sourcemgr.py:409–441

source管理类值传递处理类 用于将主机ip，与资源路径进行传递

class SourceMiddleInfo(object):
    """
    source管理类值传递处理类
    用于将主机ip，与资源路径进行传递
    """

    # 设置一个资源管理的标注字段，传入param
    SOURCE_MID_INFO = "transfer_info"

    def __init__(self, params):
        self.params = params

    def is_marked(self):
        if self.SOURCE_MID_INFO in self.params.keys():
            return True
        return False

    def upload(self, path_dict):
        transfer_info = TransferMgr().upload_file(path_dict)
        if not transfer_info:
            raise SourceMgrError("upload middle data error。 上传跨机器执行所需数据失败，请检查数据通路！")
        self.params[self.SOURCE_MID_INFO] = transfer_info
        return self.params

    def download(self):
        transfer_info = self.params[self.SOURCE_MID_INFO]
        TransferMgr().download_file(transfer_info)
        # transfer_info 格式参考task/transfermgr的注释
        # 将下载的内容载入params中
        path_list = transfer_info.get("path_list", [])
        for path_item in path_list:
            self.params[path_item["params_key_name"]] = os.path.join(DATA_DIR, path_item["rel_path"])
        return True


if __name__ == "__main__":