sfa/dummy/dummyslices.py

   1 import time
   2 from types import StringTypes
   3 from collections import defaultdict
   4
   5 from sfa.util.sfatime import utcparse, datetime_to_epoch
   6 from sfa.util.sfalogging import logger
   7 from sfa.util.xrn import Xrn, get_leaf, get_authority, urn_to_hrn
   8
   9 from sfa.rspecs.rspec import RSpec
  10 from sfa.storage.model import SliverAllocation
  11 from sfa.storage.alchemy import dbsession
  12
  13 from sfa.dummy.dummyxrn import DummyXrn, hrn_to_dummy_slicename
  14
  15 MAXINT =  2L**31-1
  16
  17 class DummySlices:
  18
  19     def __init__(self, driver):
  20         self.driver = driver
  21
  22     def get_slivers(self, xrn, node=None):
  23         hrn, type = urn_to_hrn(xrn)
  24
  25         slice_name = hrn_to_dummy_slicename(hrn)
  26
  27         slices = self.driver.shell.GetSlices({'slice_name': slice_name})
  28         slice = slices[0]
  29         # Build up list of users and slice attributes
  30         user_ids = slice['user_ids']
  31         # Get user information
  32         all_users_list = self.driver.shell.GetUsers({'user_id':user_ids})
  33         all_users = {}
  34         for user in all_users_list:
  35             all_users[user['user_id']] = user
  36
  37         # Build up list of keys
  38         all_keys = set()
  39         for user in all_users_list:
  40             all_keys.extend(user['keys'])
  41
  42         slivers = []
  43         for slice in slices:
  44             keys = all_keys
  45             # XXX Sanity check; though technically this should be a system invariant
  46             # checked with an assertion
  47             if slice['expires'] > MAXINT:  slice['expires']= MAXINT
  48
  49             slivers.append({
  50                 'hrn': hrn,
  51                 'name': slice['name'],
  52                 'slice_id': slice['slice_id'],
  53                 'expires': slice['expires'],
  54                 'keys': keys,
  55             })
  56
  57         return slivers
  58
  59
  60     def verify_slice_nodes(self, slice_urn, slice, rspec_nodes):
  61
  62         slivers = {}
  63         for node in rspec_nodes:
  64             hostname = node.get('component_name')
  65             client_id = node.get('client_id')
  66             component_id = node.get('component_id').strip()
  67             if hostname:
  68                 hostname = hostname.strip()
  69             elif component_id:
  70                 hostname = xrn_to_hostname(component_id)
  71             if hostname:
  72                 slivers[hostname] = {'client_id': client_id, 'component_id': component_id}
  73         all_nodes = self.driver.shell.GetNodes()
  74         requested_slivers = []
  75         for node in all_nodes:
  76             if node['hostname'] in slivers.keys():
  77                 requested_slivers.append(node['node_id'])
  78
  79         if 'node_ids' not in slice.keys():
  80             slice['node_ids']=[]
  81         nodes = self.driver.shell.GetNodes({'node_ids': slice['node_ids']})
  82         current_slivers = [node['node_id'] for node in nodes]
  83
  84         # remove nodes not in rspec
  85         deleted_nodes = list(set(current_slivers).difference(requested_slivers))
  86
  87         # add nodes from rspec
  88         added_nodes = list(set(requested_slivers).difference(current_slivers))
  89
  90         try:
  91             self.driver.shell.AddSliceToNodes({'slice_id': slice['slice_id'], 'node_ids': added_nodes})
  92             self.driver.shell.DeleteSliceFromNodes({'slice_id': slice['slice_id'], 'node_ids': deleted_nodes})
  93
  94         except:
  95             logger.log_exc('Failed to add/remove slice from nodes')
  96
  97         slices = self.driver.shell.GetSlices({'slice_name': slice['slice_name']})
  98         resulting_nodes = self.driver.shell.GetNodes({'node_ids': slices[0]['node_ids']})
  99
 100         # update sliver allocations
 101         for node in resulting_nodes:
 102             client_id = slivers[node['hostname']]['client_id']
 103             component_id = slivers[node['hostname']]['component_id']
 104             sliver_hrn = '%s.%s-%s' % (self.driver.hrn, slice['slice_id'], node['node_id'])
 105             sliver_id = Xrn(sliver_hrn, type='sliver').urn
 106             record = SliverAllocation(sliver_id=sliver_id, client_id=client_id,
 107                                       component_id=component_id,
 108                                       slice_urn = slice_urn,
 109                                       allocation_state='geni_allocated')
 110             record.sync()
 111         return resulting_nodes
 112
 113
 114     def verify_slice(self, slice_hrn, slice_record, expiration, options={}):
 115         slicename = hrn_to_dummy_slicename(slice_hrn)
 116         parts = slicename.split("_")
 117         login_base = parts[0]
 118         slices = self.driver.shell.GetSlices({'slice_name': slicename})
 119         if not slices:
 120             slice = {'slice_name': slicename}
 121             # add the slice
 122             slice['slice_id'] = self.driver.shell.AddSlice(slice)
 123             slice['node_ids'] = []
 124             slice['user_ids'] = []
 125         else:
 126             slice = slices[0]
 127             if slice_record and slice_record.get('expires'):
 128                 requested_expires = int(datetime_to_epoch(utcparse(slice_record['expires'])))
 129                 if requested_expires and slice['expires'] != requested_expires:
 130                     self.driver.shell.UpdateSlice( {'slice_id': slice['slice_id'], 'fields':{'expires' : expiration}})
 131
 132         return slice
 133
 134     def verify_users(self, slice_hrn, slice_record, users, options={}):
 135         slice_name = hrn_to_dummy_slicename(slice_hrn)
 136         users_by_email = {}
 137         for user in users:
 138             user['urn'] = user['urn'].lower()
 139             hrn, type = urn_to_hrn(user['urn'])
 140             username = get_leaf(hrn)
 141             user['username'] = username
 142
 143             if 'email' in user:
 144                 user['email'] = user['email'].lower()
 145                 users_by_email[user['email']] = user
 146
 147         # start building a list of existing users
 148         existing_users_by_email = {}
 149         existing_slice_users_by_email = {}
 150         existing_users = self.driver.shell.GetUsers()
 151         existing_slice_users_ids = self.driver.shell.GetSlices({'slice_name': slice_name})[0]['user_ids']
 152         for user in existing_users:
 153             existing_users_by_email[user['email']] = user
 154             if user['user_id'] in existing_slice_users_ids:
 155                 existing_slice_users_by_email[user['email']] = user
 156
 157         add_users_by_email = set(users_by_email).difference(existing_slice_user_by_email)
 158         delete_users_by_email = set(existing_slice_user_by_email).difference(users_by_email)
 159         try:
 160             for user in add_users_by_email:
 161                 self.driver.shell.AddUser()
 162         except:
 163             pass
 164
 165
 166     def verify_keys(self, old_users, new_users, options={}):
 167         # existing keys
 168         existing_keys = []
 169         for user in old_users:
 170              existing_keys.append(user['keys'])
 171         userdict = {}
 172         for user in old_users:
 173             userdict[user['email']] = user
 174
 175         # add new keys
 176         requested_keys = []
 177         updated_users = []
 178         for user in new_users:
 179             user_keys = user.get('keys', [])
 180             updated_users.append(user)
 181             for key_string in user_keys:
 182                 requested_keys.append(key_string)
 183                 if key_string not in existing_keys:
 184                     key = key_string
 185                     try:
 186                         self.driver.shell.AddUserKey({'user_id': user['user_id'], 'key':key})
 187
 188                     except:
 189                         pass
 190         # remove old keys (only if we are not appending)
 191         append = options.get('append', True)
 192         if append == False:
 193             removed_keys = set(existing_keys).difference(requested_keys)
 194             for key in removed_keys:
 195                  try:
 196                      self.driver.shell.DeleteKey({'key': key})
 197                  except:
 198                      pass
 199