1 from django.http import HttpResponse
2 from manifold.core.query import Query
3 from manifold.manifoldapi import execute_query
4 from portal.models import PendingUser, PendingSlice
7 # Get the list of authorities
9 def authority_get_pis(authority_hrn):
10 query = Query.get('authority').filter_by('authority_hrn', '==', authority_hrn).select('pi_users')
11 results = execute_query(query)
13 raise Exception, "Authority not found: %s" % authority_hrn
15 return result['pi_users']
17 def authority_get_pi_emails(authority_hrn):
18 user_hrns = authority_get_pis(authority_hrn)
20 query = Query.get('user').filter_by('user_hrn', 'included', user_hrns).select('user_email')
21 results = execute_query(query)
23 return [result['user_email'] for result in results]
25 # SFA add record (user, slice)
27 def sfa_add_user(user_params):
28 # sfi.py add --xrn=fed4fire.upmc.timur_friedman --type=user --key=/root/.sfi/timur.pub --email=timur.friedman@lip6.fr --extra=first_name=Timur --extra=last_name=Friedman --extra=enabled=true
29 # user_params: xrn type key email + first_name last_name enabled
30 query = Query.create('user').set(user_params).select('user_hrn')
31 results = execute_query(query)
33 raise Exception, "Failed creating SFA user: %s" % user_params['user_hrn']
35 return result['user_hrn']
37 def sfa_add_slice(slice_params):
42 def manifold_add_user(user_params):
43 # user_params: email, password
44 query = Query.create('local:user').set(user_params).select('email')
45 results = execute_query(query)
47 raise Exception, "Failed creating manifold user: %s" % user_params['email']
49 return result['email']
51 def manifold_add_account(account_params):
52 query = Query.create('local:account').set(account_params).select(['user', 'platform'])
53 results = execute_query(query)
55 raise Exception, "Failed creating manifold account on platform %s for user: %s" % (account_params['platform'], account_params['user'])
57 return (result['user'], result['platform'])
59 def make_request_user(user):
61 request['type'] = 'user'
62 request['id'] = user.id
63 request['timestamp'] = 'TODO' # XXX in DB ?
64 request['authority_hrn'] = user.authority_hrn
65 request['first_name'] = user.first_name
66 request['last_name'] = user.last_name
67 request['email'] = user.email
70 def make_request_slice(slice):
72 request['type'] = 'slice'
73 request['id'] = slice.id
74 request['timestamp'] = 'TODO' # XXX in DB ?
75 request['authority_hrn'] = slice.authority_hrn
76 request['number_of_nodes'] = slice.number_of_nodes
77 request['type_of_nodes'] = slice.type_of_nodes
78 request['purpose'] = slice.purpose
81 def make_requests(pending_users, pending_slices):
82 print "pending users =", pending_users
83 print "pending slices =", pending_slices
86 for user in pending_users:
87 requests.append(make_request_user(user))
88 for slice in pending_slices:
89 requests.append(make_request_slice(slice))
92 def get_request_by_id(ids):
93 sorted_ids = { 'user': [], 'slice': [] }
95 type, id = type__id.split('__')
96 sorted_ids[type].append(id)
99 pending_users = PendingUser.objects.all()
100 pending_slices = PendingSlice.objects.all()
102 pending_users = PendingUser.objects.filter(id__in=sorted_ids['user']).all()
103 pending_slices = PendingSlice.objects.filter(id__in=sorted_ids['slice']).all()
105 return make_requests(pending_users, pending_slices)
107 def get_request_by_authority(authority_hrns):
108 if not authority_hrns:
109 pending_users = PendingUser.objects.all()
110 pending_slices = PendingSlice.objects.all()
112 pending_users = PendingUser.objects.filter(authority_hrn__in=authority_hrns).all()
113 pending_slices = PendingSlice.objects.filter(authority_hrn__in=authority_hrns).all()
115 return make_requests(pending_users, pending_slices)
117 SFA_USER_KEYS = ['xrn', 'type', 'key', 'first_name', 'last_name', 'email']
119 MANIFOLD_USER_KEYS = ['email', 'password']
120 MANIFOLD_ACCOUNT_KEYS = []
122 def portal_validate_request(request_ids):
125 if not isinstance(request_ids, list):
126 request_ids = [request_ids]
128 requests = get_request_by_id(request_ids)
129 for request in requests:
130 # type, id, timestamp, details, allowed -- MISSING: authority_hrn
131 # CAREFUL about details
132 # user : first name, last name, email, password, keypair
133 # slice : number of nodes, type of nodes, purpose
137 if request['type'] == 'user':
139 sfa_user_params = { key: request[key] for key in SFA_USER_KEYS }
140 sfa_user_params['enabled'] = True
141 # XXX # sfa_add_user(sfa_user_params)
142 request_status['SFA user'] = {'status': True }
144 request_status['SFA user'] = {'status': False, 'description': str(e)}
147 manifold_user_params = { key: request[key] for key in MANIFOLD_USER_KEYS }
148 # XXX # manifold_add_user(manifold_user_params)
149 request_status['MySlice user'] = {'status': True }
151 request_status['MySlice user'] = {'status': False, 'description': str(e)}
154 #manifold_account_params = { key: request[key] for key in MANIFOLD_ACCOUNT_KEYS }
155 #manifold_add_account(manifold_account_params)
156 request_status['MySlice testbed accounts'] = {'status': False }
158 elif request['type'] == 'slice':
160 sfa_slice_params = { key: request[key] for key in SFA_SLICE_KEYS }
161 # XXX # sfa_add_slice(sfa_slice_params)
162 request_status['SFA slice'] = {'status': True }
164 request_status['SFA slice'] = {'status': False, 'description': str(e)}
166 status['%s__%s' % (request['type'], request['id'])] = request_status
168 # XXX remove from database succeeded actions
173 def validate_action(*args, **kwargs):
174 ids = filter(None, kwargs['id'].split('/'))
175 status = portal_validate_request(ids)
176 json_answer = json.dumps(status)
177 return HttpResponse (json_answer, mimetype="application/json")
180 # http://djangosnippets.org/snippets/942/