Added update_file() function to sdk

This commit is contained in:
Frikky
2024-10-24 12:20:37 +02:00
parent 8c91af54fa
commit 7cd2aa5146
2 changed files with 21 additions and 0 deletions
+19
View File
@@ -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)
+2
View File
@@ -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)