def saveImage(user, data, delete):
imageID = getID(data["imageImagesSelect"])
name = data["imageNameOnImageForm"]
if imageID == "0":
return {"updates": {}, "message": "deleted image"}
authAccount = data["authAccount"]
rootdir = data["installDirectory"]
if imageID == None:
image = Image(name, authAccount, rootdir, user)
SESSION.add(image)
SESSION.commit()
elif delete:
Image.delete(SESSION, imageID, user)
return {"updates": {}, "message": "deleted image"}
else:
image = Image.findByID(SESSION, imageID, user)
if image != None:
image.updateValues(name, authAccount, rootdir, user)
SESSION.add(image)
SESSION.commit()
else:
return {"updates": {}, "message": "user permissions error on image"}
# image = Image(name, authAccount, user)
result = {"updates": {"images": {str(image.id): image.dictForJSON()}}, "message": "updated image"}
return result
def saveDisk(user, data, variables, delete):
if delete:
diskID = getID(data["diskDisksSelect"])
workflow = WorkflowTemplate.findByID(SESSION, getID(data["diskWorkflowsSelect"]), user)
if workflow == None:
return {"updates": {}, "message": "user permissions error on workflow"}
if diskID != None:
DiskTemplate.delete(SESSION, diskID, user)
elif variables:
diskID = getID(data["diskVarDisksSelect"])
workflow = WorkflowTemplate.findByID(SESSION, getID(data["diskVarWorkflowsSelect"]), user)
if workflow == None:
return {"updates": {}, "message": "user permissions error on workflow"}
disk = DiskTemplate.findByID(SESSION, diskID, user)
if disk == None:
return {"updates": {}, "message": "user permissions error on disk"}
disk.updateVarDict(parseVariables(data), user)
else:
diskID = getID(data["diskDisksSelect"])
workflow = WorkflowTemplate.findByID(SESSION, getID(data["diskWorkflowsSelect"]), user)
if workflow == None:
return {"updates": {}, "message": "user permissions error on workflow"}
name = data["diskName"]
if getID(data["diskImagesSelect"]) == "0":
image = None
else:
image = Image.findByID(SESSION, getID(data["diskImagesSelect"]), user)
if image == None:
return {"updates": {}, "message": "user permissions error on image"}
diskSize = data["diskSize"]
diskType = data["diskTypeSelector"]
location = data["diskLocationSelector"]
if VERBOSE:
print "diskID", diskID
if diskID == None:
disk = DiskTemplate(name.lower(), workflow, image, diskSize, diskType, location)
else:
disk = DiskTemplate.findByID(SESSION, diskID, user)
if disk != None:
disk.updateValues(name.lower(), workflow, image, diskSize, diskType, location, user)
else:
return {"updates": {}, "message": "user permissions error on disk"}
# disk = DiskTemplate(name, workflow, image, diskSize, diskType, location)
SESSION.add(workflow)
SESSION.commit()
result = {"updates": {"workflows": {str(workflow.id): workflow.dictForJSON()}}, "message": "updated disk"}
return result
def getImage(self):
user = self.getUser()
image = Image.findByName(SESSION, "test_image", user)
if image==None:
image = Image("test_image", "*****@*****.**", "/home/cmelton/", user)
return image
