ASJServer/Controller/ApplicationController.py

#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""
@Copyright (C) ansjer cop Video Technology Co.,Ltd.All rights reserved.
@AUTHOR: ASJRD019
@NAME: AnsjerFormal
@software: PyCharm
@DATE: 2020/3/18 9:38
@Version: python3.6
@MODIFY DECORD:ansjer dev
@file: AppSetController.py
@Contact: pzb3076@163.com
"""
import requests
from Ansjer.config import SERVER_TYPE
from Model.models import ApplicationModel
from django.views.generic.base import View
from Object.RedisObject import RedisObject
from Object.TokenObject import TokenObject
from Service.ModelService import ModelService
from Service.CommonService import CommonService
from django.http import JsonResponse
from django.contrib import auth
import time,json
from Object.ResponseObject import ResponseObject
    # http://192.168.136.39:8000/login/oauth/authorize

# http://192.168.136.39:8000/application/query
class AuthView(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 == 'authorize':
                return self.do_authorize(request_dict, userID, response)
            elif operation == 'access_token':
                return self.do_token(request_dict, userID, response)
            elif operation == 'user':
                return self.do_user(request_dict, userID, response)
            else:
                return response.json(tko.code)
        else:
            return response.json(414)

    def do_authorize(self,request_dict, userID, response):
        state = request_dict.get("state", '')
        client_id = request_dict.get("client_id", '')
        response_type = request_dict.get("response_type", '')
        scope = request_dict.get("scope", '')
        redirect_uri = request_dict.get("redirect_uri", '')
        code = CommonService.encrypt_data(32)
        print(redirect_uri)
        redirect_uri = redirect_uri + '?code=' + code + '&state=' + state
        return response.json(0, res=redirect_uri)

        # 增加对code和client_id的校验代码，返回access_token和refresh_token
    def do_token(self,request_dict, userID, response):
        code = request_dict.get("code", None)
        client_id = request_dict.get("client_id", None)
        refresh_token = request_dict.get("refresh_token", None)
        print('refresh_token:')
        print(refresh_token)
        print('code:')
        print(code)
        print('client_id:')
        print(client_id)
        access_token = CommonService.encrypt_data(randomlength=32)
        refresh_token = CommonService.encrypt_data(randomlength=32)
        res_json = {
            "access_token": access_token,
            "token_type": "bearer",
            "expires_in": 3600,
            "refresh_token": refresh_token
        }
        print(res_json)
        return JsonResponse(res_json)

    def do_user(self, request_dict, userID, response):
        code = request_dict.get("code", None)
        client_id = request_dict.get("client_id", None)
        refresh_token = request_dict.get("refresh_token", None)
        print('refresh_token:')
        print(refresh_token)
        print('code:')
        print(code)
        print('client_id:')
        print(client_id)
        access_token = CommonService.encrypt_data(randomlength=32)
        refresh_token = CommonService.encrypt_data(randomlength=32)
        res_json ={
          "login": "pengzhibo168",
          "id": 1,
          "node_id": "MDQ6VXNlcjE=",
          "gravatar_id": "",
          "type": "User",
          "name": "monalisa octocat",
          "company": "asdfda",
          "location": "San Francisco",
          "email": "sadf@sadf.com",
          "bio": "There once was...",
          "public_repos": 2,
          "public_gists": 1,
          "followers": 20,
          "following": 0,
          "created_at": "2008-01-14T04:33:35Z",
          "updated_at": "2008-01-14T04:33:35Z"
        }
        print(res_json)
        return JsonResponse(res_json)








class ApplicationView(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 == 'query':
                return self.query(request_dict, userID, response)
            elif operation == 'add':
                return self.add(request_dict, userID, response)
            elif operation == 'update':
                return self.update(request_dict, userID, response)
            elif operation == 'delete':
                return self.delete(request_dict, userID, response)
            else:
                return response.json(414)

    def query(self, request_dict, userID, response):
        page = int(request_dict.get('page', 0))
        line = int(request_dict.get('line', 0))
        if page == 0:
            page=1
        if line == 0:
            line=10
        qs = ApplicationModel.objects.all()
        if qs.exists():
            count = qs.count()
            res = qs[(page - 1) * line:page * line]
            send_json = CommonService.qs_to_dict(res)
            send_json['count'] = count
            return response.json(0, send_json)
        return response.json(0, {'datas': [], 'count': 0})

    # 管理员的编辑
    def update(self, request_dict, userID, response):
        own_perm = ModelService.check_perm(userID=userID, permID=50)
        if own_perm is not True:
            return response.json(404)
        deviceContent = request_dict.get('content', None)
        id = request_dict.get('id', None)
        if not deviceContent or not id:
            return response.json(444, 'content,id')
        try:
            timestamp = int(time.time())
            deviceData = json.loads(deviceContent)
            uid_set = ApplicationModel.objects.filter(id=id)
            if uid_set.exists():
                uid_set.update(updTime=timestamp, **deviceData)
                return response.json(0)
            else:
                return response.json(173)
        except Exception:
            errorInfo = traceback.format_exc()
            print(errorInfo)
            return response.json(500, {'details': errorInfo})



    def delete(self, request_dict, userID, response):
        own_perm = ModelService.check_perm(userID=userID, permID=10)
        if own_perm is not True:
            return response.json(404)
        id_list = request_dict.getlist('id', None)
        if not id_list:
            return response.json(444, 'id must list')
        try:
            for id in id_list:
                ApplicationModel.objects.filter(id=id).delete()
        except Exception as e:
            errorInfo = traceback.format_exc()
            print(errorInfo)
            return response.json(424, {'details': repr(e)})
        else:
            return response.json(0)