Pi check in each view to make the topmenu show correctly

[myslice.git] / portal / homeview.py

diff --git a/portal/homeview.py b/portal/homeview.py
index 80e9347..ce12ee2 100644 (file)
--- a/portal/homeview.py
+++ b/portal/homeview.py
@@ -100,7 +100,26 @@ class HomeView (FreeAccessView, ThemeView):
 
     def get (self, request, state=None):
         env = self.default_env()
+        ## check user is pi or not
+        platform_query  = Query().get('local:platform').select('platform_id','platform','gateway_type','disabled')
+        account_query  = Query().get('local:account').select('user_id','platform_id','auth_type','config')
+        platform_details = execute_query(self.request, platform_query)
+        account_details = execute_query(self.request, account_query)
+        for platform_detail in platform_details:
+            for account_detail in account_details:
+                if platform_detail['platform_id'] == account_detail['platform_id']:
+                    if 'config' in account_detail and account_detail['config'] is not '':
+                        account_config = json.loads(account_detail['config'])
+                        if 'myslice' in platform_detail['platform']:
+                            acc_auth_cred = account_config.get('delegated_authority_credentials','N/A')
+        # assigning values
+        if acc_auth_cred == {}:
+            pi = "is_not_pi"
+        else:
+            pi = "is_pi"
 
+ 
+        env['pi'] = pi
         if request.user.is_authenticated(): 
             env['person'] = self.request.user
         else: