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

from django.views.generic.base import View

from Model.models import SysMsgModel, FeedBackModel
from Object.ResponseObject import ResponseObject
from Object.TokenObject import TokenObject
from Service.ModelService import ModelService


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)
            else:
                return response.json(414)
        else:
            return response.json(tko.code)

    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')

                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')