add roles in login get method

24552ea0 · haowang · 34f24c51 · 24552ea0
Commit 24552ea0 authored Apr 26, 2019 by haowang
Show whitespace changes
Inline Side-by-side

Showing with 3 additions and 0 deletions

account.py api/account.py +3 -0

No files found.
--- a/api/account.py
+++ b/api/account.py
@@ -75,6 +75,9 @@ class LoginView(APIView):
                'avatar': settings.AVATAR
            })
            if data['is_staff']:
+                if data.get('staff_level'):
+                    data.update({'roles': ['mbhoperator']})
+                else:    
                    data.update({'roles': ['staff']})
            else:
                data.update({'roles': ['anonymous']})