from PLC.Faults import *
from PLC.Method import Method
from PLC.Parameter import Parameter, Mixed
-from PLC.Attributes import Attribute, Attributes
+from PLC.TagTypes import TagType, TagTypes
from PLC.Slices import Slice, Slices
from PLC.Nodes import Node, Nodes
from PLC.SliceAttributes import SliceAttribute, SliceAttributes
-from PLC.Auth import PasswordAuth
+from PLC.NodeGroups import NodeGroup, NodeGroups
+from PLC.InitScripts import InitScript, InitScripts
+from PLC.Auth import Auth
class AddSliceAttribute(Method):
"""
roles = ['admin', 'pi', 'user']
accepts = [
- PasswordAuth(),
- Mixed(SliceAttribute.fields['slice_id'],
- SliceAttribute.fields['name']),
- Mixed(SliceAttribute.fields['attribute_id'],
- SliceAttribute.fields['name']),
- SliceAttribute.fields['value'],
+ Auth(),
+ Mixed(Slice.fields['slice_id'],
+ Slice.fields['name']),
+ Mixed(SliceAttribute.fields['tag_type_id'],
+ SliceAttribute.fields['tagname']),
+ Mixed(SliceAttribute.fields['value'],
+ InitScript.fields['name']),
Mixed(Node.fields['node_id'],
- Node.fields['hostname'])
+ Node.fields['hostname'],
+ None),
+ Mixed(NodeGroup.fields['nodegroup_id'],
+ NodeGroup.fields['groupname'])
]
returns = Parameter(int, 'New slice_attribute_id (> 0) if successful')
- def call(self, auth, slice_id_or_name, attribute_id_or_name, value, node_id_or_hostname = None):
- slices = Slices(self.api, [slice_id_or_name]).values()
+ def call(self, auth, slice_id_or_name, tag_type_id_or_name, value, node_id_or_hostname = None, nodegroup_id_or_name = None):
+ slices = Slices(self.api, [slice_id_or_name])
if not slices:
- raise PLCInvalidArgument, "No such slice"
+ raise PLCInvalidArgument, "No such slice %r"%slice_id_or_name
slice = slices[0]
- attributes = Attributes(self.api, [attribute_id_or_name]).values()
- if not attributes:
- raise PLCInvalidArgument, "No such attribute"
- attribute = attributes[0]
+ tag_types = TagTypes(self.api, [tag_type_id_or_name])
+ if not tag_types:
+ raise PLCInvalidArgument, "No such tag type %r"%tag_type_id_or_name
+ tag_type = tag_types[0]
if 'admin' not in self.caller['roles']:
if self.caller['person_id'] in slice['person_ids']:
elif slice['site_id'] not in self.caller['site_ids']:
raise PLCPermissionDenied, "Specified slice not associated with any of your sites"
- if attribute['min_role_id'] is not None and \
- min(self.caller['role_ids']) > attribute['min_role_id']:
- raise PLCPermissionDenied, "Not allowed to set the specified attribute"
+ if tag_type['min_role_id'] is not None and \
+ min(self.caller['role_ids']) > tag_type['min_role_id']:
+ raise PLCPermissionDenied, "Not allowed to set the specified slice attribute"
+
+ # if initscript is specified, validate value
+ if tag_type['tagname'] in ['initscript']:
+ initscripts = InitScripts(self.api, {'enabled': True, 'name': value})
+ if not initscripts:
+ raise PLCInvalidArgument, "No such plc initscript %r"%value
slice_attribute = SliceAttribute(self.api)
slice_attribute['slice_id'] = slice['slice_id']
- slice_attribute['attribute_id'] = attribute['attribute_id']
- slice_attribute['value'] = value
+ slice_attribute['tag_type_id'] = tag_type['tag_type_id']
+ slice_attribute['value'] = unicode(value)
# Sliver attribute if node is specified
if node_id_or_hostname is not None:
- nodes = Nodes(self.api, [node_id_or_hostname]).values()
+ nodes = Nodes(self.api, [node_id_or_hostname])
if not nodes:
raise PLCInvalidArgument, "No such node"
node = nodes[0]
if node['node_id'] not in slice['node_ids']:
raise PLCInvalidArgument, "Node not in the specified slice"
-
slice_attribute['node_id'] = node['node_id']
+ # Sliver attribute shared accross nodes if nodegroup is sepcified
+ if nodegroup_id_or_name is not None:
+ nodegroups = NodeGroups(self.api, [nodegroup_id_or_name])
+ if not nodegroups:
+ raise PLCInvalidArgument, "No such nodegroup %r"%nodegroup_id_or_name
+ nodegroup = nodegroups[0]
+
+ slice_attribute['nodegroup_id'] = nodegroup['nodegroup_id']
+
+ # Check if slice attribute alreay exists
+ slice_attributes_check = SliceAttributes(self.api, {'slice_id': slice['slice_id'],
+ 'tagname': tag_type['tagname'],
+ 'value': value})
+ for slice_attribute_check in slice_attributes_check:
+ if 'node_id' in slice_attribute and slice_attribute['node_id'] == slice_attribute_check['node_id']:
+ raise PLCInvalidArgument, "Sliver attribute already exists"
+ if 'nodegroup_id' in slice_attribute and slice_attribute['nodegroup_id'] == slice_attribute_check['nodegroup_id']:
+ raise PLCInvalidArgument, "Slice attribute already exists for this nodegroup"
+ if node_id_or_hostname is None and nodegroup_id_or_name is None:
+ raise PLCInvalidArgument, "Slice attribute already exists"
+
slice_attribute.sync()
+ self.event_objects = {'SliceAttribute': [slice_attribute['slice_attribute_id']]}
return slice_attribute['slice_attribute_id']