init:3528

@@ -9,6 +9,7 @@ from settings import conf
 from django.urls import reverse
 from django.urls import reverse
 from django.http import HttpResponseRedirect
 from django.http import HttpResponseRedirect
 from django.contrib.auth import login as auth_login
 from django.contrib.auth import login as auth_login
+from apps.doc.models import UserRole
 # from django.conf import settings
 # from django.conf import settings
 # from django.shortcuts import resolve_url, redirect
 # from django.shortcuts import resolve_url, redirect
 
 
@@ -46,10 +47,12 @@ class LoginView(ObtainJSONWebToken, GenericView):
         serializer = self.get_serializer(data=request.data)
         serializer = self.get_serializer(data=request.data)
         serializer.is_valid()
         serializer.is_valid()
         user = serializer.object.get('user')
         user = serializer.object.get('user')
+        user_role = UserRole.objects.filter(auth_user_id=user.id).first()
         data = {
         data = {
             'user_id': user.id,
             'user_id': user.id,
             'user_name': user.username,
             'user_name': user.username,
             'token': res.data.get('token'),
             'token': res.data.get('token'),
+            'role':  user_role.role if user_role else -1
         }
         }
         rh.set_token(res.data.get('token')[-10:], user.username)
         rh.set_token(res.data.get('token')[-10:], user.username)
         return response.ok(data=data)
         return response.ok(data=data)
@@ -83,6 +86,8 @@ class IWALoginView(IWABaseView, GenericView):
 
 
         if is_valid:
         if is_valid:
             rh.set_token(data.get('token')[-10:], data.get('user_name'))
             rh.set_token(data.get('token')[-10:], data.get('user_name'))
+            user_role = UserRole.objects.filter(auth_user_id=data.get('user_id')).first()
+            data['role'] = user_role.role if user_role else -1
             return response.ok(data=data)
             return response.ok(data=data)
         else:
         else:
             self.no_permission(data)
             self.no_permission(data)

@@ -1099,3 +1099,12 @@ class HILCmsStatusInfo(models.Model):
     class Meta:
     class Meta:
         managed = False
         managed = False
         db_table = 'hil_cms_status_info'
         db_table = 'hil_cms_status_info'
+class UserRole(models.Model):
+    id = models.AutoField(primary_key=True, verbose_name="id")  # 主键
+    auth_user_id = models.SmallIntegerField(null=False, default=0, verbose_name="auth_user_id")
+    role = models.SmallIntegerField(null=False, default=0, verbose_name="role")
+    menu_list = models.CharField(max_length=64, verbose_name="menu_list")
+
+    class Meta:
+        managed = False
+        db_table = 'user_role'
\ No newline at end of file