ASJServer/Controller/FeedBack.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/1/18 10:45
@Version: python3.6
@MODIFY DECORD:ansjer dev
@file: SysMsg.py
@Contact: chanjunkai@163.com
"""

import time
import json
import oss2
from django.views.generic.base import View
from var_dump import var_dump
from Ansjer.config import OSS_STS_ACCESS_KEY, OSS_STS_ACCESS_SECRET
from Model.models import FeedBackModel, StatResModel
from Object.ResponseObject import ResponseObject
from Object.TokenObject import TokenObject
from Service.CommonService import CommonService
from Service.ModelService import ModelService


class FeedBackView(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 == 'add':
                return self.do_add(userID, request_dict, response)
            elif operation == 'getUploadUrl':
                return self.do_get_upload_url(request_dict, response)
            elif operation == 'query':
                return self.do_query(userID, request_dict, response)
            elif operation == 'delete':
                return self.do_delete(userID, request_dict, response)
            elif operation == 'queryByAdmin':
                return self.do_query_by_admin(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_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)
        else:
            return response.json(444, 'page,line')
        fb_qs = FeedBackModel.objects.filter(userID_id=userID)[(page - 1) * line:page * line].values('id', 'status',
                                                                                                     'content',
                                                                                                     'addTime',
                                                                                                     'updTime',
                                                                                                     'FS__name')
        res = {}
        auth = oss2.Auth(OSS_STS_ACCESS_KEY, OSS_STS_ACCESS_SECRET)
        bucket = oss2.Bucket(auth, 'oss-cn-hongkong.aliyuncs.com', 'statres')
        for fb in fb_qs:
            if fb['id'] in res:
                if fb['FS__name']:
                    img_url = bucket.sign_url('GET', fb['FS__name'], 600)
                    res[fb['id']]['img_url'].append(img_url)
            else:
                print(fb['FS__name'])
                if fb['FS__name']:
                    img_url = bucket.sign_url('GET', fb['FS__name'], 600)
                    fb['img_url'] = []
                    fb['img_url'].append(img_url)
                res[fb['id']] = fb

        # print(fb_qs.FS_set.all())

        return response.json(0, res)

    def do_add(self, userID, request_dict, response):
        nowTime = int(time.time())
        content = request_dict.get('content', None)
        res_1 = request_dict.get('res_1', None)
        res_2 = request_dict.get('res_2', None)
        res_3 = request_dict.get('res_3', None)
        # res_list = []
        try:
            fb = FeedBackModel()
            fb.addTime = nowTime
            fb.updTime = nowTime
            fb.content = content
            fb.userID_id = userID
            fb.save()
            if res_1:
                fb.FS.add(StatResModel.objects.create(addTime=nowTime, name=res_1))
            if res_2:
                fb.FS.add(StatResModel.objects.create(addTime=nowTime, name=res_2))
            if res_3:
                fb.FS.add(StatResModel.objects.create(addTime=nowTime, name=res_3))
            if not res_1 and not res_2 and not res_3:
                fb.delete()
                return response.json(444, 'res_1,2,3')
        except Exception as e:
            print(repr(e))
            # fb.delete()
            return response.json(500, repr(e))
        else:
            return response.json(0)

    def do_get_upload_url(self, request_dict, response):
        uploadType = request_dict.get('uploadType', None)
        uploadType = json.loads(uploadType)
        if len(uploadType):
            res = []
            auth = oss2.Auth(OSS_STS_ACCESS_KEY, OSS_STS_ACCESS_SECRET)
            bucket = oss2.Bucket(auth, 'oss-cn-hongkong.aliyuncs.com', 'statres')
            for upType in uploadType:
                name = CommonService.createOrderID()
                obj = str(name) + '.' + upType
                url = bucket.sign_url('PUT', obj, 7200)
                res.append({'put_url': url, 'res_name': obj})
            return response.json(0, res)
        else:
            return response.json(444, 'uploadType')

    def do_delete(self, userID, request_dict, response):
        id = request_dict.get('id', None)
        try:
            fb_qs = FeedBackModel.objects.get(userID_id=userID, id=id)
            sr_id_list = fb_qs.FS.values_list('id', flat=True)
            StatResModel.objects.filter(id__in=sr_id_list).delete()
            fb_qs.delete()
            # is_delete = StatResModel.objects.filter(id__in=sr_id_list).delete()

            # print(is_delete)
            print('----------')

        except Exception as e:
            return response.json(500, repr(e))
        else:
            return response.json(0)

    def do_query_by_admin(self, userID, request_dict, response):
        own_perm = ModelService.check_perm(userID, 40)
        if own_perm:
            page = request_dict.get('page', None)
            line = request_dict.get('line', None)
            status = request_dict.get('status', None)
            username = request_dict.get('username', None)
            if page and line:
                page = int(page)
                line = int(line)
                filter_data = {}
                if status:
                    filter_data = {'status': status}
                if username:
                    filter_data = {'userID_id': ModelService.get_userID_byname(username)}
                fb_qs = FeedBackModel.objects.filter()
                if filter_data:
                    fb_qs = fb_qs.filter(**filter_data)
                fb_qs = fb_qs[(page - 1) * line:page * line]. \
                    values('id', 'status', 'content', 'addTime', 'updTime', 'FS__name')
                res = {}
                auth = oss2.Auth(OSS_STS_ACCESS_KEY, OSS_STS_ACCESS_SECRET)
                bucket = oss2.Bucket(auth, 'oss-cn-hongkong.aliyuncs.com', 'statres')
                for fb in fb_qs:
                    if fb['id'] in res:
                        if fb['FS__name']:
                            img_url = bucket.sign_url('GET', fb['FS__name'], 600)
                            res[fb['id']]['img_url'].append(img_url)
                    else:
                        print(fb['FS__name'])
                        if fb['FS__name']:
                            img_url = bucket.sign_url('GET', fb['FS__name'], 600)
                            fb['img_url'] = []
                            fb['img_url'].append(img_url)
                        res[fb['id']] = fb
                return response.json(0, res)
            else:
                return response.json(444, 'page,line')
        else:
            return response.json(404)

    def do_delete_by_admin(self, userID, request_dict, response):
        own_perm = ModelService.check_perm(userID, 10)
        if own_perm:

            id = request_dict.get('id', None)
            try:
                fb_qs = FeedBackModel.objects.get(id=id)
                sr_id_list = fb_qs.FS.values_list('id', flat=True)
                StatResModel.objects.filter(id__in=sr_id_list).delete()
                fb_qs.delete()
                print('----------')
            except Exception as e:
                return response.json(500, repr(e))
            else:
                return response.json(0)
        else:
            return response.json(404)