fix typo. When sending email notification, use values from plc_config instead of...

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

diff --git a/PLC/Methods/UpdateSliceAttribute.py b/PLC/Methods/UpdateSliceAttribute.py
index 2efd280..3c21403 100644 (file)
--- a/PLC/Methods/UpdateSliceAttribute.py
+++ b/PLC/Methods/UpdateSliceAttribute.py
@@ -3,7 +3,8 @@ 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.InitScripts import InitScript, InitScripts
+from PLC.Auth import Auth
 
 class UpdateSliceAttribute(Method):
     """
@@ -20,28 +21,26 @@ class UpdateSliceAttribute(Method):
     roles = ['admin', 'pi', 'user']
 
     accepts = [
-        PasswordAuth(),
-        Mixed(SliceAttribute.fields['slice_id'],
-              SliceAttribute.fields['name']),
+        Auth(),
         SliceAttribute.fields['slice_attribute_id'],
-        SliceAttribute.fields['value']
+       Mixed(SliceAttribute.fields['value'],
+              InitScript.fields['name'])
         ]
 
     returns = Parameter(int, '1 if successful')
 
-    def call(self, auth, slice_id_or_name, slice_attribute_id, value):
-        slices = Slices(self.api, [slice_id_or_name]).values()
-        if not slices:
-            raise PLCInvalidArgument, "No such slice"
-        slice = slices[0]
-
-        slice_attributes = SliceAttributes(self.api, [slice_attribute_id]).values()
+    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]
 
-        if slice_attribute['slice_attribute_id'] not in slice['slice_attribute_ids']:
-            raise PLCInvalidArgument, "Invalid slice attribute ID"
+        slices = Slices(self.api, [slice_attribute['slice_id']])
+        if not slices:
+            raise PLCInvalidArgument, "No such slice"
+        slice = slices[0]
+
+        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']:
@@ -54,8 +53,13 @@ class UpdateSliceAttribute(Method):
             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['tagname'] in ['initscript']:
+            initscripts = InitScripts(self.api, {'enabled': True, 'name': value})
+            if not initscripts:
+                raise PLCInvalidArgument, "No such plc initscript"     
 
-        slice_attribute['value'] = value
+        slice_attribute['value'] = unicode(value)
         slice_attribute.sync()
-
+       self.event_objects = {'SliceAttribute': [slice_attribute['slice_attribute_id']]}
         return 1