1 from PLC.Faults import *
2 from PLC.Method import Method
3 from PLC.Parameter import Parameter, Mixed
4 from PLC.AddressTypes import AddressType, AddressTypes
5 from PLC.Addresses import Address, Addresses
6 from PLC.Auth import Auth
8 class AddAddressTypeToAddress(Method):
10 Adds an address type to the specified address.
12 PIs may only update addresses of their own sites.
14 Returns 1 if successful, faults otherwise.
17 roles = ['admin', 'pi']
21 Mixed(AddressType.fields['address_type_id'],
22 AddressType.fields['name']),
23 Address.fields['address_id']
26 returns = Parameter(int, '1 if successful')
29 object_type = 'Address'
32 def call(self, auth, address_type_id_or_name, address_id):
33 address_types = AddressTypes(self.api, [address_type_id_or_name])
35 raise PLCInvalidArgument, "No such address type"
36 address_type = address_types[0]
38 addresses = Addresses(self.api, [address_id])
40 raise PLCInvalidArgument, "No such address"
41 address = addresses[0]
43 if 'admin' not in self.caller['roles']:
44 if address['site_id'] not in self.caller['site_ids']:
45 raise PLCPermissionDenied, "Address must be associated with one of your sites"
47 address.add_address_type(address_type)
48 self.object_ids = [address['address_id']]