PLC/Methods/AddPersonKey.py

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