首页数据隔离

src/inspect_report/api/dataapi.py

@@ -6,7 +6,7 @@ from rest_framework.request import Request
 from inspect_report.models import Tasks, CheckSession, RulesStat, SeatStat, ScoreStat, Round
 import json
 from datetime import datetime, timedelta
-from config.config import TABLE_PRE
+from config.config import TABLE_PRE, name_list
 from django.http import HttpResponse
 import logging
 from pypinyin import Style, pinyin
@@ -14,6 +14,7 @@ import csv
 import codecs
 from django.utils.http import urlquote
 
+from inspect_report.utils.account import get_account_info
 
 logger = logging.getLogger(__name__)
 
@@ -23,12 +24,14 @@ class DataApi(viewsets.ViewSet):
     permission_classes = ()
 
     @action(['get'], detail=False)
+    @get_account_info
     def static_score_export(self, req: Request):
         """
         概览-地市得分统计-导出
         :param req:
         :return:
         """
+        username = req.data.get('username', '')
         task_id = req.GET.get('task', '')
         sort = req.GET.get('sort', '-avg_score')
         start_date = req.GET.get('start_date', (datetime.now() + timedelta(days=-1)).strftime('%Y-%m-%d'))
@@ -38,6 +41,8 @@ class DataApi(viewsets.ViewSet):
         task_condition = {'create_date__gte': q_start_date, 'create_date__lt': q_end_date}
         if task_id:
             task_condition['task'] = task_id
+        if username in name_list:
+            task_condition['task'] = username
         rules = ScoreStat.objects.filter(**task_condition).extra(select={'area': "task"}) \
             .values('area').annotate(avg_score_svc=Round(Avg('service_score'), 2),
                                      avg_score_bus=Round(Avg('business_score'), 2),

src/inspect_report/api/tasksapi.py

@@ -440,18 +440,22 @@ class TasksApi(viewsets.ViewSet):
         return Response({'code': 0, 'msg': 'success', 'count': total_count, 'data': seats.object_list})
 
     @action(['post'], detail=False)
+    @get_account_info
     def static_rule(self, req: Request):
         """
         首页概述-违规项统计
         :param req:
         :return:
         """
+        username = req.data.get('username', '')
         task_id = req.data.get('task', '')
         start_date = req.data.get('start_date', (datetime.now() + timedelta(days=-1)).strftime('%Y-%m-%d'))
         end_date = req.data.get('end_date', datetime.now().strftime('%Y-%m-%d'))
         task_condition = {'hasCheck': 1, 'createdAt__gte': start_date, 'createdAt__lt': end_date + ' 23:59:59'}
         if task_id:
             task_condition['name__startswith'] = task_id
+        if username in name_list:
+            task_condition['name__startswith'] = username
         tasks = Tasks.objects.filter(**task_condition).values('id', 'sessionCollectionId')
         task_dict = {}
         for t in tasks:
@@ -493,7 +497,7 @@ class TasksApi(viewsets.ViewSet):
             v['call_count'] = call_count
             data_sort.append(v)
         # data_sort = sorted(data_sort, key=lambda x: x['count'], reverse=True)
-        return Response({'code': 0, 'msg': 'success', 'data': data_sort})
+        return Response({'code': 0, 'msg': 'success', 'data': data_sort, 'username': username})
 
     @action(['post'], detail=False)
     def rule_detail(self, req: Request):

src/inspect_report/templates/inspect/group.html

@@ -87,7 +87,7 @@
               end-placeholder="结束日期"
               :default-time="['00:00:00', '23:59:59']">
             </el-date-picker>
-            <el-select v-model="taskvalue" filterable placeholder="请选择" @change="changeTask">
+            <el-select v-if="sel_is_show == true" v-model="taskvalue" filterable placeholder="请选择" @change="changeTask">
               <el-option
                 v-for="item in taskList"
                 :key="item.id"

src/inspect_report/utils/account.py

@@ -38,7 +38,12 @@ def get_account_info(func):
         cookie_jar = transfer_cookie(req.COOKIES)
         info = requests.get(USER_INFO_URL, cookies=cookie_jar)
         # logger.info('req中的query_params为: ', req.query_params)
-        if info.json()['code'] != 200:
+        try:
+            if info.json()['code'] != 200:
+                logger.info('未能获取用户信息: ' + info.text)
+                return Response({'code': -1, 'msg': '未能获取用户信息'})
+        except Exception:
+            logger.info('未能获取用户信息: ' + info.text)
             return Response({'code': -1, 'msg': '未能获取用户信息'})
         req.data['username'] = info.json()['data']['userInfo']['name']
         return func(obj, req, *args, **kw)

static/inspect/js/group.js

@@ -15,7 +15,7 @@ new Vue({
       poppagesize: 10,
       poptotal: 10,
       poprule: '',
-      poptables: '', 
+      poptables: '',
       popcurrentPage: 1,
       isHidePage: true,
       scrollTop: 0,
@@ -31,10 +31,11 @@ new Vue({
 			dateRange:[new Date(), new Date()],
       takevalue: '',
       titleInfo: {
-        "total_session": 0, 
-        "validate_session": 0, 
-        "ratio": 0, 
+        "total_session": 0,
+        "validate_session": 0,
+        "ratio": 0,
       },
+            sel_is_show: true,
 			loading: true,
       poptableData: [],
       tableData: []
@@ -135,7 +136,7 @@ new Vue({
     },
     popCurrentChange(page){
       this.poppage = page;
-      this.getDetail(this.taskvalue,this.start_date,this.end_date,this.poptables,this.poprule, this.poppage, this.poppagesize); 
+      this.getDetail(this.taskvalue,this.start_date,this.end_date,this.poptables,this.poprule, this.poppage, this.poppagesize);
     },
 	getRule(task,start_date,end_date,agentName,page,page_size){
       let that = this;
@@ -183,6 +184,12 @@ new Vue({
 					return false;
 				}
 				that.titleInfo = data.data;
+				if (['合肥', '芜湖', '淮北', '安庆', '黄山', '滁州', '阜阳', '亳州', '宿州', '六安', '宣城', '巢湖', '池州', '淮南', '安徽省营业部', '蚌埠', '马鞍山', '铜陵'].indexOf(data.username) !== -1) {
+				    that.sel_is_show = false;
+				    that.taskvalue = data.username;
+                } else {
+				    that.sel_is_show = true;
+                }
 				//this.$loading().close();
 			})
 		},
@@ -191,7 +198,7 @@ new Vue({
 		},
     rowDetail(index,tableData){
       let data = tableData;
-      console.log(data); 
+      console.log(data);
 
       //this.poptableData = tableData[index].details;
       this.poptables = data.tables;
@@ -199,7 +206,7 @@ new Vue({
       this.popcurrentPage = 1;
       this.poppage = 1;
       this.taskvalue = tableData.area;
-      this.getDetail(this.taskvalue,this.start_date,this.end_date,this.poptables,this.poprule, this.poppage, this.poppagesize); 
+      this.getDetail(this.taskvalue,this.start_date,this.end_date,this.poptables,this.poprule, this.poppage, this.poppagesize);
       this.dialogTableVisible = true;
 
     },