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