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 d55aa4d..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,20 +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')
 
-    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.event_objects = {'AddressType' : [address_type['address_type_id']]}
+
         return address_type['address_type_id']