PLC/Methods/AdmIsPersonInRole.py

   1 from types import StringTypes
   2
   3 from PLC.Faults import *
   4 from PLC.Method import Method
   5 from PLC.Parameter import Parameter, Mixed
   6 from PLC.Persons import Person, Persons
   7 from PLC.Auth import PasswordAuth
   8 from PLC.Roles import Role, Roles
   9
  10 class AdmIsPersonInRole(Method):
  11     """
  12     Returns 1 if the specified account has the specified role, 0
  13     otherwise. This function differs from AdmGetPersonRoles() in that
  14     any authorized user can call it. It is currently restricted to
  15     verifying PI roles.
  16     """
  17
  18     roles = ['admin', 'pi', 'user', 'tech']
  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 account has role, 0 otherwise")
  29
  30     status = "useless"
  31
  32     def call(self, auth, person_id_or_email, role_id_or_name):
  33         # This is a totally fucked up function. I have no idea why it
  34         # exists or who calls it, but here is how it is supposed to
  35         # work.
  36
  37         # Only allow PI roles to be checked
  38         roles = {}
  39         for role_id, role in Roles(self.api).iteritems():
  40             roles[role_id] = role['name']
  41             roles[role['name']] = role_id
  42
  43         if role_id_or_name not in roles:
  44             raise PLCInvalidArgument, "Invalid role identifier or name"
  45
  46         if isinstance(role_id_or_name, int):
  47             role_id = role_id_or_name
  48         else:
  49             role_id = roles[role_id_or_name]
  50
  51         if roles[role_id] != "pi":
  52             raise PLCInvalidArgument, "Only the PI role may be checked"
  53
  54         # Get account information
  55         persons = Persons(self.api, [person_id_or_email])
  56
  57         # Rather than raise an error, and indicate whether or not
  58         # the person is real, return 0.
  59         if not persons:
  60             return 0
  61
  62         person = persons.values()[0]
  63
  64         if role_id in person['role_ids']:
  65             return 1
  66
  67         return 0