2 from PLC.Faults import *
3 from PLC.Method import Method
4 from PLC.Parameter import Parameter, Mixed
5 from PLC.Keys import Key, Keys
6 from PLC.Persons import Person, Persons
7 from PLC.Auth import Auth
9 can_update = lambda (field, value): field in ['key_type','key']
11 class AddPersonKey(Method):
13 Adds a new key to the specified account.
15 Non-admins can only modify their own keys.
17 Returns the new key_id (> 0) if successful, faults otherwise.
20 roles = ['admin', 'pi', 'tech', 'user']
22 key_fields = dict(filter(can_update, Key.fields.items()))
26 Mixed(Person.fields['person_id'],
27 Person.fields['email']),
31 returns = Parameter(int, 'New key_id (> 0) if successful')
33 def call(self, auth, person_id_or_email, key_fields):
34 key_fields = dict(filter(can_update, key_fields.items()))
37 persons = Persons(self.api, [person_id_or_email])
39 raise PLCInvalidArgument, "No such account"
42 if person['peer_id'] is not None:
43 raise PLCInvalidArgument, "Not a local account"
45 # If we are not admin, make sure caller is adding a key to their account
46 if 'admin' not in self.caller['roles']:
47 if person['person_id'] != self.caller['person_id']:
48 raise PLCPermissionDenied, "You may only modify your own keys"
50 key = Key(self.api, key_fields)
51 key.sync(commit = False)
52 person.add_key(key, commit = True)
55 self.event_objects = {'Person': [person['person_id']],
56 'Key': [key['key_id']]}
57 self.message = 'Key %d added to person %d' % \
58 (key['key_id'], person['person_id'])