#!/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'] nowTime = int(time.time()) user_qs = UserModel.objects.filter(userID=userID) if not user_qs.exists(): user_qs = UserModel.objects.create(userID=userID, 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'], 'password': uid_a['password'], '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) is_update = user_qs.update(access_token=access_token, refresh_token=refresh_token) print(is_update) if is_update: res_json = { "access_token": access_token, "token_type": "bearer", "expires_in": 3600, "refresh_token": refresh_token } print(res_json) return JsonResponse(res_json) else: return JsonResponse({'msg':'error'}) 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 class oa2DiscoveryDevice(TemplateView): def post(self, request, *args, **kwargs): request.encoding = '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 access_token = request_dict.get("access_token", None) user_qs = UserModel.objects.filter(access_token=access_token) response = ResObject() if user_qs.exists(): userID = user_qs[0].userID ur_qs = UidRtspModel.objects.filter(user_id=userID).values('uid', 'nick', 'rtsp_url', 'id') if ur_qs.exists(): res_json = [] for ur in ur_qs: ur_data = { 'endpointId': ur['id'], 'manufacturerName': 'Axis', 'manufacturerId': 'axis-ACCC8E5E7513', 'modelName': 'P1425-LE', 'friendlyName': 'Camera {nick}'.format(nick=ur['nick']), 'description': 'Camera {nick}'.format(nick=ur['nick']), 'resolutions': [{'width': 1280, 'height': 720}], 'videoCodecs': ['H264'], 'audioCodecs': ['ACC'], 'protocols': ['RTSP'], 'authorizationTypes': ['NONE'], 'uri': 'rtsp://www.zositech.org:443/{stream_name}'.format(stream_name=ur['rtsp_url']) } res_json.append(ur_data) return JsonResponse(res_json,safe=False) # return response.json(0, res_json) else: return response.json(0, 'data is none~') else: return JsonResponse({'res':'11'}) return response.json(0, 'error')