add roles in login get method

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

Showing with 4 additions and 1 deletion

account.py api/account.py +4 -1

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