Upgrade/downgrade PI

[unfold.git] / portal / actions.py

diff --git a/portal/actions.py b/portal/actions.py
index e8a4d01..38e70e8 100644 (file)
--- a/portal/actions.py
+++ b/portal/actions.py
@@ -40,6 +40,34 @@ def authority_get_pis(request, authority_hrn):
     #return result['pi_users']
     return results
 
+
+def authority_add_pis(request, authority_hrn,user_hrn):
+
+    # getting pis of the authority of the user
+    pis = authority_get_pis (request, authority_hrn)
+    for pi in pis:
+        pi_list = pi['pi_users']
+   
+    updated_pi_list = pi_list.append(user_hrn) 
+    query = Query.update('authority').filter_by('authority_hrn', '==', authority_hrn).set({'pi_users':pi_list})
+    results = execute_admin_query(request,query)
+    newpis = authority_get_pis (request, authority_hrn)
+    return newpis
+
+def authority_remove_pis(request, authority_hrn,user_hrn):
+
+    # getting pis of the authority of the user
+    pis = authority_get_pis (request, authority_hrn)
+    for pi in pis:
+        pi_list = pi['pi_users']
+ 
+    updated_pi_list = pi_list.remove(user_hrn) 
+    query = Query.update('authority').filter_by('authority_hrn', '==', authority_hrn).set({'pi_users':pi_list})
+    results = execute_admin_query(request,query)
+    newpis = authority_get_pis (request, authority_hrn)
+    return newpis
+
+
 def authority_get_pi_emails(request, authority_hrn):
     pi_users = authority_get_pis(request,authority_hrn)
     print "pi_users = %s" % pi_users