def allocate(self, urn, rspec_string, expiration, options={}):
xrn = Xrn(urn)
aggregate = PlAggregate(self)
slices = PlSlices(self)
sfa_peer = slices.get_sfa_peer(xrn.get_hrn())
slice_record = None
users = options.get('geni_users', [])
if users:
slice_record = users[0].get('slice_record', {})
# parse rspec
rspec = RSpec(rspec_string)
requested_attributes = rspec.version.get_slice_attributes()
# ensure site record exists
site = slices.verify_site(xrn.hrn,
slice_record,
sfa_peer,
options=options)
# ensure slice record exists
slice = slices.verify_slice(xrn.hrn,
slice_record,
sfa_peer,
expiration=expiration,
options=options)
# ensure person records exists
persons = slices.verify_persons(xrn.hrn,
slice,
users,
sfa_peer,
options=options)
# ensure slice attributes exists
slices.verify_slice_attributes(slice,
requested_attributes,
options=options)
# add/remove slice from nodes
request_nodes = rspec.version.get_nodes_with_slivers()
nodes = slices.verify_slice_nodes(urn, slice, request_nodes)
# add/remove links links
slices.verify_slice_links(slice, rspec.version.get_link_requests(),
nodes)
# add/remove leases
rspec_requested_leases = rspec.version.get_leases()
leases = slices.verify_slice_leases(slice, rspec_requested_leases)
return aggregate.describe([xrn.get_urn()], version=rspec.version)
def prepare_slice(self, api, slice_xrn, creds, users):
reg_objects = self._get_registry_objects(slice_xrn, creds, users)
(hrn, type) = urn_to_hrn(slice_xrn)
slices = PlSlices(self.driver)
peer = slices.get_peer(hrn)
sfa_peer = slices.get_sfa_peer(hrn)
slice_record = None
if users:
slice_record = users[0].get('slice_record', {})
registry = api.registries[api.hrn]
credential = api.getCredential()
# ensure site record exists
site = slices.verify_site(hrn, slice_record, peer, sfa_peer)
# ensure slice record exists
slice = slices.verify_slice(hrn, slice_record, peer, sfa_peer)
# ensure person records exists
persons = slices.verify_persons(hrn, slice, users, peer, sfa_peer)
def provision(self, urns, options={}):
# update users
slices = PlSlices(self)
aggregate = PlAggregate(self)
slivers = aggregate.get_slivers(urns)
if not slivers:
sliver_id_parts = Xrn(urns[0]).get_sliver_id_parts()
filter = {}
try:
filter['slice_id'] = int(sliver_id_parts[0])
except ValueError:
filter['name'] = sliver_id_parts[0]
slices = self.shell.GetSlices(filter, ['hrn'])
if not slices:
raise Forbidden(
"Unable to locate slice record for sliver: %s" % xrn)
slice = slices[0]
slice_urn = hrn_to_urn(slice['hrn'], type='slice')
urns = [slice_urn]
else:
slice_id = slivers[0]['slice_id']
slice_hrn = self.shell.GetSliceHrn(slice_id)
slice = self.shell.GetSlices({'slice_id': slice_id})[0]
slice['hrn'] = slice_hrn
sfa_peer = slices.get_sfa_peer(slice['hrn'])
users = options.get('geni_users', [])
persons = slices.verify_persons(slice['hrn'],
slice,
users,
sfa_peer,
options=options)
# update sliver allocation states and set them to geni_provisioned
sliver_ids = [sliver['sliver_id'] for sliver in slivers]
dbsession = self.api.dbsession()
SliverAllocation.set_allocations(sliver_ids, 'geni_provisioned',
dbsession)
version_manager = VersionManager()
rspec_version = version_manager.get_version(
options['geni_rspec_version'])
return self.describe(urns, rspec_version, options=options)
