- support updating 'enabled' field

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

diff --git a/PLC/Methods/UpdateSliceAttribute.py b/PLC/Methods/UpdateSliceAttribute.py
index 425b7e6..5b7d48f 100644 (file)
--- a/PLC/Methods/UpdateSliceAttribute.py
+++ b/PLC/Methods/UpdateSliceAttribute.py
@@ -3,7 +3,7 @@ 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.Auth import PasswordAuth
+from PLC.Auth import Auth
 
 class UpdateSliceAttribute(Method):
     """
@@ -20,24 +20,29 @@ class UpdateSliceAttribute(Method):
     roles = ['admin', 'pi', 'user']
 
     accepts = [
-        PasswordAuth(),
+        Auth(),
         SliceAttribute.fields['slice_attribute_id'],
         SliceAttribute.fields['value']
         ]
 
     returns = Parameter(int, '1 if successful')
 
+    object_type = 'Slice'
+
     def call(self, auth, slice_attribute_id, value):
-        slice_attributes = SliceAttributes(self.api, [slice_attribute_id]).values()
+        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']]).values()
+        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']:
@@ -54,5 +59,5 @@ class UpdateSliceAttribute(Method):
 
         slice_attribute['value'] = value
         slice_attribute.sync()
-
+       self.object_ids = [slice_attribute['slice_attribute_id']]
         return 1