Example #1
0
    def run(self):
        ie = self.prepare_endpoint(InfraEndpoint(), "infra")
        req = self.setup_request(FindStorageRequest())
        req.storageId = self.storageid
        response = ie.findStorage(req)
        if response.storage is None:
            print("! Storage '%s' doesn't exist" % self.storageid)
            return

        req = self.setup_request(UpdateStorageRequest())

        storage = response.storage

        req.storage = storage
        storage._id = self.storageid
        storage.name = self.name

        if self.localpath is not None:
            localAccess = LocalStorageAccess()
            localAccess.path = self.localpath
            localAccess.type = 'local'
            storage.defaultLocalAccess = localAccess

        if self.remotepath is not None:

            if self.remoteaccesstype is None or self.host is None or self.user is None or self.password is None:
                raise Exception("All options for remote access need to be specified")

            if self.remoteaccesstype is not None and self.remoteaccesstype not in validStorageAccessTypes:
                raise Exception("Option --accesstype %s has to be supplied" % validStorageAccessTypes)

            remoteAccess = RemoteStorageAccess()
            remoteAccess.type = 'remote'
            remoteAccess.accessType = self.remoteaccesstype
            ac = AccessCredentials()
            ac.authType = AccessCredentials_AuthType.PASSWORD
            ac.hostname = self.host
            ac.username = self.user
            ac.password = self.password
            ac.path = self.remotepath
            ac.port = self.determine_default_port(self.remoteaccesstype) if self.port is None else self.port

            remoteAccess.accessCredentials = ac
            storage.defaultRemoteStorageAccess = remoteAccess

        resp = ie.updateStorage(req)

        if resp.status != EndpointResponseStatus.OK:
            print("! Error updating storage %s: %s" %(self.storageid, resp.message))

        print("Storage %s has been updated" %(self.storageid))
Example #2
0
    def run(self):

        ie = self.prepare_endpoint(InfraEndpoint(), "infra")
        req = self.setup_request(FindStorageRequest())
        req.storageId = self.storageid
        response = ie.findStorage(req)
        if response.storage is not None:
            print("! Storage '%s' already exists, remove it before adding again" % self.storageid)
            return

        if self.remoteaccesstype not in validStorageAccessTypes:
            raise Exception("Option --accesstype %s has to be supplied" % validStorageAccessTypes)


        req = self.setup_request(CreateStorageRequest())

        storage = Storage()
        req.storage = storage
        storage._id = self.storageid
        storage.name = self.name

        localAccess = LocalStorageAccess()
        localAccess.path = self.localpath
        localAccess.type = 'local'
        storage.defaultLocalAccess = localAccess

        remoteAccess = RemoteStorageAccess()
        remoteAccess.type = 'remote'
        remoteAccess.accessType = self.remoteaccesstype
        ac = AccessCredentials()
        ac.authType = AccessCredentials_AuthType.PASSWORD
        ac.hostname = self.host
        ac.username = self.user
        ac.password = self.password
        ac.path = self.remotepath
        ac.port = self.determine_default_port(self.remoteaccesstype) if self.port is None else self.port

        remoteAccess.accessCredentials = ac
        storage.defaultRemoteStorageAccess = remoteAccess

        resp = ie.createStorage(req)

        if resp.status != EndpointResponseStatus.OK:
            print("Error creating storage %s: %s" %(self.storageid, resp.message))

        print("Storage %s has been created" %(self.storageid))
Example #3
0
    def run(self):

        ie = self.prepare_endpoint(InfraEndpoint(), "infra")

        node = self.find_node(ie, self.hostname)

        if self.validate is not None and self.add is None and self.remove is None:

            return self._validate_storage_access(ie, node)

        elif self.add is not None:

            if self.type is None:
                print("! --type option must be specified for --add")

            req = self.setup_request(FindStorageRequest())
            req.storageId = self.storageid
            response = ie.findStorage(req)
            if response.status != EndpointResponseStatus.OK:
                print("! Error retrieving stroage '%s'" % self.storageid)
                return

            storage = response.storage
            if storage is None:
                print("! Storage '%s' doesn't exist" % self.storageid)
                return


            if self.type.lower() == 'local':

                sa = LocalStorageAccess()
                sa.type = 'local'
                sa.storageId = self.storageid

                if self.usedefault is not None:

                    if storage.defaultLocalAccess is not None:
                        sa.path = storage.defaultLocalAccess.path
                    else:
                        raise Exception("Default local access has not been specified in storage '%s'" % self.storageid )

                else:
                    if self.path is None:
                        raise Exception("Option --path has to be specified")

                    sa.path = self.path

                self.add_or_replace_storage_access(ie, node, sa)


            elif self.type.lower() == 'remote':

                sa = RemoteStorageAccess()
                sa.type = 'remote'
                sa.accessType = self.accesstype
                sa.storageId = self.storageid

                if self.usedefault is not None:


                    if storage.defaultRemoteStorageAccess is not None:
                        sa.accessType =  storage.defaultRemoteStorageAccess.accessType
                        sa.accessCredentials = storage.defaultRemoteStorageAccess.accessCredentials
                    else:
                        raise Exception("Default remote access has not been specified in storage '%s'" % self.storageid )

                else:
                    if self.accesstype not in validStorageAccessTypes:
                        raise Exception("Option --accesstype %s has to be supplied, or --usedefault" % validStorageAccessTypes)

                    if self.path is None:
                        raise Exception("Option --path has to be specified, or --usedefault")

                    ac = AccessCredentials()
                    if self.path is None or self.user is None or self.password is None:
                        raise Exception("Options --path, --user, --password must be specified for remote storage access")

                    ac.authType = AccessCredentials_AuthType.PASSWORD
                    ac.path = self.path
                    ac.username = self.user
                    ac.password = self.password
                    ac.hostname = self.host
                    ac.port = determine_default_port() if self.port is None else self.port
                    sa.accessCredentials = ac

                self.add_or_replace_storage_access(ie, node, sa)

        elif self.remove is not None:

            for item in node.storageAccess:
                if item.storageId != self.storageid:
                    newStorageAccess.append(item)
            node.storageAccess = newStorageAccess
            self.update_node(ie, node)
            print("Access to storage %s has been removed from node %s" % (self.storageid, self.hostname))