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
18 from manifoldapi.manifoldapi import execute_admin_query
20 from unfold.loginrequired import LoginRequiredView
22 from myslice.settings import logger, config
24 from rest.json_encoder import MyEncoder
26 def dispatch(request, method):
39 if request.method == 'POST':
40 req_items = request.POST
41 elif request.method == 'GET':
42 req_items = request.GET
44 for el in req_items.items():
45 if el[0].startswith('rspec'):
47 elif el[0].startswith('platform'):
48 platforms = req_items.getlist('platform[]')
49 #elif el[0].startswith('options'):
50 # options += req_items.getlist('options[]')
51 elif el[0].startswith('hrn'):
53 elif el[0].startswith('urn'):
55 elif el[0].startswith('type'):
57 elif el[0].startswith('recursive'):
62 elif el[0].startswith('display'):
65 results = sfa_client(request, method, hrn=hrn, urn=urn, object_type=object_type, recursive=recursive, options=options, platforms=platforms)
66 if display == 'table':
67 return render_to_response('table-default.html', {'data' : data, 'fields' : columns, 'id' : '@component_id', 'options' : None})
69 return HttpResponse(json.dumps(results, cls=MyEncoder), content_type="application/json")
71 def get_user_account(request, user_email, platform_name):
73 Returns the user configuration for a given platform.
74 This function does not resolve references.
76 user_query = Query().get('local:user').filter_by('email', '==', user_email).select('user_id')
77 user_details = execute_admin_query(request, user_query)
78 platform_query = Query().get('local:platform').filter_by('platform', '==', platform_name).select('platform_id')
79 platform_details = execute_admin_query(request, platform_query)
81 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')
82 accounts = execute_admin_query(request, account_query)
85 raise Exception, "this account does not exist"
87 if accounts[0]['auth_type'] == 'reference':
88 pf = json.loads(accounts[0]['config'])['reference_platform']
89 return get_user_account(request, user_email, pf)
93 def sfa_client(request, method, hrn=None, urn=None, object_type=None, rspec=None, recursive=None, options=None, platforms=None, admin=False):
95 Config = ConfigParser.ConfigParser()
96 monitor_file = os.path.abspath(os.path.dirname(__file__) + '/../myslice/monitor.ini')
97 Config.read(monitor_file)
100 user_email, admin_password = config.manifold_admin_user_password()
102 #logger.debug(request.session['user']['email'])
103 user_email = request.session['user']['email']
111 if object_type is None:
115 if recursive is None:
119 if platforms is None:
122 if method not in ['GetVersion','ListResources']:
125 hrn = urn_to_hrn(urn)
127 urn = hrn_to_urn(hrn, object_type)
130 raise Exception, "Provide urn OR hrn + type as parameters of method %s" % method
132 if len(platforms)==0:
133 platforms = get_platforms(request)
134 #platforms.append('myslice')
135 #results = {'method':method,'platforms':platforms,'rspec':rspec,'options':options}
142 api_options['list_leases'] = 'all'
145 platform = get_platform_config(request, pf)
146 if 'rspec_type' in platform and 'rspec_version' in platform:
147 api_options['geni_rspec_version'] = {'type': platform['rspec_type'],'version': platform['rspec_version']}
149 api_options['geni_rspec_version'] = {'type': 'GENI', 'version': '3'}
150 if 'sm' in platform and len(platform['sm']) > 0:
152 server_url = platform['sm']
153 if 'rm' in platform and len(platform['rm']) > 0:
155 server_url = platform['rm']
156 if 'registry' in platform and len(platform['registry']) > 0:
158 server_url = platform['registry']
160 if not Config.has_option('monitor', 'cert') :
161 #return HttpResponse(json.dumps({'error' : '-1'}), content_type="application/json")
162 return {'error' : '-1', 'msg': 'monitor.ini has no cert configured'}
164 cert = os.path.abspath(Config.get('monitor', 'cert'))
165 if not os.path.isfile(cert) :
166 #return HttpResponse(json.dumps({'error' : '-1'}), content_type="application/json")
167 return {'error' : '-1', 'msg': 'check cert file at %s'%cert}
169 if not Config.has_option('monitor', 'pkey') :
170 #return HttpResponse(json.dumps({'error' : '-2'}), content_type="application/json")
171 return {'error' : '-2'}
173 pkey = os.path.abspath(Config.get('monitor', 'pkey'))
174 if not os.path.isfile(pkey) :
175 #return HttpResponse(json.dumps({'error' : '-2'}), content_type="application/json")
176 return {'error' : '-2'}
178 server = SfaServerProxy(server_url, pkey, cert)
181 # Get user config from Manifold
182 user_config = get_user_config(request, user_email, pf)
183 if 'delegated_user_credential' in user_config:
184 user_cred = user_config['delegated_user_credential']
185 elif 'user_credential' in user_config:
186 user_cred = user_config['user_credential']
188 logger.error("no user credentials for user = ", user_email)
192 if 'delegated_%s_credentials'%object_type in user_config:
193 for obj_name, cred in user_config['delegated_%s_credentials'%object_type].items():
196 elif '%s_credentials'%object_type in user_config:
197 for obj_name, cred in user_config['%s_credentials'%object_type].items():
201 logger.error("no credentials for object")
202 logger.error(object_type)
203 logger.error(object_name)
207 if method == "GetVersion":
208 result = server.GetVersion()
212 if method == "ListResources":
213 result = server.ListResources([user_cred], api_options)
214 dict_result = xmltodict.parse(result['value'])
215 result['parsed'] = dict_result
216 if isinstance(dict_result['rspec']['node'], list):
217 columns.extend(dict_result['rspec']['node'][0].keys())
219 columns.extend(dict_result['rspec']['node'].keys())
221 elif method == "Describe":
222 version = server.GetVersion()
224 if version['geni_api'] == 2:
225 # ListResources(slice_hrn)
226 api_options['geni_slice_urn'] = urn
227 result = server.ListResources([object_cred], api_options)
228 dict_result = xmltodict.parse(result['value'])
229 # else GetVersion = v3
231 result = server.Describe([urn] ,[object_cred], api_options)
232 if isinstance(result, dict):
233 if result['value'] != 0:
234 dict_result = xmltodict.parse(result['value']['geni_rspec'])
236 result['parsed'] = dict_result
237 if 'rspec' in dict_result and 'node' in dict_result['rspec']:
238 if isinstance(dict_result['rspec']['node'], list):
239 columns.extend(dict_result['rspec']['node'][0].keys())
241 columns.extend(dict_result['rspec']['node'].keys())
243 elif method == 'Renew':
244 # Renew till 1 month from now
245 d = datetime.datetime.utcnow() + datetime.timedelta(365/12)
246 date = d.isoformat("T") + "Z"
247 result = server.Renew([urn] ,[object_cred], date, api_options)
248 elif method == 'Delete':
249 result = server.Delete([urn] ,[object_cred], api_options)
250 elif method == 'Allocate':
251 api_options['call_id'] = unique_call_id()
252 # List of users comes from the Registry
253 api_options['sfa_users'] = sfa_users
254 api_options['geni_users'] = geni_users
256 version = server.GetVersion()
257 if version['geni_api'] == 2:
258 result = server.CreateSliver([urn] ,[object_cred], rspec, api_options)
259 # else GetVersion = v3
261 result = server.Allocate([urn] ,[object_cred], rspec, api_options)
262 elif method == 'Provision':
264 # Nothing it is not supported by v2 AMs
265 version = server.GetVersion()
266 if version['geni_api'] == 3:
267 api_options['call_id'] = unique_call_id()
268 # List of users comes from the Registry
269 api_options['sfa_users'] = sfa_users
270 api_options['geni_users'] = geni_users
271 result = server.Provision([urn] ,[object_cred], api_options)
272 elif method == 'Status':
273 result = server.Status([urn] ,[object_cred], api_options)
274 elif method == 'PerformOperationalAction':
276 # Nothing it is not supported by v2 AMs
277 version = server.GetVersion()
278 if version['geni_api'] == 3:
279 result = server.PerformOperationalAction([urn] ,[object_cred], action, api_options)
280 elif method == 'Shutdown':
281 result = server.Shutdown(urn ,[object_cred], api_options)
283 #return HttpResponse(json.dumps({'error' : '-3','msg':'method not supported by AM'}), content_type="application/json")
284 logger.debug('method %s not handled by AM' % method)
289 record_dict = {'urn': urn, 'hrn': hrn, 'type': object_type}
292 api_options['recursive'] = recursive
293 result = server.List(hrn, user_cred, api_options)
295 result = filter_records(object_type, result)
296 elif method == "Resolve":
298 # details can be True or False
299 api_options['details']=True
300 result = server.Resolve(hrn, user_cred, api_options)
302 result = filter_records(object_type, result)
303 elif method == "Register":
304 # record_dict must be crafted
305 # auth_cred must be selected in the list of auth_creds from user's account
306 result = server.Register(record_dict, auth_cred)
307 elif method == "Update":
308 # record_dict must be crafted
309 # object_cred must be selected in the list of creds for the object type
310 # from user's account
311 result = server.Update(record_dict, object_cred)
312 elif method == "Remove":
314 # auth_cred must be selected in the list of auth_creds from user's account
315 # object_type is required
316 result = server.Remove(hrn, auth_cred, object_type)
318 #return HttpResponse(json.dumps({'error' : '-3','msg':'method not supported by Registry'}), content_type="application/json")
319 logger.debug('method %s not handled by Registry' % method)
324 if 'rspec' in dict_result and 'node' in dict_result['rspec']:
325 if isinstance(dict_result['rspec']['node'], list):
326 data = data + dict_result['rspec']['node']
328 data.append(dict_result['rspec']['node'])
331 logger.error(traceback.format_exc())
333 results[pf] = {'error':'-3', 'error_msg': str(e)}
335 results['columns'] = columns
338 def get_user_config(request, user_email, platform_name):
339 account = get_user_account(request, user_email, platform_name)
340 return json.loads(account['config']) if account['config'] else {}
342 def get_platforms(request):
344 platform_query = Query().get('local:platform').filter_by('gateway_type', '==', 'sfa').filter_by('disabled','==',0).select('platform')
345 platforms = execute_admin_query(request, platform_query)
348 ret.append(p['platform'])
351 def get_platform_config(request, platform_name):
352 platform_query = Query().get('local:platform').filter_by('platform', '==', platform_name).select('platform', 'config')
353 platforms = execute_admin_query(request, platform_query)
355 return json.loads(platforms[0]['config']) if platforms[0]['config'] else {}
357 def filter_records(type, records):
358 filtered_records = []
359 for record in records:
360 if (record['type'] == type) or (type == "all"):
361 filtered_records.append(record)
362 return filtered_records