004ac686eddf1ad184300a4062da83b818eb00bf
[myslice.git] / portal / actions.py
1 from django.http                 import HttpResponse
2 from manifold.core.query         import Query
3 from manifold.manifoldapi        import execute_query,execute_admin_query
4 from portal.models               import PendingUser, PendingSlice, PendingAuthority
5 import json
6
7 # Thierry: moving this right into the code so 
8 # most people can use myslice without having to install sfa
9 # XXX tmp sfa dependency, should be moved to SFA gateway
10 #from sfa.util.xrn                import Xrn 
11
12
13 # Get the list of authorities
14
15 def authority_get_pis(request, authority_hrn):
16     query = Query.get('authority').filter_by('authority_hrn', '==', authority_hrn).select('pi_users')
17     results = execute_query(request, query)
18     # NOTE: temporarily commented. Because results is giving empty list. 
19     # Needs more debugging
20     #if not results:
21     #    raise Exception, "Authority not found: %s" % authority_hrn
22     #result, = results
23     #return result['pi_users']
24     return results
25
26 def authority_get_pi_emails(request, authority_hrn):
27     return ['jordan.auge@lip6.fr', 'loic.baron@lip6.fr']
28
29     pi_users = authority_get_pis(request,authority_hrn)
30     pi_user_hrns = [ hrn for x in pi_users for hrn in x['pi_users'] ]
31     query = Query.get('user').filter_by('user_hrn', 'included', pi_user_hrns).select('email')
32     results = execute_query(request, query)
33     print "mails",  [result['email'] for result in results]
34     return [result['email'] for result in results]
35
36 # SFA add record (user, slice)
37
38 def sfa_add_user(request, user_params):
39     query = Query.create('user').set(user_params).select('user_hrn')
40     results = execute_query(request, query)
41     if not results:
42         raise Exception, "Could not create %s. Already exists ?" % user_params['hrn']
43     return results
44
45 def sfa_update_user(request, user_hrn, user_params):
46     # user_params: keys [public_key] 
47     query = Query.update('user').filter_by('user_hrn', '==', user_hrn).set(user_params).select('user_hrn')
48     results = execute_query(request,query)
49     return results
50
51 def sfa_add_slice(request, slice_params):
52     query = Query.create('slice').set(slice_params).select('slice_hrn')
53     results = execute_query(request, query)
54     if not results:
55         raise Exception, "Could not create %s. Already exists ?" % slice_params['hrn']
56     return results
57
58 def sfa_add_user_to_slice(request, user_hrn, slice_params):
59     query = Query.update('slice').filter_by('user_hrn', '==', user_hrn).set(slice_params).select('slice_hrn')
60     results = execute_query(request, query)
61     if not results:
62         raise Exception, "Could not create %s. Already exists ?" % slice_params['hrn']
63     return results
64
65 # Propose hrn
66
67 def manifold_add_user(request, user_params):
68     # user_params: email, password e.g., user_params = {'email':'aa@aa.com','password':'demo'}
69     query = Query.create('local:user').set(user_params).select('email')
70     results = execute_admin_query(request, query)
71     if not results:
72         raise Exception, "Failed creating manifold user: %s" % user_params['email']
73     result, = results
74     return result['email']
75
76 def manifold_update_user(request, email, user_params):
77     # user_params: password, config e.g., 
78     query = Query.update('local:user').filter_by('email', '==', email).set(user_params).select('email')
79     results = execute_admin_query(request,query)
80     # NOTE: results remains empty and goes to Exception. However, it updates the manifold DB.
81     # That's why I commented the exception part. -- Yasin 
82     #if not results:
83     #    raise Exception, "Failed updating manifold user: %s" % user_params['email']
84     #result, = results
85     return results
86
87 def manifold_add_account(request, account_params):
88     query = Query.create('local:account').set(account_params).select(['user', 'platform'])
89     results = execute_admin_query(request,query)
90     if not results:
91         raise Exception, "Failed creating manifold account on platform %s for user: %s" % (account_params['platform'], account_params['user'])
92     result, = results
93     return result['user_id']
94
95 def manifold_update_account(request,account_params):
96     # account_params: config
97     query = Query.update('local:account').filter_by('platform', '==', 'myslice').set(account_params).select('user_id')
98     results = execute_query(request,query)
99     # NOTE: results remains empty and goes to Exception. However, it updates the manifold DB.
100     # That's why I commented the exception part. -- Yasin 
101     #if not results:
102     #    raise Exception, "Failed updating manifold account: config %s" % account_params['config']
103     #result, = results
104     return results
105
106 #explicitly mention the platform_id
107 def manifold_delete_account(request, platform_id, account_params):
108     query = Query.delete('local:account').filter_by('platform_id', '==', platform_id).set(account_params).select('user_id')
109     results = execute_query(request,query)
110     return results
111
112
113 #not tested
114 def manifold_add_platform(request, platform_params):
115     query = Query.create('local:platform').set(platform_params).select(['user', 'platform'])
116     results = execute_admin_query(request,query)
117     if not results:
118         raise Exception, "Failed creating manifold  platform %s for user: %s" % (platform_params['platform'], platform_params['user'])
119     result, = results
120     return result['platform_id']
121
122
123 def make_request_user(user):
124     request = {}
125     request['type']          = 'user'
126     request['id']            = user.id
127     request['timestamp']     = user.created # XXX in DB ?
128     request['authority_hrn'] = user.authority_hrn
129     request['first_name']    = user.first_name
130     request['last_name']     = user.last_name
131     request['email']         = user.email
132     request['login']         = user.login
133     request['keypair']       = user.keypair
134     return request
135
136 def make_request_slice(slice):
137     request = {}
138     request['type'] = 'slice'
139     request['id'] = slice.id
140     request['user_email'] = slice.user_email
141     request['timestamp'] = slice.created
142     request['authority_hrn'] = slice.authority_hrn
143     request['slice_name'] = slice.slice_name
144     request['number_of_nodes'] = slice.number_of_nodes
145     request['type_of_nodes'] = slice.type_of_nodes
146     request['purpose'] = slice.purpose
147     return request
148
149 def make_request_authority(authority):
150     request = {}
151     request['type']                  = 'authority'
152     request['id']                    = authority.id
153     request['site_name']             = authority.site_name
154     request['site_latitude']         = authority.site_latitude
155     request['site_longitude']        = authority.site_longitude
156     request['site_url']              = authority.site_url
157     request['site_authority']        = authority.site_authority
158     request['site_abbreviated_name'] = authority.site_abbreviated_name
159     request['address_line1']         = authority.address_line1
160     request['address_line2']         = authority.address_line2
161     request['address_line3']         = authority.address_line3
162     request['address_city']          = authority.address_city
163     request['address_postalcode']    = authority.address_postalcode
164     request['address_state']         = authority.address_state
165     request['address_country']       = authority.address_country
166     request['authority_hrn']         = authority.authority_hrn
167     request['timestamp']             = authority.created
168     return request
169
170 def make_requests(pending_users, pending_slices, pending_authorities):
171     requests = []
172     for user in pending_users:
173         requests.append(make_request_user(user))
174     for slice in pending_slices:
175         requests.append(make_request_slice(slice))
176     for authority in pending_authorities:
177         requests.append(make_request_authority(authority))
178     return requests   
179
180 def get_request_by_id(ids):
181     sorted_ids = { 'user': [], 'slice': [], 'authority': [] }
182     for type__id in ids:
183         type, id = type__id.split('__')
184         sorted_ids[type].append(id)
185         
186     if not ids:
187         pending_users  = PendingUser.objects.all()
188         pending_slices = PendingSlice.objects.all()
189         pending_authorities = PendingAuthority.objects.all()
190     else:
191         pending_users  = PendingUser.objects.filter(id__in=sorted_ids['user']).all()
192         pending_slices = PendingSlice.objects.filter(id__in=sorted_ids['slice']).all()
193         pending_authorities = PendingAuthority.objects.filter(id__in=sorted_ids['authority']).all()
194
195     return make_requests(pending_users, pending_slices, pending_authorities)
196
197 def get_request_by_authority(authority_hrns):
198     print "get_request_by_authority auth_hrns = ", authority_hrns
199     if not authority_hrns:
200         pending_users  = PendingUser.objects.all()
201         pending_slices = PendingSlice.objects.all()
202         pending_authorities = PendingAuthority.objects.all()
203     else:
204         pending_users  = PendingUser.objects.filter(authority_hrn__in=authority_hrns).all()
205         pending_slices = PendingSlice.objects.filter(authority_hrn__in=authority_hrns).all()
206         pending_authorities = PendingAuthority.objects.filter(authority_hrn__in=authority_hrns).all()
207
208     return make_requests(pending_users, pending_slices, pending_authorities)
209     
210 # XXX Is it in sync with the form fields ?
211
212 def portal_validate_request(wsgi_request, request_ids):
213     status = {}
214
215     if not isinstance(request_ids, list):
216         request_ids = [request_ids]
217
218     requests = get_request_by_id(request_ids)
219     for request in requests:
220         # type, id, timestamp, details, allowed -- MISSING: authority_hrn
221         # CAREFUL about details
222         # user  : first name, last name, email, password, keypair
223         # slice : number of nodes, type of nodes, purpose
224         
225         request_status = {}
226
227         print "REQUEST", request
228         if request['type'] == 'user':
229
230             try:
231                 # XXX tmp user_hrn inside the keypair column of pendiguser table
232                 hrn = json.loads(request['keypair'])['user_hrn']
233                 #hrn = "%s.%s" % (request['authority_hrn'], request['login'])
234                 # XXX tmp sfa dependency
235                 from sfa.util.xrn import Xrn 
236                 urn = Xrn(hrn, request['type']).get_urn()
237
238                 sfa_user_params = {
239                     'hrn'        : hrn, 
240                     'urn'        : urn,
241                     'type'       : request['type'],
242                     'keys'       : [json.loads(request['keypair'])['user_public_key']],
243                     'first_name' : request['first_name'],
244                     'last_name'  : request['last_name'],
245                     'email'      : request['email'],
246                     #'slices'    : None,
247                     #'researcher': None,
248                     #'pi'        : None,
249                     'enabled'    : True
250                 }
251                 # ignored in request: id, timestamp, password
252
253                 sfa_add_user(wsgi_request, sfa_user_params)
254
255                 # XXX Remove from database
256
257
258                 request_status['SFA user'] = {'status': True }
259
260             except Exception, e:
261                 request_status['SFA user'] = {'status': False, 'description': str(e)}
262
263             # MANIFOLD user should be added beforehand, during registration
264             #try:
265             #    manifold_user_params = { key: request[key] for key in MANIFOLD_USER_KEYS }
266             #    # XXX # manifold_add_user(manifold_user_params)
267             #    request_status['MySlice user'] = {'status': True }
268             #except Exception, e:
269             #    request_status['MySlice user'] = {'status': False, 'description': str(e)}
270
271             # XXX
272             #manifold_account_params = { key: request[key] for key in MANIFOLD_ACCOUNT_KEYS }
273             #manifold_add_account(manifold_account_params)
274             #request_status['MySlice testbed accounts'] = {'status': False }
275
276         elif request['type'] == 'slice':
277             try:
278                 hrn = "%s.%s" % (request['authority_hrn'], request['slice_name'])
279                 # XXX tmp sfa dependency
280                 from sfa.util.xrn import Xrn 
281                 urn = Xrn(hrn, request['type']).get_urn()
282
283                 sfa_slice_params = {
284                     'hrn'        : hrn, 
285                     'urn'        : urn,
286                     'type'       : request['type'],
287                     #'slices'    : None,
288                     #'researcher': None,
289                     #'pi'        : None,
290                     'enabled'    : True
291                 }
292                 # ignored in request: id, timestamp,  number_of_nodes, type_of_nodes, purpose
293
294                 sfa_add_slice(wsgi_request, sfa_slice_params)
295                 #sfa_add_user_to_slice(wsgi_request, user_hrn, sfa_slice_params)
296
297                 # XXX Remove from database
298
299             
300                 request_status['SFA slice'] = {'status': True }
301
302             except Exception, e:
303                 request_status['SFA slice'] = {'status': False, 'description': str(e)}
304
305         status['%s__%s' % (request['type'], request['id'])] = request_status
306
307     return status
308
309
310 def validate_action(request, **kwargs):
311     ids = filter(None, kwargs['id'].split('/'))
312     status = portal_validate_request(request, ids)
313     json_answer = json.dumps(status)
314     return HttpResponse (json_answer, mimetype="application/json")
315
316 # Django and ajax
317 # http://djangosnippets.org/snippets/942/