5 from time import mktime
8 from django.shortcuts import render_to_response
9 from django.http import HttpResponse
11 from sfa.trust.certificate import Keypair, Certificate
12 from sfa.client.sfaserverproxy import SfaServerProxy
13 from sfa.client.return_value import ReturnValue
14 from sfa.util.xrn import Xrn, get_leaf, get_authority, hrn_to_urn, urn_to_hrn
16 from manifold.core.query import Query
17 from manifold.operators.rename import do_rename
19 from manifoldapi.manifoldapi import execute_admin_query
21 from unfold.loginrequired import LoginRequiredView
23 from myslice.settings import logger, config
25 from rest.json_encoder import MyEncoder
28 def unique_call_id(): return uuid.uuid4().urn
30 def dispatch(request, method):
43 if request.method == 'POST':
44 req_items = request.POST
45 elif request.method == 'GET':
46 req_items = request.GET
48 logger.debug("dispatch got = %s" % req_items.dict())
49 #t = dict(req_items.iterlists())
50 #rspec = req_items.getlist('rspec')
51 #logger.debug("dispatch got = %s" % t)
53 platforms = req_items.getlist('platform[]')
54 logger.debug("req_items type = %s" % type(req_items.dict()))
55 for k in req_items.dict():
58 rspec = req_items.get(k)
60 hrn = req_items.get(k)
62 urn = req_items.get(k)
64 object_type = req_items.get(k)
71 display = req_items.get(k)
75 rspec = json.loads(rspec)
77 logger.debug("rspec type = %s" % type(rspec))
78 if type(rspec) is dict:
79 rspec = xmltodict.unparse(rspec)
80 results = sfa_client(request, method, hrn=hrn, urn=urn, object_type=object_type, rspec=rspec, recursive=recursive, options=options, platforms=platforms)
81 if display == 'table':
82 return render_to_response('table-default.html', {'data' : data, 'fields' : columns, 'id' : '@component_id', 'options' : None})
84 return HttpResponse(json.dumps(results, cls=MyEncoder), content_type="application/json")
86 def get_user_account(request, user_email, platform_name):
88 Returns the user configuration for a given platform.
89 This function does not resolve references.
91 user_query = Query().get('local:user').filter_by('email', '==', user_email).select('user_id')
92 user_details = execute_admin_query(request, user_query)
93 platform_query = Query().get('local:platform').filter_by('platform', '==', platform_name).select('platform_id')
94 platform_details = execute_admin_query(request, platform_query)
96 account_query = Query().get('local:account').filter_by('platform_id','==',platform_details[0]['platform_id']).filter_by('user_id', '==', user_details[0]['user_id']).select('user_id','platform_id','auth_type','config')
97 accounts = execute_admin_query(request, account_query)
100 raise Exception, "this account does not exist"
102 if accounts[0]['auth_type'] == 'reference':
103 pf = json.loads(accounts[0]['config'])['reference_platform']
104 return get_user_account(request, user_email, pf)
108 def sfa_client(request, method, hrn=None, urn=None, object_type=None, rspec=None, recursive=None, options=None, platforms=None, admin=False):
110 Config = ConfigParser.ConfigParser()
111 monitor_file = os.path.abspath(os.path.dirname(__file__) + '/../myslice/monitor.ini')
112 Config.read(monitor_file)
115 user_email, admin_password = config.manifold_admin_user_password()
117 #logger.debug(request.session['user']['email'])
118 user_email = request.session['user']['email']
126 if object_type is None:
131 logger.debug("RSPEC = %s" % rspec)
132 if recursive is None:
136 if platforms is None:
139 if method not in ['GetVersion','ListResources']:
142 hrn = urn_to_hrn(urn)
144 urn = hrn_to_urn(hrn, object_type)
147 raise Exception, "Provide urn OR hrn + type as parameters of method %s" % method
149 if len(platforms)==0:
150 platforms = get_platforms(request)
151 #platforms.append('myslice')
152 #results = {'method':method,'platforms':platforms,'rspec':rspec,'options':options}
159 api_options['list_leases'] = 'all'
162 platform = get_platform_config(request, pf)
163 if 'rspec_type' in platform and 'rspec_version' in platform:
164 api_options['geni_rspec_version'] = {'type': platform['rspec_type'],'version': platform['rspec_version']}
166 api_options['geni_rspec_version'] = {'type': 'GENI', 'version': '3'}
167 if 'sm' in platform and len(platform['sm']) > 0:
169 server_url = platform['sm']
170 if 'rm' in platform and len(platform['rm']) > 0:
172 server_url = platform['rm']
173 if 'registry' in platform and len(platform['registry']) > 0:
175 server_url = platform['registry']
177 if not Config.has_option('monitor', 'cert') :
178 #return HttpResponse(json.dumps({'error' : '-1'}), content_type="application/json")
179 return {'error' : '-1', 'msg': 'monitor.ini has no cert configured'}
181 cert = os.path.abspath(Config.get('monitor', 'cert'))
182 if not os.path.isfile(cert) :
183 #return HttpResponse(json.dumps({'error' : '-1'}), content_type="application/json")
184 return {'error' : '-1', 'msg': 'check cert file at %s'%cert}
186 if not Config.has_option('monitor', 'pkey') :
187 #return HttpResponse(json.dumps({'error' : '-2'}), content_type="application/json")
188 return {'error' : '-2'}
190 pkey = os.path.abspath(Config.get('monitor', 'pkey'))
191 if not os.path.isfile(pkey) :
192 #return HttpResponse(json.dumps({'error' : '-2'}), content_type="application/json")
193 return {'error' : '-2'}
195 server = SfaServerProxy(server_url, pkey, cert)
198 # Get user config from Manifold
199 user_config = get_user_config(request, user_email, pf)
200 if 'delegated_user_credential' in user_config:
201 user_cred = user_config['delegated_user_credential']
202 elif 'user_credential' in user_config:
203 user_cred = user_config['user_credential']
205 logger.error("no user credentials for user = ", user_email)
209 if 'delegated_%s_credentials'%object_type in user_config:
210 for obj_name, cred in user_config['delegated_%s_credentials'%object_type].items():
213 elif '%s_credentials'%object_type in user_config:
214 for obj_name, cred in user_config['%s_credentials'%object_type].items():
218 logger.error("no credentials for object")
219 logger.error(object_type)
220 logger.error(object_name)
224 if method == "GetVersion":
225 result = server.GetVersion()
229 if method == "ListResources":
230 result = server.ListResources([user_cred], api_options)
231 dict_result = xmltodict.parse(result['value'])
232 result['parsed'] = dict_result
233 if isinstance(dict_result['rspec']['node'], list):
234 columns.extend(dict_result['rspec']['node'][0].keys())
236 columns.extend(dict_result['rspec']['node'].keys())
238 elif method == "Describe":
239 version = server.GetVersion()
241 if version['geni_api'] == 2:
242 # ListResources(slice_hrn)
243 api_options['geni_slice_urn'] = urn
244 result = server.ListResources([object_cred], api_options)
245 dict_result = xmltodict.parse(result['value'])
246 # else GetVersion = v3
248 result = server.Describe([urn] ,[object_cred], api_options)
249 if isinstance(result, dict):
250 if result['value'] != 0:
251 dict_result = xmltodict.parse(result['value']['geni_rspec'])
253 result['parsed'] = dict_result
254 if 'rspec' in dict_result and 'node' in dict_result['rspec']:
255 if isinstance(dict_result['rspec']['node'], list):
256 columns.extend(dict_result['rspec']['node'][0].keys())
258 columns.extend(dict_result['rspec']['node'].keys())
260 elif method == 'Renew':
261 # Renew till 1 month from now
262 d = datetime.datetime.utcnow() + datetime.timedelta(365/12)
263 date = d.isoformat("T") + "Z"
264 result = server.Renew([urn] ,[object_cred], date, api_options)
265 elif method == 'Delete':
266 result = server.Delete([urn] ,[object_cred], api_options)
267 elif method == 'Allocate':
268 api_options['call_id'] = unique_call_id()
269 # List of users comes from the Registry
270 users = get_users_in_slice(request, hrn)
271 api_options['sfa_users'] = users
272 api_options['geni_users'] = users
274 version = server.GetVersion()
275 if version['geni_api'] == 2:
276 result = server.CreateSliver([urn] ,[object_cred], rspec, api_options)
277 # else GetVersion = v3
279 result = server.Allocate(urn ,[object_cred], rspec, api_options)
280 elif method == 'Provision':
282 # Nothing it is not supported by v2 AMs
283 version = server.GetVersion()
284 # List of users comes from the Registry
285 users = get_users_in_slice(request, hrn)
286 api_options['sfa_users'] = users
287 api_options['geni_users'] = users
288 if version['geni_api'] == 3:
289 api_options['call_id'] = unique_call_id()
290 result = server.Provision([urn] ,[object_cred], api_options)
291 elif method == 'Status':
292 result = server.Status([urn] ,[object_cred], api_options)
293 elif method == 'PerformOperationalAction':
295 # Nothing it is not supported by v2 AMs
296 version = server.GetVersion()
297 if version['geni_api'] == 3:
298 result = server.PerformOperationalAction([urn] ,[object_cred], action, api_options)
299 elif method == 'Shutdown':
300 result = server.Shutdown(urn ,[object_cred], api_options)
302 #return HttpResponse(json.dumps({'error' : '-3','msg':'method not supported by AM'}), content_type="application/json")
303 logger.debug('method %s not handled by AM' % method)
308 record_dict = {'urn': urn, 'hrn': hrn, 'type': object_type}
311 api_options['recursive'] = recursive
312 result = server.List(hrn, user_cred, api_options)
314 result = filter_records(object_type, result)
315 elif method == "Resolve":
317 # details can be True or False
318 api_options['details']=True
319 result = server.Resolve(hrn, user_cred, api_options)
321 result = filter_records(object_type, result)
322 elif method == "Register":
323 # record_dict must be crafted
324 # auth_cred must be selected in the list of auth_creds from user's account
325 result = server.Register(record_dict, auth_cred)
326 elif method == "Update":
327 # record_dict must be crafted
328 # object_cred must be selected in the list of creds for the object type
329 # from user's account
330 result = server.Update(record_dict, object_cred)
331 elif method == "Remove":
333 # auth_cred must be selected in the list of auth_creds from user's account
334 # object_type is required
335 result = server.Remove(hrn, auth_cred, object_type)
337 #return HttpResponse(json.dumps({'error' : '-3','msg':'method not supported by Registry'}), content_type="application/json")
338 logger.debug('method %s not handled by Registry' % method)
343 if 'rspec' in dict_result and 'node' in dict_result['rspec']:
344 if isinstance(dict_result['rspec']['node'], list):
345 data = data + dict_result['rspec']['node']
347 data.append(dict_result['rspec']['node'])
350 logger.error(traceback.format_exc())
352 results[pf] = {'error':'-3', 'error_msg': str(e)}
354 results['columns'] = columns
357 def rename(self,key,new_key):
358 ind = self._keys.index(key) #get the index of old key, O(N) operation
359 self._keys[ind] = new_key #replace old key with new key in self._keys
360 self[new_key] = self[key] #add the new key, this is added at the end of self._keys
361 self._keys.pop(-1) #pop the last item in self._keys
363 def get_users_in_slice(request, slice_hrn):
364 # select users.user_hrn, users.user_email, users.keys
366 # where slice_hrn=='onelab.upmc.r2d2.slice1'
367 users_query = Query().get('myslice:slice').filter_by('slice_hrn', '==', slice_hrn).select('users.user_hrn', 'users.user_urn', 'users.user_email','users.keys')
368 users = execute_admin_query(request, users_query)
369 rmap = {'user_urn':'urn','user_email':'email','user_hrn':'hrn'}
371 for u in users[0]['users']:
373 for k,v in u.items():
381 def get_user_config(request, user_email, platform_name):
382 account = get_user_account(request, user_email, platform_name)
383 return json.loads(account['config']) if account['config'] else {}
385 def get_platforms(request):
387 platform_query = Query().get('local:platform').filter_by('gateway_type', '==', 'sfa').filter_by('disabled','==',0).select('platform')
388 platforms = execute_admin_query(request, platform_query)
391 ret.append(p['platform'])
394 def get_platform_config(request, platform_name):
395 platform_query = Query().get('local:platform').filter_by('platform', '==', platform_name).select('platform', 'config')
396 platforms = execute_admin_query(request, platform_query)
398 return json.loads(platforms[0]['config']) if platforms[0]['config'] else {}
400 def filter_records(type, records):
401 filtered_records = []
402 for record in records:
403 if (record['type'] == type) or (type == "all"):
404 filtered_records.append(record)
405 return filtered_records