ASJServer/Controller/PushDeploy.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/12/23 10:30
@Version: python3.6
@MODIFY DECORD:ansjer dev
@file: PushDeploy.py
@Contact: pzb3076@163.com
"""
import json
import math

from django.utils.decorators import method_decorator
from django.views.decorators.csrf import csrf_exempt
from django.views.generic.base import View

from Model.models import pushDeployModel
from Object.ResponseObject import ResponseObject
from Object.TokenObject import TokenObject
from Service.CommonService import CommonService
from Service.ModelService import ModelService
import time


class PushDeployView(View):
    @method_decorator(csrf_exempt)
    def dispatch(self, *args, **kwargs):
        return super(PushDeployView, self).dispatch(*args, **kwargs)

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

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

    def validation(self, request_dict, request, operation):
        response = ResponseObject()
        if operation is None:
            return response.json(444, 'error path')
        token = request_dict.get('token', None)
        # 设备主键uid
        tko = TokenObject(token)
        response.lang = tko.lang
        if tko.code != 0:
            return response.json(tko.code)
        userID = tko.userID
        if operation == 'adminAdd':
            return self.do_add(request_dict, response,userID)
        elif operation == 'adminDelete':
            return self.do_delete(request_dict, response,userID)
        elif operation == 'adminUpdate':
            return self.do_update(request_dict, response,userID)
        elif operation == 'adminQuery':
            return self.do_query(request_dict,response, userID)
        else:
            return response.json(414)

    def do_add(self, request_dict, response, userID):
        own_perm = ModelService.check_perm(userID, 40)
        if not own_perm:
            return response.json(404)
        name = request_dict.get('name', None)
        key = request_dict.get('key', None)
        secret = request_dict.get('secret', None)
        pem = request_dict.get('pem', None)
        type = request_dict.get('type', None)
        nowTime = int(time.time())
        try:
            pushDeployModel.objects.create(addTime=nowTime, updTime=nowTime, name=name, key=key, secret=secret,pem=pem,
                                           type=type)
        except Exception as e:
            return response.json(10, repr(e))
        res = list(pushDeployModel.objects.values())
        return response.json(0, res)

    def do_query(self,request_dict, response,userID):
        own_perm = ModelService.check_perm(userID, 20)
        if own_perm is True:
            page = request_dict.get('page', None)
            line = request_dict.get('line', None)
            page = int(page)
            line = int(line)
            deviceContent = request_dict.get('content', None)
            if deviceContent:
                try:
                    searchCondition = json.loads(deviceContent)
                except Exception as e:
                    return response.json(10, repr(e))
                else:
                    kwargs = CommonService.get_kwargs(data=searchCondition)
                    uid_user_qs = pushDeployModel.objects.filter(**kwargs).order_by('-id')
                    if not uid_user_qs.exists():
                        return response.json(0, [])
                    count = uid_user_qs.count()
                    uid_user_ql = uid_user_qs[(page - 1) * line:page * line]. \
                        values('id', 'name', 'key', 'secret', 'pem', 'type',
                               'addTime', 'updTime')
                    uid_user_ls = CommonService.qs_to_list(uid_user_ql)
                    return response.json(0, {'datas': uid_user_ls, 'count': count})
            else:
                uid_user_qs = pushDeployModel.objects.all().order_by('-id')
                if not uid_user_qs.exists():
                    return response.json(0, [])
                count = uid_user_qs.count()
                uid_user_ql = uid_user_qs[(page - 1) * line:page * line]. \
                    values('id', 'name', 'key', 'secret', 'pem', 'type',
                           'addTime', 'updTime')
                uid_user_ls = CommonService.qs_to_list(uid_user_ql)
                return response.json(0, {'datas': uid_user_ls, 'count': count})
        else:
            return response.json(404)

    def do_update(self, request_dict, response,userID):
        own_perm = ModelService.check_perm(userID, 30)
        if not own_perm:
            return response.json(404)
        request_dict.pop('token')
        nowTime = int(time.time())
        id = request_dict.get('id', None)
        try:
            pushDeployModel.objects.filter(id=id).update(updTime=nowTime, **request_dict)
        except Exception as e:
            return response.json(10, repr(e))
        # res = list(pushDeployModel.objects.values())
        return response.json(0, {'updTime': nowTime})

    def do_delete(self,request_dict, response, userID):
        own_perm = ModelService.check_perm(userID, 10)
        if not own_perm:
            return response.json(404)
        id = request_dict.get('id', None)
        try:
            pushDeployModel.objects.filter(id=id).delete()
        except Exception as e:
            return response.json(10, repr(e))
        res = list(pushDeployModel.objects.values())
        return response.json(0, res)