#!/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 auth_request_url = 'http://test.dvema.com/account/login?userName=13800138001&userPwd=ansjer999999' requests_data = { 'userName': user, 'userPwd': pwd } res = requests.post(url=auth_request_url, data=requests_data) print(res.json()['result_code']) if res.json()['result_code'] == 0: redirect_uri = redirect_uri + '?code=' + code + '&state=' + state return 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 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): ''' 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