def run()

in api/pages/node/modify.py [0:0]

• 44 lines of code
• 19 McCabe index (conditional complexity)

def run(API, environ, indata, session):
    
    method = environ['REQUEST_METHOD']
    
    # Modifying a node?
    if method == "POST":
        # Super users only!
        if not session.user or session.user['userlevel'] != 'superuser':
            raise API.exception(403, "You need to be logged in as super user to perform this action")
        
        
        # Try finding the node in the registry
        nodeid = indata['id']
        node = None
        try:
            node = plugins.registry.node(session, nodeid)
        except:
            raise API.exception(404, "Could not find a node by this ID to modify.")
        
        # Save any changes we get
        if 'verified' in indata:
            node.verified = indata['verified']
        if 'enabled' in indata:
            node.enabled = indata['enabled']
        if 'hostname' in indata and indata['hostname']:
            node.hostname = indata['hostname']
        if 'pubkey' in indata and indata['pubkey']:
            # Verify that PEM works:
            try:
                plugins.crypto.loads(indata['pubkey'])
            except:
                raise API.exception(400, "Could not save changes: Bad PEM payload passed!")
            node.pubkey = indata['pubkey']
        if 'description' in indata and indata['description']:
            node.description = indata['description']
        if 'location' in indata and indata['location']:
            node.location = indata['location']
        
        # All done, save node and exit
        node.save()
        del node # just to be sure
        
        yield json.dumps({"okay": True, "message": "Changes saved"}, indent = 2)
        return
    
    # Deleting a node?
    if method == "DELETE":
        # Super users only!
        if not session.user or session.user['userlevel'] != 'superuser':
            raise API.exception(403, "You need to be logged in as super user to perform this action")
        
        # Try finding the node in the registry
        nodeid = indata['id']
        node = None
        try:
            node = plugins.registry.node(session, nodeid)
        except:
            raise API.exception(404, "Could not find a node by this ID to modify.")
        
        # Remove node, say cheese!
        node.remove()
        yield json.dumps({"okay": True, "message": "Node removed from registry"}, indent = 2)
        return
        
    # Finally, if we hit a method we don't know, balk!
    yield API.exception(400, "I don't know this request method!!")