Throw 500 error, when grant some roles to some users OpenBankProject#83

PengfeiLi0218 · Feb 20, 2019 · fa61cfa · fa61cfa
1 parent fd3971f
commit fa61cfa
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 7 deletions.
diff --git a/apimanager/obp/directlogin.py b/apimanager/obp/directlogin.py
@@ -39,7 +39,7 @@ def login_to_api(self, data):
 
         result = response.json()
         if response.status_code != 201:
-            raise AuthenticatorError(result['error'])
+            raise AuthenticatorError(result['message'])
         else:
             self.token = result['token']
 

diff --git a/apimanager/users/views.py b/apimanager/users/views.py
@@ -151,12 +151,15 @@ def form_valid(self, form):
         except:
             messages.error(self.request, 'Unknown Error')
             return super(DetailView, self).form_invalid(form)
-
-        msg = 'Entitlement with role {} has been added.'.format(
-            entitlement['role_name'])
-        messages.success(self.request, msg)
-        self.success_url = self.request.path
-        return super(DetailView, self).form_valid(form)
+        if 'code' in entitlement and entitlement['code']>=400:
+            messages.error(self.request, entitlement[''])
+            return super(DetailView, self).form_invalid(form)
+        else:
+            msg = 'Entitlement with role {} has been added.'.format(
+                entitlement['role_name'])
+            messages.success(self.request, msg)
+            self.success_url = self.request.path
+            return super(DetailView, self).form_valid(form)
 
     def get_context_data(self, **kwargs):
         context = super(DetailView, self).get_context_data(**kwargs)