PLC/Methods/AddSlice.py

   1 import re
   2
   3 from PLC.Faults import *
   4 from PLC.Method import Method
   5 from PLC.Parameter import Parameter, Mixed
   6 from PLC.Slices import Slice, Slices
   7 from PLC.Auth import Auth
   8 from PLC.Sites import Site, Sites
   9
  10 can_update = lambda (field, value): field in \
  11              ['name', 'instantiation', 'url', 'description', 'max_nodes']
  12
  13 class AddSlice(Method):
  14     """
  15     Adds a new slice. Any fields specified in slice_fields are used,
  16     otherwise defaults are used.
  17
  18     Valid slice names are lowercase and begin with the login_base
  19     (slice prefix) of a valid site, followed by a single
  20     underscore. Thereafter, only letters, numbers, or additional
  21     underscores may be used.
  22
  23     PIs may only add slices associated with their own sites (i.e.,
  24     slice prefixes must always be the login_base of one of their
  25     sites).
  26
  27     Returns the new slice_id (> 0) if successful, faults otherwise.
  28     """
  29
  30     roles = ['admin', 'pi']
  31
  32     slice_fields = dict(filter(can_update, Slice.fields.items()))
  33
  34     accepts = [
  35         Auth(),
  36         slice_fields
  37         ]
  38
  39     returns = Parameter(int, 'New slice_id (> 0) if successful')
  40
  41     event_type = 'Add'
  42     object_type = 'Slice'
  43
  44     def call(self, auth, slice_fields):
  45         slice_fields = dict(filter(can_update, slice_fields.items()))
  46
  47         # 1. Lowercase.
  48         # 2. Begins with login_base (only letters).
  49         # 3. Then single underscore after login_base.
  50         # 4. Then letters, numbers, or underscores.
  51         name = slice_fields['name']
  52         good_name = r'^[a-z]+_[a-z0-9_]+$'
  53         if not name or \
  54            not re.match(good_name, name):
  55             raise PLCInvalidArgument, "Invalid slice name"
  56
  57         # Get associated site details
  58         login_base = name.split("_")[0]
  59         sites = Sites(self.api, [login_base])
  60         if not sites:
  61             raise PLCInvalidArgument, "Invalid slice prefix %s in %s"%(login_base,name)
  62         site = sites[0]
  63
  64         if 'admin' not in self.caller['roles']:
  65             if site['site_id'] not in self.caller['site_ids']:
  66                 raise PLCPermissionDenied, "Slice prefix %s must be the same as the login_base of one of your sites"%login_base
  67
  68         if len(site['slice_ids']) >= site['max_slices']:
  69             raise PLCInvalidArgument, "Site %s has reached (%d) its maximum allowable slice count (%d)"%(site['name'],
  70                                                                                                          len(site['slice_ids']),
  71                                                                                                          site['max_slices'])
  72
  73         slice = Slice(self.api, slice_fields)
  74         slice['creator_person_id'] = self.caller['person_id']
  75         slice['site_id'] = site['site_id']
  76         slice.sync()
  77
  78         self.object_ids = [slice['slice_id']]
  79
  80         return slice['slice_id']