confluent/plugins/configuration/attributes.py

import confluent.exceptions as exc
import confluent.messages as msg
import confluent.config.attributes as allattributes

def retrieve(nodes, element, configmanager, inputdata):
    attributes = configmanager.get_node_attributes(nodes)
    if element[-1] == 'all':
        for node in nodes:
            for attribute in sorted(allattributes.node.iterkeys()):
                if attribute in attributes[node]: #have a setting for it
                    val = attributes[node][attribute]
                elif attribute == 'groups': # no setting, provide a blank
                    val = []
                else: # no setting, provide a blank
                    val = {'value': None}
                if attribute.startswith('secret.'):
                    yield msg.CryptedAttributes(node,
                        {attribute: val},
                        allattributes.node[attribute]['description'])
                elif isinstance(val, list):
                    yield msg.ListAttributes(node,
                        {attribute: val},
                        allattributes.node[attribute]['description'])
                else:
                    yield msg.Attributes(node,
                        {attribute: val['value']},
                        allattributes.node[attribute]['description'])
    elif element[-1] == 'current':
        for node in attributes.iterkeys():
            for attribute in sorted(attributes[node].iterkeys()):
                currattr = attributes[node][attribute]
                if 'value' in currattr:
                    yield msg.Attributes(node,
                        {attribute: currattr['value']},
                        allattributes.node[attribute]['description'])
                elif 'cryptvalue' in currattr:
                    yield msg.CryptedAttributes(node,
                        {attribute: currattr},
                        allattributes.node[attribute]['description'])
                elif isinstance(currattr, list):
                    yield msg.ListAttributes(node,
                        {attribute: currattr},
                        allattributes.node[attribute]['description'])
                else:
                    print repr(currattr)
                    raise Exception("BUGGY ATTRIBUTE FOR NODE")


def update(nodes, element, configmanager, inputdata):
    updatedict = {}
    for node in nodes:
        updatenode = inputdata.get_attributes(node)
        if updatenode:
            updatedict[node] = updatenode
    try:
        configmanager.set_node_attributes(updatedict)
    except ValueError:
        raise exc.InvalidArgumentException()
    return retrieve(nodes, element, configmanager, inputdata)
Have configmanager protect against unrecognized data. When doing a set, check the keynames and value types as appropriate. raise ValueError in the configmanager case since it could be code or human mistake. attributes plugin then catches this error and propogates it up as an InvalidArgumentException if it is just trying to pass in data from user. 2014-02-13 12:58:03 -05:00			`import confluent.exceptions as exc`
Add attribute retrieval facility 2013-11-02 18:24:04 -04:00			`import confluent.messages as msg`
Add support for showing all as well as current attributes 2013-11-03 10:36:03 -05:00			`import confluent.config.attributes as allattributes`
Add attribute retrieval facility 2013-11-02 18:24:04 -04:00
Wire up input to plugins for create/read 2013-11-02 19:45:10 -04:00			`def retrieve(nodes, element, configmanager, inputdata):`
Add attribute retrieval facility 2013-11-02 18:24:04 -04:00			`attributes = configmanager.get_node_attributes(nodes)`
Have rest explorer actually mostly work as expected 2013-11-07 14:39:34 -05:00			`if element[-1] == 'all':`
Add support for showing all as well as current attributes 2013-11-03 10:36:03 -05:00			`for node in nodes:`
			`for attribute in sorted(allattributes.node.iterkeys()):`
			`if attribute in attributes[node]: #have a setting for it`
			`val = attributes[node][attribute]`
Enumerate nodegroups 2013-11-13 15:12:57 -05:00			`elif attribute == 'groups': # no setting, provide a blank`
			`val = []`
Add support for showing all as well as current attributes 2013-11-03 10:36:03 -05:00			`else: # no setting, provide a blank`
Have configmanager protect against unrecognized data. When doing a set, check the keynames and value types as appropriate. raise ValueError in the configmanager case since it could be code or human mistake. attributes plugin then catches this error and propogates it up as an InvalidArgumentException if it is just trying to pass in data from user. 2014-02-13 12:58:03 -05:00			`val = {'value': None}`
Add support for showing all as well as current attributes 2013-11-03 10:36:03 -05:00			`if attribute.startswith('secret.'):`
			`yield msg.CryptedAttributes(node,`
