ASJServer/Controller/SysMsg.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
@Copyright (C) ansjer cop Video Technology Co.,Ltd.All rights reserved.
@AUTHOR: ASJRD018
@NAME: AnsjerFormal
@software: PyCharm
@DATE: 2019/5/18 10:45
@Version: python3.6
@MODIFY DECORD:ansjer dev
@file: SysMsg.py
@Contact: chanjunkai@163.com
"""
import time
import math
from django.views.generic.base import View

from Model.models import SysMsgModel, FeedBackModel, UserExtendModel
from Object.ResponseObject import ResponseObject
from Object.TokenObject import TokenObject
from Service.ModelService import ModelService
from django.db.models import Q


class SysMsgView(View):
    def get(self, request, *args, **kwargs):
        request.encoding = 'utf-8'
        operation = kwargs.get('operation', None)
        return self.validation(request.GET, operation)

    def post(self, request, *args, **kwargs):
        request.encoding = 'utf-8'
        operation = kwargs.get('operation', None)
        return self.validation(request.POST, operation)

    def validation(self, request_dict, operation):
        response = ResponseObject()
        token = request_dict.get('token', None)
        tko = TokenObject(token)
        if tko.code == 0:
            userID = tko.userID
            if operation == 'pushFeedBack':
                return self.do_push_msg_feed_back(userID, request_dict, response)
            elif operation == 'query':
                return self.do_query(userID, request_dict, response)
            elif operation == 'queryByAdmin':
                return self.do_query_by_admin(userID, request_dict, response)
            elif operation == 'update':
                return self.do_update(userID, request_dict, response)
            elif operation == 'delete':
                return self.do_delete(userID, request_dict, response)
            elif operation == 'deleteByAdmin':
                return self.do_delete_by_admin(userID, request_dict, response)
            elif operation == 'massPush':
                return self.do_mass_push(userID, request_dict, response)
            else:
                return response.json(414)
        else:
            return response.json(tko.code)

    def do_mass_push(self, userID, request_dict, response):
        own_perm = ModelService.check_perm(userID, 10)
        if own_perm:
            app_list = request_dict.get('app_list', None)
            msg = request_dict.get('msg', None)
            region = request_dict.get('region', None)
            app_list = app_list.split(',')
            region = region.split(',')
            nowTime = int(time.time())
            ue_filter = UserExtendModel.objects.filter()
            if app_list:
                ue_filter = ue_filter.filter(appBundleId__in=app_list)
            if region:
                ue_filter.filter(userID__language__in=region)
            ue_count = ue_filter.count()
            range_count = math.ceil(ue_count / 5000)
            for i in range(range_count):
                querysetlist = []
                userID_list = ue_filter.values_list('userID_id', flat=True)
                for urd in userID_list:
                    querysetlist.append(SysMsgModel(userID_id=urd, msg=msg, addTime=nowTime, updTime=nowTime))
                SysMsgModel.objects.bulk_create(querysetlist)
            return response.json(0)
        else:
            return response.json(404)

    def do_push_msg_feed_back(self, userID, request_dict, response):
        own_perm = ModelService.check_perm(userID, 10)
        if own_perm:
            fid = request_dict.get('fid', None)
            msg = request_dict.get('msg', None)
            if all([fid, msg]):
                nowTime = int(time.time())
                fb_qs = FeedBackModel.objects.filter(id=fid)
                fb_list = fb_qs.values('userID_id')
                if fb_qs.exists():
                    create_data = {'userID_id': fb_list[0]['userID_id'], 'msg': msg, 'addTime': nowTime,
                                   'updTime': nowTime}
                    try:
                        SysMsgModel.objects.create(**create_data)
                        fb_qs.update(status=1)
                    except Exception as e:
                        return response.json(500, repr(e))
                    else:
                        return response.json(0, {'status': 1, 'updTime': nowTime})
                else:
                    return response.json(173)
            else:
                return response.json(444, 'fid,msg')
        else:
            return response.json(404)

    def do_query(self, userID, request_dict, response):
        page = request_dict.get('page', None)
        line = request_dict.get('line', None)
        if page and line:
            page = int(page)
            line = int(line)
            sm_qs = SysMsgModel.objects.filter(userID_id=userID)
            count = sm_qs.count()
            sm_qs = sm_qs[(page - 1) * line:page * line]. \
                values('status', 'id', 'msg', 'addTime', 'updTime')
            # sm_qs = SysMsgModel.objects.filter(userID_id=userID)[(page - 1) * line:page * line].values('sta')
            if sm_qs.exists():
                return response.json(0, {'data': list(sm_qs), 'count': count})
            else:
                return response.json(0, [])
        else:
            return response.json(444, 'page,line')

    def do_query_by_admin(self, userID, request_dict, response):
        own_perm = ModelService.check_perm(userID, 30)
        if own_perm:
            page = request_dict.get('page', None)
            line = request_dict.get('line', None)
            username = request_dict.get('username', None)
            if page and line:
                page = int(page)
                line = int(line)
                sm_qs = SysMsgModel.objects.filter()
                if username:
                    userID_list = ModelService.get_user_list_by_username(username)
                    sm_qs = sm_qs.filter(userID_id__in=userID_list)
                count = sm_qs.count()
                sm_qs = sm_qs[(page - 1) * line:page * line]. \
                    values('status', 'id', 'msg', 'addTime', 'updTime', 'userID__username', 'userID__userEmail',
                           'userID__phone')
                for k, v in enumerate(sm_qs):

                    if sm_qs[k]['userID__username'] == '':
                        sm_qs[k]['userID__username'] = sm_qs[k]['userID__userEmail']
                    if sm_qs[k]['userID__username'] == '':
                        sm_qs[k]['userID__username'] = sm_qs[k]['userID__phone']
                    sm_qs[k].pop('userID__userEmail')
                    sm_qs[k].pop('userID__phone')
                return response.json(0, {'datas': list(sm_qs), 'count': count})
            else:
                return response.json(444, 'page,line')
        else:
            return response.json(404)

    def do_update(self, userID, request_dict, response):
        sid = request_dict.get('sid', None)
        if sid:
            sm_qs = SysMsgModel.objects.filter(userID_id=userID, id=sid)
            if sm_qs.exists():
                sm_qs.update(status=1)
                return response.json(0)
            else:
                return response.json(173)
        else:
            sm_qs = SysMsgModel.objects.filter(userID_id=userID).update(status=1)
            return response.json(0)

    def do_delete(self, userID, request_dict, response):
        sid = request_dict.get('sid', None)
        if sid:
            sm_qs = SysMsgModel.objects.filter(userID_id=userID, id=sid)
            if sm_qs.exists():
                sm_qs.delete()
                return response.json(0)
            else:
                return response.json(173)
        else:
            sm_qs = SysMsgModel.objects.filter(userID_id=userID)
            if sm_qs.exists():
                sm_qs.delete()
                return response.json(0)
            else:
                return response.json(173)

    def do_delete_by_admin(self, userID, request_dict, response):
        own_perm = ModelService.check_perm(userID, 10)
        if not own_perm:
            return response.json(404)
        sid = request_dict.get('sid', None)
        if sid:
            sm_qs = SysMsgModel.objects.filter(id=sid)
            if sm_qs.exists():
                sm_qs.delete()
                return response.json(0)
            else:
                return response.json(173)
        else:
            return response.json(444, 'sid')