Merge from trunk

[plcapi.git] / trunk / PLC / Methods / UpdateSliceAttribute.py

diff --git a/trunk/PLC/Methods/UpdateSliceAttribute.py b/trunk/PLC/Methods/UpdateSliceAttribute.py
new file mode 100644 (file)
index 0000000..43cf1f3
--- /dev/null
+++ b/trunk/PLC/Methods/UpdateSliceAttribute.py
@@ -0,0 +1,68 @@
+from PLC.Faults import *
+from PLC.Method import Method
+from PLC.Parameter import Parameter, Mixed
+from PLC.SliceAttributes import SliceAttribute, SliceAttributes
+from PLC.Slices import Slice, Slices
+from PLC.InitScripts import InitScript, InitScripts
+from PLC.Auth import Auth
+
+class UpdateSliceAttribute(Method):
+    """
+    Updates the value of an existing slice or sliver attribute.
+
+    Users may only update attributes of slices or slivers of which
+    they are members. PIs may only update attributes of slices or
+    slivers at their sites, or of which they are members. Admins may
+    update attributes of any slice or sliver.
+
+    Returns 1 if successful, faults otherwise.
+    """
+
+    roles = ['admin', 'pi', 'user']
+
+    accepts = [
+        Auth(),
+        SliceAttribute.fields['slice_attribute_id'],
+       Mixed(SliceAttribute.fields['value'],
+              InitScript.fields['initscript_id'])
+        ]
+
+    returns = Parameter(int, '1 if successful')
+
+    def call(self, auth, slice_attribute_id, value):
+        slice_attributes = SliceAttributes(self.api, [slice_attribute_id])
+        if not slice_attributes:
+            raise PLCInvalidArgument, "No such slice attribute"
+        slice_attribute = slice_attributes[0]
+
+        slices = Slices(self.api, [slice_attribute['slice_id']])
+        if not slices:
+            raise PLCInvalidArgument, "No such slice"
+        slice = slices[0]
+
+        if slice['peer_id'] is not None:
+            raise PLCInvalidArgument, "Not a local slice"
+
+        assert slice_attribute['slice_attribute_id'] in slice['slice_attribute_ids']
+
+        if 'admin' not in self.caller['roles']:
+            if self.caller['person_id'] in slice['person_ids']:
+                pass
+            elif 'pi' not in self.caller['roles']:
+                raise PLCPermissionDenied, "Not a member of the specified slice"
+            elif slice['site_id'] not in self.caller['site_ids']:
+                raise PLCPermissionDenied, "Specified slice not associated with any of your sites"
+
+            if slice_attribute['min_role_id'] is not None and \
+               min(self.caller['role_ids']) > slice_attribute['min_role_id']:
+                raise PLCPermissionDenied, "Not allowed to update the specified attribute"
+       
+       if slice_attribute['name'] in ['plc_initscript_id']:
+            initscripts = InitScripts(self.api, {'enabled': True, 'initscript_id': int(value)})
+            if not initscripts:
+                raise PLCInvalidArgument, "No such plc initscript"     
+
+        slice_attribute['value'] = unicode(value)
+        slice_attribute.sync()
+       self.event_objects = {'SliceAttribute': [slice_attribute['slice_attribute_id']]}
+        return 1