Setting tag plcapi-5.4-2

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

diff --git a/PLC/Methods/AddAddressType.py b/PLC/Methods/AddAddressType.py
index 075fca2..2847714 100644 (file)
--- a/PLC/Methods/AddAddressType.py
+++ b/PLC/Methods/AddAddressType.py
@@ -2,9 +2,9 @@ from PLC.Faults import *
 from PLC.Method import Method
 from PLC.Parameter import Parameter, Mixed
 from PLC.AddressTypes import AddressType, AddressTypes
-from PLC.Auth import PasswordAuth
+from PLC.Auth import Auth
 
-can_update = lambda (field, value): field in ['description']
+can_update = lambda (field, value): field not in ['address_type_id']
 
 class AddAddressType(Method):
     """
@@ -16,25 +16,21 @@ class AddAddressType(Method):
 
     roles = ['admin']
 
-    update_fields = dict(filter(can_update, AddressType.fields.items()))
+    address_type_fields = dict(filter(can_update, AddressType.fields.items()))
 
     accepts = [
-        PasswordAuth(),
-        AddressType.fields['name'],
-        update_fields
+        Auth(),
+        address_type_fields
         ]
 
     returns = Parameter(int, 'New address_type_id (> 0) if successful')
-        
-    event_type = 'Add'    
-    object_type = 'AddressType'
-    object_ids = []
 
-    def call(self, auth, name, address_type_fields = {}):
+
+    def call(self, auth, address_type_fields):
         address_type_fields = dict(filter(can_update, address_type_fields.items()))
         address_type = AddressType(self.api, address_type_fields)
-        address_type['name'] = name
         address_type.sync()
-       self.object_ids = [address_type['address_type_id']]
-        
-       return address_type['address_type_id']
+
+        self.event_objects = {'AddressType' : [address_type['address_type_id']]}
+
+        return address_type['address_type_id']