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 object.ResObject import ResObject
import subprocess
from gevent.pool import Pool


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):
        rq_data = 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
        code = user
        print(user)
        print(pwd)
        auth_request_url = 'http://test.dvema.com/account/login'
        requests_data = {
            'userName': user,
            'userPwd': pwd
        }
        res = requests.post(url=auth_request_url, data=requests_data)
        print(res.json())
        print(res.json()['result_code'])

        if res.json()['result_code'] == 0:
            redirect_uri = redirect_uri + '?code=' + code + '&state=' + state
            from var_dump import var_dump
            var_dump(redirect_uri)
            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'))
        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)
        access_token = "aaaaaaaaaaaaaaa"
        refresh_token = "tGzv3JOkF0XG5Qx2TlKWIA"
        res_json = {
            "access_token": access_token,
            "token_type": "bearer",
            "expires_in": 3600,
            "refresh_token": refresh_token
        }
        response = ResObject()
        return response.json(0, res=res_json)


class oa2RtspStartView(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):
        '''
        VVDHCVBYDKFMJRWA111A
        '''
        UID = request_dict.get("UID", '')
        PWD = request_dict.get("PWD", '')
        MSG = request_dict.get("MSG", '')
        # po = Pool(10)
        # po.apply_async(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=4)
        print("back0----", back[0].decode())  # 注意需要进行解码操作，默认输出的是字节
        print("back1----", back[1].decode())  # back是一个元祖，可以通过元祖取值的方式获取结果
        return