Added update_file() function to sdk
This commit is contained in:
@@ -1645,6 +1645,25 @@ class AppBase:
|
||||
#return value.json()
|
||||
return {"success": False}
|
||||
|
||||
def update_file(self, file_id, content):
|
||||
full_execution = self.full_execution
|
||||
workflow_id = full_execution["workflow"]["id"]
|
||||
org_id = full_execution["workflow"]["execution_org"]["id"]
|
||||
|
||||
new_headers = {
|
||||
"Authorization": f"Bearer {self.authorization}",
|
||||
"User-Agent": "Shuffle 1.1.0",
|
||||
}
|
||||
|
||||
upload_path = "/api/v1/files/%s/edit?execution_id=%s" % (file_id, full_execution["execution_id"])
|
||||
|
||||
files={"shuffle_file": ("filename", content)}
|
||||
|
||||
ret = requests.put("%s%s" % (self.url, upload_path), files=files, headers=new_headers, verify=False, proxies=self.proxy_config)
|
||||
print(ret.status_code)
|
||||
return ret.json()
|
||||
|
||||
|
||||
# Wrapper for set_files
|
||||
def set_file(self, infiles):
|
||||
return self.set_files(infiles)
|
||||
|
||||
@@ -3726,7 +3726,9 @@ func sendWorkerRequest(workflowExecution shuffle.ExecutionRequest, image string,
|
||||
|
||||
newresp, err := client.Do(req)
|
||||
if err != nil {
|
||||
// Connection refused?
|
||||
log.Printf("[ERROR] Error running worker request to %s (1): %s", streamUrl, err)
|
||||
|
||||
if strings.Contains(fmt.Sprintf("%s", err), "connection refused") || strings.Contains(fmt.Sprintf("%s", err), "EOF") {
|
||||
workerImage := fmt.Sprintf("%s/%s/shuffle-worker:%s", baseimageregistry, baseimagename, workerVersion)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user