PLC/Methods/AddRoleToPerson.py

   1 from PLC.Faults import *
   2 from PLC.Method import Method
   3 from PLC.Parameter import Parameter, Mixed
   4 from PLC.Persons import Person, Persons
   5 from PLC.Auth import PasswordAuth
   6 from PLC.Roles import Roles
   7
   8 class AddRoleToPerson(Method):
   9     """
  10     Grants the specified role to the person.
  11
  12     PIs can only grant the tech and user roles to users and techs at
  13     their sites. ins can grant any role to any user.
  14
  15     Returns 1 if successful, faults otherwise.
  16     """
  17
  18     roles = ['admin', 'pi']
  19
  20     accepts = [
  21         PasswordAuth(),
  22         Mixed(Person.fields['person_id'],
  23               Person.fields['email']),
  24         Mixed(Parameter(int, "Role identifier"),
  25               Parameter(str, "Role name"))
  26         ]
  27
  28     returns = Parameter(int, '1 if successful')
  29
  30     def call(self, auth, person_id_or_email, role_id_or_name):
  31         # Get all roles
  32         roles = {}
  33         for role_id, role in Roles(self.api).iteritems():
  34             roles[role_id] = role['name']
  35             roles[role['name']] = role_id
  36
  37         if role_id_or_name not in roles:
  38             raise PLCInvalidArgument, "Invalid role identifier or name"
  39
  40         if isinstance(role_id_or_name, int):
  41             role_id = role_id_or_name
  42         else:
  43             role_id = roles[role_id_or_name]
  44
  45         # Get account information
  46         persons = Persons(self.api, [person_id_or_email])
  47         if not persons:
  48             raise PLCInvalidArgument, "No such account"
  49
  50         person = persons.values()[0]
  51
  52         # Authenticated function
  53         assert self.caller is not None
  54
  55         # Check if we can update this account
  56         if not self.caller.can_update(person):
  57             raise PLCPermissionDenied, "Not allowed to update specified account"
  58
  59         # Can only grant lesser (higher) roles to others
  60         if 'admin' not in self.caller['roles'] and \
  61            role_id <= min(self.caller['role_ids']):
  62             raise PLCInvalidArgument, "Not allowed to grant that role"
  63
  64         if role_id not in person['role_ids']:
  65             person.add_role(role_id)
  66
  67         return 1