PLC/Methods/DeleteLeases.py

   1 from PLC.Faults import *
   2 from PLC.Method import Method
   3 from PLC.Parameter import Parameter, Mixed
   4 from PLC.Auth import Auth
   5 from PLC.Leases import Lease, Leases
   6 from PLC.Slices import Slice, Slices
   7
   8 class DeleteLeases(Method):
   9     """
  10     Deletes a lease.
  11
  12     Users may only delete leases attached to their slices.
  13     PIs may delete any of the leases for slices at their sites, or any
  14     slices of which they are members. Admins may delete any lease.
  15
  16     Returns 1 if successful, faults otherwise.
  17     """
  18
  19     roles = ['admin', 'pi', 'tech', 'user']
  20
  21     accepts = [
  22         Auth(),
  23         Mixed(Lease.fields['lease_id'],[ Lease.fields['lease_id']]),
  24         ]
  25
  26     returns = Parameter(int, '1 if successful')
  27
  28
  29     def call(self, auth, lease_ids):
  30         # Get associated lease details
  31         leases = Leases(self.api, lease_ids)
  32         if len(leases) != len(lease_ids):
  33             raise PLCInvalidArgument, "Could not find all leases %r"%lease_ids
  34
  35         # fetch related slices
  36         slices = Slices(self.api, [ lease['slice_id'] for lease in leases],['slice_id','person_ids'])
  37         # create hash on slice_id
  38         slice_map = dict ( [ (slice['slice_id'],slice) for slice in slices ] )
  39
  40         lease_ids=[lease['lease_id'] for lease in leases]
  41         for lease in leases:
  42             if 'admin' not in self.caller['roles']:
  43                 slice=slice_map[lease['slice_id']]
  44                 # check slices only once
  45                 if not slice.has_key('verified'):
  46                     if self.caller['person_id'] in slice['person_ids']:
  47                         pass
  48                     elif 'pi' not in self.caller['roles']:
  49                         raise PLCPermissionDenied, "Not a member of slice %r"%slice['name']
  50                     elif slice['site_id'] not in self.caller['site_ids']:
  51                         raise PLCPermissionDenied, "Slice %r not associated with any of your sites"%slice['name']
  52                 slice['verified']=True
  53
  54             lease.delete()
  55
  56         # Logging variables
  57         self.event_objects = {'Lease': lease_ids }
  58         self.message = 'Leases %r deleted' % lease_ids
  59
  60         return 1