Commit 850774d6 authored by Cristiano Urban's avatar Cristiano Urban
Browse files

Added support for single node and list of nodes.

parent 032650eb
Loading
Loading
Loading
Loading
+9 −6
Original line number Diff line number Diff line
@@ -18,12 +18,15 @@ class RetrievePreprocessor(TaskExecutor):
        super(RetrievePreprocessor, self).__init__()
        
    def execute(self):
        nodeType = self.jobObj.jobInfo["transfer"]["protocols"][0]["param"][0]["value"]
        vospacePath = self.jobObj.jobInfo["transfer"]["target"].split("!vospace")[1]
        if nodeType == "single":                    
            self.nodeList.append(vospacePath)
        else:
            self.dbConn.connect()
        osPath = self.dbConn.getOSPath(vospacePath)
            self.nodeList = self.dbConn.getVOSpacePathList(vospacePath)
            self.dbConn.disconnect()
        self.nodeList.append(osPath)
        self.jobObj.jobInfo["transfer"]["nodeList"] = self.nodeList
        self.jobObj.jobInfo["nodeList"] = self.nodeList        

    def run(self):
        print("Starting retrieve preprocessor...")