ASJAzoauth/controller/index.py

#!/usr/bin/env python3  
# -*- coding: utf-8 -*-  
"""
@Copyright (C) ansjer cop Video Technology Co.,Ltd.All rights reserved.
@AUTHOR: ASJRD018
@NAME: azoauth
@software: PyCharm
@DATE: 2020/1/13 17:01
@Version: python3.6
@MODIFY DECORD:ansjer dev
@file: index.py
@Contact: chanjunkai@163.com
"""
import json
import time
import requests
from django.views.generic import TemplateView
from django.shortcuts import render_to_response
from django.http import JsonResponse
from object.ResObject import ResObject
import subprocess
# from gevent.pool import Pool
from model.models import UserModel, UidRtspModel
from object.tkObject import tkObject
from service.CommonService import CommonService


class authView(TemplateView):
    def post(self, request, *args, **kwargs):
        request.encoding = 'utf-8'
        request_dict = json.loads(request.body.decode('utf-8'))
        return self.validate(request_dict)

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

    def validate(self, request_dict):
        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", '')
        context = {
            'state': state,
            'client_id': client_id,
            'response_type': response_type,
            'scope': scope,
            'redirect_uri': redirect_uri,
        }
        return render_to_response("login.html", context)
        # return render_template('./login.html', **context)


class loginHandleView(TemplateView):
    def post(self, request, *args, **kwargs):
        request.encoding = 'utf-8'
        request_dict = json.loads(request.body.decode('utf-8'))
        return self.validate(request_dict)

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

    def validate(self, request_dict):
        response = ResObject()
        user = request_dict.get("user", '')
        pwd = request_dict.get("pwd", '')
        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
        print(user)
        print(pwd)
        auth_request_url = 'http://test.dvema.com/oalexa/auth'
        # auth_request_url = 'http://192.168.136.40:8077/oalexa/auth'
        requests_data = {'userName': user, 'userPwd': pwd}
        res = requests.post(url=auth_request_url, data=requests_data)
        print(res.json())
        print(res.json()['result_code'])
        res_json = res.json()
        if res_json['result_code'] == 0:
            userID = res_json['result']['userID']
            uid_arr = res_json['result']['uid_arr']
            tko = tkObject(tk='', rank=3)
            token = tko.encrypt(data=userID)
            nowTime = int(time.time())
            user_qs = UserModel.objects.filter(userID=userID)
            if not user_qs.exists():
                user_qs = UserModel.objects.create(userID=userID, token=token, addTime=nowTime, updTime=nowTime)
                userID = user_qs.userID
                user_qs = UserModel.objects.filter(userID=userID)
            else:
                userID = user_qs[0].userID
            querysetlist = []
            rtko = tkObject(rank=1)
            UidRtspModel.objects.filter(user_id=userID).delete()
            for uid_a in uid_arr:
                rtsp_url = rtko.encrypt(data=uid_a['uid'])
                create_data = {
                    'uid': uid_a['uid'],
                    'nick': uid_a['nick'],
                    'addTime': nowTime,
                    'updTime': nowTime,
                    'user_id': userID,
                    'rtsp_url': rtsp_url
                }
                querysetlist.append(UidRtspModel(**create_data))
            UidRtspModel.objects.bulk_create(querysetlist)
            code = CommonService.encrypt_data(32)
            user_qs.update(code=code)
            redirect_uri = redirect_uri + '?code=' + code + '&state=' + state
            return response.json(0, res=redirect_uri)
        else:
            return response.json(0, res={'msg': 'error'})


class oa2TokenView(TemplateView):

    def post(self, request, *args, **kwargs):
        request.encoding = 'utf-8'
        # request_dict = json.loads(request.body.decode('utf-8'))
        request_dict = request.POST
        return self.validate(request_dict)

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

    def validate(self, request_dict):
        # 增加对code和client_id的校验代码，返回access_token和refresh_token
        code = request_dict.get("code", None)
        client_id = request_dict.get("client_id", None)

        print('code:')
        print(code)
        print('client_id:')
        print(client_id)
        user_qs = UserModel.objects.filter(code=code)
        if user_qs.exists():

            access_token = CommonService.encrypt_data(randomlength=32)
            refresh_token = CommonService.encrypt_data(randomlength=32)

            user_qs.update(access_token=access_token, refresh_token=refresh_token)
            res_json = {
                "access_token": access_token,
                "token_type": "bearer",
                "expires_in": 3600,
                "refresh_token": refresh_token
            }
            print(res_json)
            return JsonResponse(res_json)
        else:
            res_json={'msg': 'code not exists'}
            print(res_json)
            return JsonResponse(res_json)


class oa2RtspStartView(TemplateView):
    def post(self, request, *args, **kwargs):
        request.encoding = 'utf-8'
        request_dict = request.POST
        # request_dict = json.loads(request.body.decode('utf-8'))
        return self.validate(request_dict)

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

    def validate(self, request_dict):
        '''
        VVDHCVBYDKFMJRWA111A
        '''
        UID = request_dict.get("UID", '')
        PWD = request_dict.get("PWD", '')
        MSG = request_dict.get("MSG", '')
        # po = Pool(10)
        # po.apply_async(self.runSendRtspMsg, (UID, PWD, MSG))
        self.runSendRtspMsg(UID, PWD, MSG)
        response = ResObject()
        return response.json(0, {'res': 'yes'})

    def runSendRtspMsg(self, UID, PWD, MSG):
        command = "./test {UID} {PWD} {MSG}".format(UID=UID, PWD=PWD, MSG=MSG)
        print('command=>{command}'.format(command=command))
        back = subprocess.Popen(command, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE).communicate(
            timeout=10)
        print("back0----", back[0].decode())  # 注意需要进行解码操作，默认输出的是字节
        print("back1----", back[1].decode())  # back是一个元祖，可以通过元祖取值的方式获取结果
        return