Have configmanager protect against unrecognized data. When doing a set, check the keynames and value types as appropriate. raise ValueError in the configmanager case since it could be code or human mistake. attributes plugin then catches this error and propogates it up as an InvalidArgumentException if it is just trying to pass in data from user. 2014-02-13 12:58:03 -05:00			`{attribute: val},`
			`allattributes.node[attribute]['description'])`
Enumerate nodegroups 2013-11-13 15:12:57 -05:00			`elif isinstance(val, list):`
			`yield msg.ListAttributes(node,`
Have configmanager protect against unrecognized data. When doing a set, check the keynames and value types as appropriate. raise ValueError in the configmanager case since it could be code or human mistake. attributes plugin then catches this error and propogates it up as an InvalidArgumentException if it is just trying to pass in data from user. 2014-02-13 12:58:03 -05:00			`{attribute: val},`
			`allattributes.node[attribute]['description'])`
Add support for showing all as well as current attributes 2013-11-03 10:36:03 -05:00			`else:`
			`yield msg.Attributes(node,`
Have configmanager protect against unrecognized data. When doing a set, check the keynames and value types as appropriate. raise ValueError in the configmanager case since it could be code or human mistake. attributes plugin then catches this error and propogates it up as an InvalidArgumentException if it is just trying to pass in data from user. 2014-02-13 12:58:03 -05:00			`{attribute: val['value']},`
			`allattributes.node[attribute]['description'])`
Have rest explorer actually mostly work as expected 2013-11-07 14:39:34 -05:00			`elif element[-1] == 'current':`
Add support for showing all as well as current attributes 2013-11-03 10:36:03 -05:00			`for node in attributes.iterkeys():`
			`for attribute in sorted(attributes[node].iterkeys()):`
			`currattr = attributes[node][attribute]`
			`if 'value' in currattr:`
			`yield msg.Attributes(node,`
Have configmanager protect against unrecognized data. When doing a set, check the keynames and value types as appropriate. raise ValueError in the configmanager case since it could be code or human mistake. attributes plugin then catches this error and propogates it up as an InvalidArgumentException if it is just trying to pass in data from user. 2014-02-13 12:58:03 -05:00			`{attribute: currattr['value']},`
			`allattributes.node[attribute]['description'])`
Add support for showing all as well as current attributes 2013-11-03 10:36:03 -05:00			`elif 'cryptvalue' in currattr:`
			`yield msg.CryptedAttributes(node,`
Have configmanager protect against unrecognized data. When doing a set, check the keynames and value types as appropriate. raise ValueError in the configmanager case since it could be code or human mistake. attributes plugin then catches this error and propogates it up as an InvalidArgumentException if it is just trying to pass in data from user. 2014-02-13 12:58:03 -05:00			`{attribute: currattr},`
			`allattributes.node[attribute]['description'])`
Support list style attributes (notably groups) in the api explorer 2013-11-13 10:36:43 -05:00			`elif isinstance(currattr, list):`
			`yield msg.ListAttributes(node,`
Have configmanager protect against unrecognized data. When doing a set, check the keynames and value types as appropriate. raise ValueError in the configmanager case since it could be code or human mistake. attributes plugin then catches this error and propogates it up as an InvalidArgumentException if it is just trying to pass in data from user. 2014-02-13 12:58:03 -05:00			`{attribute: currattr},`
			`allattributes.node[attribute]['description'])`
Add support for showing all as well as current attributes 2013-11-03 10:36:03 -05:00			`else:`
			`print repr(currattr)`
			`raise Exception("BUGGY ATTRIBUTE FOR NODE")`
Implement ability to change settings for an existing node 2013-11-03 10:21:54 -05:00

			`def update(nodes, element, configmanager, inputdata):`
			`updatedict = {}`
			`for node in nodes:`
			`updatenode = inputdata.get_attributes(node)`
			`if updatenode:`
			`updatedict[node] = updatenode`
Have configmanager protect against unrecognized data. When doing a set, check the keynames and value types as appropriate. raise ValueError in the configmanager case since it could be code or human mistake. attributes plugin then catches this error and propogates it up as an InvalidArgumentException if it is just trying to pass in data from user. 2014-02-13 12:58:03 -05:00			`try:`
			`configmanager.set_node_attributes(updatedict)`
			`except ValueError:`
			`raise exc.InvalidArgumentException()`
Have attribute set return some data Handle html form arrays with blanks, removing blank values (more complex usage requires JSON) 2013-11-13 14:52:32 -05:00			`return retrieve(nodes, element, configmanager, inputdata)`