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 'verified' not in slice:
  46                     if self.caller['person_id'] in slice['person_ids']:
  47                         slice['verified']=True
  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
  53             lease.delete()
  54
  55         # Logging variables
  56         self.event_objects = {'Lease': lease_ids }
  57         self.message = 'Leases %r deleted' % lease_ids
  58
  59         return 1