PLC/Methods/AddSlice.py

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