index.py 40 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988
  1. import json
  2. import logging
  3. import subprocess
  4. import threading
  5. import time
  6. from collections import OrderedDict
  7. from datetime import datetime
  8. from gc import enable
  9. import requests
  10. from django.http import JsonResponse
  11. from django.shortcuts import render
  12. from django.views import View
  13. from django.views.generic import TemplateView
  14. from azoauth.config import *
  15. from model.models import UserModel, UidRtspModel, AlexaAuthModel, UserCountModel, SwitchModel
  16. from object.RedisObject import RedisObject
  17. from object.ResObject import ResObject
  18. from object.ResponseObject import ResponseObject
  19. from object.tkObject import tkObject
  20. from service.CommonService import CommonService
  21. import uuid
  22. class authView(TemplateView):
  23. def post(self, request, *args, **kwargs):
  24. request.encoding = 'utf-8'
  25. request_dict = json.loads(request.body.decode('utf-8'))
  26. return self.validate(request_dict)
  27. def get(self, request, *args, **kwargs):
  28. request.encoding = 'utf-8'
  29. request_dict = request.GET
  30. return self.validate(request_dict)
  31. def validate(self, request_dict):
  32. logger = logging.getLogger('django')
  33. logger.info('------oa2/auth请求参数: {}------'.format(request_dict))
  34. state = request_dict.get("state", '')
  35. client_id = request_dict.get("client_id", '')
  36. response_type = request_dict.get("response_type", '')
  37. scope = request_dict.get("scope", '')
  38. redirect_uri = request_dict.get("redirect_uri", '')
  39. context = {
  40. 'state': state,
  41. 'client_id': client_id,
  42. 'response_type': response_type,
  43. 'scope': scope,
  44. 'redirect_uri': redirect_uri,
  45. 'skill_name': 'zosi smart'
  46. }
  47. return render(request=None, template_name="login.html", context=context)
  48. # Anlapus登录
  49. class authAnlapusView(TemplateView):
  50. def post(self, request, *args, **kwargs):
  51. request.encoding = 'utf-8'
  52. request_dict = json.loads(request.body.decode('utf-8'))
  53. return self.validate(request_dict)
  54. def get(self, request, *args, **kwargs):
  55. request.encoding = 'utf-8'
  56. request_dict = request.GET
  57. return self.validate(request_dict)
  58. def validate(self, request_dict):
  59. state = request_dict.get("state", '')
  60. client_id = request_dict.get("client_id", '')
  61. response_type = request_dict.get("response_type", '')
  62. scope = request_dict.get("scope", '')
  63. redirect_uri = request_dict.get("redirect_uri", '')
  64. context = {
  65. 'state': state,
  66. 'client_id': client_id,
  67. 'response_type': response_type,
  68. 'scope': scope,
  69. 'redirect_uri': redirect_uri,
  70. 'skill_name': 'Anlapus'
  71. }
  72. return render(request=None, template_name="login_anlapus.html", context=context)
  73. # loocam登录
  74. class authLoocamView(TemplateView):
  75. def post(self, request, *args, **kwargs):
  76. request.encoding = 'utf-8'
  77. request_dict = json.loads(request.body.decode('utf-8'))
  78. return self.validate(request_dict)
  79. def get(self, request, *args, **kwargs):
  80. request.encoding = 'utf-8'
  81. request_dict = request.GET
  82. return self.validate(request_dict)
  83. def validate(self, request_dict):
  84. state = request_dict.get("state", '')
  85. client_id = request_dict.get("client_id", '')
  86. response_type = request_dict.get("response_type", '')
  87. scope = request_dict.get("scope", '')
  88. redirect_uri = request_dict.get("redirect_uri", '')
  89. context = {
  90. 'state': state,
  91. 'client_id': client_id,
  92. 'response_type': response_type,
  93. 'scope': scope,
  94. 'redirect_uri': redirect_uri,
  95. 'skill_name': 'loocam'
  96. }
  97. logger = logging.getLogger('django')
  98. logger.info('loocam请求登录网页,参数为{}'.format(context))
  99. return render(request=None, template_name="login_loocam.html", context=context)
  100. # VSees登录
  101. class authVSeesView(TemplateView):
  102. def post(self, request, *args, **kwargs):
  103. request.encoding = 'utf-8'
  104. request_dict = json.loads(request.body.decode('utf-8'))
  105. return self.validate(request_dict)
  106. def get(self, request, *args, **kwargs):
  107. request.encoding = 'utf-8'
  108. request_dict = request.GET
  109. return self.validate(request_dict)
  110. def validate(self, request_dict):
  111. state = request_dict.get("state", '')
  112. client_id = request_dict.get("client_id", '')
  113. response_type = request_dict.get("response_type", '')
  114. scope = request_dict.get("scope", '')
  115. redirect_uri = request_dict.get("redirect_uri", '')
  116. context = {
  117. 'state': state,
  118. 'client_id': client_id,
  119. 'response_type': response_type,
  120. 'scope': scope,
  121. 'redirect_uri': redirect_uri,
  122. 'skill_name': 'VSees'
  123. }
  124. logger = logging.getLogger('django')
  125. logger.info('VSees请求登录网页,参数为{}'.format(context))
  126. return render(request=None, template_name="login_VSees.html", context=context)
  127. class loginHandleView(TemplateView):
  128. def post(self, request, *args, **kwargs):
  129. request.encoding = 'utf-8'
  130. request_dict = json.loads(request.body.decode('utf-8'))
  131. return self.validate(request_dict)
  132. def get(self, request, *args, **kwargs):
  133. request.encoding = 'utf-8'
  134. request_dict = request.GET
  135. return self.validate(request_dict)
  136. def validate(self, request_dict):
  137. response = ResObject()
  138. user = request_dict.get("user", '')
  139. pwd = request_dict.get("pwd", '')
  140. state = request_dict.get("state", '')
  141. client_id = request_dict.get("client_id", '')
  142. response_type = request_dict.get("response_type", '')
  143. scope = request_dict.get("scope", '')
  144. redirect_uri = request_dict.get("redirect_uri", '')
  145. skill_name = request_dict.get("skill_name", '')
  146. # 返回code
  147. logger = logging.getLogger('django')
  148. logger.info('------开始认证登录------参数:{}'.format(request_dict))
  149. # 请求美洲服登录接口
  150. region_code = 'US'
  151. auth_request_url = '{SERVER_PREFIX}/oalexa/auth'.format(SERVER_PREFIX=SERVER_PREFIX)
  152. requests_data = {'userName': user, 'userPwd': pwd}
  153. res = requests.post(url=auth_request_url, data=requests_data)
  154. if res.status_code != 200:
  155. return response.json(10, res={'错误': '请求响应异常'})
  156. res_json = res.json()
  157. logger.info('美洲服务器响应: {}'.format(res_json))
  158. # 账号不存在/密码错误/账号存在但用户地区为欧洲时请求欧洲服
  159. result_code = res_json['result_code']
  160. if result_code == 104 or result_code == 111 or (result_code == 0 and res_json['result']['region_code'] == 'EU'):
  161. auth_request_url = '{SERVER_PREFIX}/oalexa/auth'.format(SERVER_PREFIX=SERVER_PREFIX_EU)
  162. res = requests.post(url=auth_request_url, data=requests_data)
  163. if res.status_code != 200:
  164. return response.json(10, res={'错误': '请求响应异常'})
  165. res_json = res.json()
  166. logger.info('欧洲服务器响应: {}'.format(res_json))
  167. if res_json['result_code'] == 0:
  168. region_code = 'EU'
  169. if res_json['result_code'] != 0:
  170. return response.json(10, res={'msg': 'error'}, extra={'msg': res_json['reason']})
  171. nowTime = int(time.time())
  172. code = CommonService.encrypt_data(32)
  173. userID = res_json['result']['userID']
  174. user_qs = UserModel.objects.filter(userID=userID)
  175. if user_qs.exists():
  176. user_qs.update(region_code=region_code, code=code, updTime=nowTime)
  177. else:
  178. UserModel.objects.create(userID=userID, region_code=region_code, code=code, addTime=nowTime,
  179. updTime=nowTime)
  180. year_month = str(time.strftime('%Y%m', time.localtime(nowTime))) # 获取当前年月
  181. user_count_qs = UserCountModel.objects.filter(skill_name=skill_name, year_month=year_month).values('amount')
  182. if not user_count_qs.exists():
  183. UserCountModel.objects.create(skill_name=skill_name, year_month=year_month, amount=1)
  184. else:
  185. # 用户数量+1
  186. amount = user_count_qs[0]['amount'] + 1
  187. user_count_qs.update(amount=amount)
  188. redirect_uri += '?code=' + code + '&state=' + state
  189. logger.info('------认证登录响应------:{}'.format(redirect_uri))
  190. return response.json(0, res=redirect_uri)
  191. class oa2TokenView(TemplateView):
  192. def post(self, request, *args, **kwargs):
  193. request.encoding = 'utf-8'
  194. # request_dict = json.loads(request.body.decode('utf-8'))
  195. request_dict = request.POST
  196. return self.validate(request_dict)
  197. def get(self, request, *args, **kwargs):
  198. request.encoding = 'utf-8'
  199. request_dict = request.GET
  200. return self.validate(request_dict)
  201. def validate(self, request_dict):
  202. # 增加对code和client_id的校验代码,返回access_token和refresh_token
  203. code = request_dict.get("code", None)
  204. grant_type = request_dict.get("grant_type", None) # refresh_token, authorization_code
  205. client_id = request_dict.get("client_id", None)
  206. refresh_token = request_dict.get("refresh_token", None)
  207. logger = logging.getLogger('django')
  208. logger.info('请求获取令牌接口参数:{}'.format(request_dict))
  209. # 根据用户授权码获取令牌
  210. user_qs = UserModel.objects.filter(code=code)
  211. if not user_qs.exists():
  212. if grant_type == 'authorization_code':
  213. user_qs = UserModel.objects.filter(user_authorization_code=code)
  214. else:
  215. user_qs = UserModel.objects.filter(refresh_token=refresh_token)
  216. if not user_qs.exists():
  217. res_json = {'msg': 'code not exists'}
  218. logger.info('请求获取令牌接口响应:{}'.format(res_json))
  219. return JsonResponse(res_json)
  220. access_token = CommonService.encrypt_data(randomlength=32)
  221. refresh_token = CommonService.encrypt_data(randomlength=32)
  222. is_update = user_qs.update(access_token=access_token, refresh_token=refresh_token)
  223. if is_update:
  224. res_json = {
  225. "access_token": access_token,
  226. "token_type": "bearer",
  227. "expires_in": 3600,
  228. "refresh_token": refresh_token,
  229. }
  230. logger.info('请求获取令牌接口响应:{}'.format(res_json))
  231. return JsonResponse(res_json)
  232. else:
  233. logger.info({'msg': 'error'})
  234. return JsonResponse({'msg': 'error'})
  235. def runSendRtspMsg_thread(UID, PWD, MSG):
  236. command = "./pushtool {UID} {PWD} {MSG} 1".format(UID=UID, PWD=PWD, MSG=MSG)
  237. print('command=>{command}'.format(command=command))
  238. try:
  239. back = subprocess. \
  240. Popen(command, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE). \
  241. communicate(timeout=10)
  242. except Exception as e:
  243. return False
  244. else:
  245. print("back0----", back[0].decode()) # 注意需要进行解码操作,默认输出的是字节
  246. print("back1----", back[1].decode()) # back是一个元祖,可以通过元祖取值的方式获取结果
  247. return True
  248. class oa2RtspStartView(TemplateView):
  249. def post(self, request, *args, **kwargs):
  250. request.encoding = 'utf-8'
  251. request_dict = request.POST
  252. return self.validate(request_dict)
  253. def get(self, request, *args, **kwargs):
  254. request.encoding = 'utf-8'
  255. request_dict = request.GET
  256. return self.validate(request_dict)
  257. def validate(self, request_dict):
  258. st = request_dict.get("st", 0)
  259. uid = request_dict.get("id", '')
  260. access_token = request_dict.get("access_token", '')
  261. skill_name = request_dict.get("skill_name", 'zosi smart')
  262. user_qs = UserModel.objects.filter(access_token=access_token)
  263. if not user_qs.exists():
  264. return JsonResponse({'错误': '用户数据不存在'})
  265. ur_qs = UidRtspModel.objects.filter(uid=uid).values('user_id', 'uid', 'nick', 'rtsp_url', 'password')
  266. if not ur_qs.exists():
  267. return JsonResponse({'错误': 'uid数据不存在'})
  268. # 确认设备的用户地区
  269. region = 'US'
  270. user_id = ur_qs[0]['user_id']
  271. user_qs = UserModel.objects.filter(userID=user_id).values('region_code')
  272. if user_qs.exists():
  273. region = user_qs[0]['region_code']
  274. UID = ur_qs[0]['uid']
  275. nick = ur_qs[0]['nick']
  276. PWD = ur_qs[0]['password']
  277. stream_name = ur_qs[0]['rtsp_url']
  278. channel = '0'
  279. if '_' in UID:
  280. # 多通道设备
  281. channel = UID[-1:]
  282. UID = UID[:-2]
  283. # 根据用户地区确认域名
  284. domain_name = SERVER_PREFIX_EU if region == 'EU' else SERVER_PREFIX
  285. RESP_SERVER_DOMAIN = RESP_SERVER_DOMAIN_DATA[region]
  286. MSG = '{}://{}:8554/{}'.format(RTSP_PREFIX, RESP_SERVER_DOMAIN, stream_name)
  287. # 测试往go2rtc推流
  288. if UID == '3VCV5HZD9DRPA6EU111A':
  289. MSG = 'rtsp://144.24.9.228:8554/camera1'
  290. logger = logging.getLogger('django')
  291. logger.info('------{} 开始向设备下发推流指令------'.format(skill_name))
  292. # 此处用线程异步发送指令
  293. if int(st) == 1:
  294. send_flag = self.runSendStop(UID, PWD, MSG)
  295. logger.info('----------send_flag---st=1-----------------')
  296. if send_flag:
  297. return JsonResponse({'msg': 'stop yes', 'code': 0})
  298. else:
  299. return JsonResponse({'msg': 'stop no', 'code': 0})
  300. # 创建并启动线程
  301. thread = threading.Thread(
  302. target=self.send_rtsp_command_thread,
  303. args=(domain_name, UID, PWD, MSG, channel, logger, region)
  304. )
  305. thread.start()
  306. # 拉流地址
  307. rtsp_uri = '{}://{}:443/{}'.format(RTSP_PREFIX, RESP_SERVER_DOMAIN, stream_name)
  308. stop_time = int(time.time()) + 2 * 60
  309. expirationTime = time.strftime('%Y-%m-%dT%H:%MZ', time.localtime(stop_time))
  310. res_json = {
  311. 'uid': UID,
  312. 'pwd': PWD,
  313. 'msg': MSG,
  314. 'uri': rtsp_uri,
  315. 'endpointId': uid,
  316. 'friendlyName': nick,
  317. 'manufacturerName': skill_name,
  318. 'expirationTime': expirationTime,
  319. 'description': 'Camera connected via {}'.format(skill_name),
  320. 'audioCodecs': 'G711',
  321. 'videoCodecs': 'H264',
  322. 'protocols': 'RTSP',
  323. 'authorizationTypes': 'NONE',
  324. 'resolutions': {'width': 640, 'height': 360},
  325. }
  326. logger.info('------------返回控制摄像头的信息---------------: {}'.format(res_json))
  327. return JsonResponse(res_json, safe=False)
  328. def send_rtsp_command_thread(self, domain_name, UID, PWD, MSG, channel, logger, region):
  329. """异步发送RTSP命令的线程函数"""
  330. thing_name = CommonService.query_serial_with_uid(UID)
  331. topic_name = 'ansjer/generic/{}'.format(thing_name)
  332. msg = OrderedDict(
  333. [
  334. ('alexaRtspCommand', MSG),
  335. ('enable', 1),
  336. ]
  337. )
  338. publish_result = CommonService.req_publish_mqtt_msg(thing_name, topic_name, msg)
  339. logger.info('{}下发指令结果:{}'.format(UID, publish_result))
  340. command = "./pushtool {UID} {PWD} {MSG} 1 {channel}".format(
  341. UID=UID, PWD=PWD, MSG=MSG, channel=channel)
  342. # 请求MQTT发布消息
  343. url = '{}/iot/requestPublishMessage'.format(domain_name)
  344. requests_data = {'UID': UID, 'rtsp': MSG, 'enable': '1'}
  345. try:
  346. r = requests.post(url, requests_data)
  347. if r.status_code == 200:
  348. res = r.json()
  349. logger.info('请求MQTT发布消息参数:{},result_code: {}'.format(requests_data, res['result_code']))
  350. if res['result_code'] == 0:
  351. logger.info('请求MQTT下发指令成功---正式服')
  352. elif res['result_code'] == 10044:
  353. url = '{}/iot/requestPublishMessage'.format(SERVER_PREFIX_TEST)
  354. r = requests.post(url, requests_data)
  355. if r.status_code == 200:
  356. res = r.json()
  357. if res['result_code'] == 0:
  358. logger.info('请求MQTT下发指令成功---测试服')
  359. else:
  360. self.runSendRtspMsg(logger, region, command)
  361. else:
  362. self.runSendRtspMsg(logger, region, command)
  363. else:
  364. self.runSendRtspMsg(logger, region, command)
  365. else:
  366. self.runSendRtspMsg(logger, region, command)
  367. except Exception as e:
  368. logger.error('发送RTSP命令异常: {}'.format(str(e)))
  369. self.runSendRtspMsg(logger, region, command)
  370. def runReqRtspMsg(self, UID, PWD, MSG):
  371. request_url = 'http://localhost:5000/?UID={UID}&MSG={MSG}&CMD=1&PWD={PWD}'. \
  372. format(UID=UID, PWD=PWD, MSG=MSG)
  373. res = requests.get(url=request_url)
  374. print(res)
  375. return True
  376. # 触发此方法,让摄像头推流到MSG流地址
  377. def runSendRtspMsg(self, logger, region, command):
  378. logger.info('------------推流指令: {}---------------'.format(command))
  379. if region == 'CN':
  380. logger.info('------------国内发送推流指令---------------')
  381. url = "http://52.83.252.41:7880/alexa/command?command={command}".format(command=command)
  382. # 请求国内服务器调用pushtool
  383. try:
  384. requests.get(url=url, timeout=10)
  385. except Exception as e:
  386. logger.info('请求国内服务器调用pushtool异常: {}'.format(repr(e)))
  387. else:
  388. logger.info('------------国外发送推流指令---------------')
  389. try:
  390. back = subprocess.Popen(command, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE). \
  391. communicate(timeout=10)
  392. logger.info('back: {}'.format(str(back[0].decode()) + str(back[1].decode())))
  393. except Exception as e:
  394. self.runSendRtspMsg(logger, 'CN', command) # 调用失败时尝试用国内的发送
  395. logger.info('调用pushtool异常: {}'.format(repr(e)))
  396. def runSendStop(self, UID, PWD, MSG):
  397. command = "./pushtool {UID} {PWD} {MSG} 0".format(UID=UID, PWD=PWD, MSG=MSG)
  398. print('command=>{command}'.format(command=command))
  399. try:
  400. back = subprocess.Popen(command, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE). \
  401. communicate(timeout=10)
  402. except Exception as e:
  403. return False
  404. else:
  405. print("back0----", back[0].decode()) # 注意需要进行解码操作,默认输出的是字节
  406. print("back1----", back[1].decode()) # back是一个元祖,可以通过元祖取值的方式获取结果
  407. return True
  408. class oa2DiscoveryDevice(TemplateView):
  409. def post(self, request, *args, **kwargs):
  410. request.encoding = 'utf-8'
  411. request_dict = request.POST
  412. return self.validate(request_dict)
  413. def get(self, request, *args, **kwargs):
  414. request.encoding = 'utf-8'
  415. request_dict = request.GET
  416. return self.validate(request_dict)
  417. def validate(self, request_dict):
  418. # 增加对code和client_id的校验代码,返回access_token和refresh_token
  419. skill_name = request_dict.get("skill_name", 'zosi smart')
  420. access_token = request_dict.get("access_token", None)
  421. logger = logging.getLogger('django')
  422. logger.info('--------{} 开始搜索设备--------'.format(skill_name))
  423. response = ResObject()
  424. user_qs = UserModel.objects.filter(access_token=access_token)
  425. if not user_qs.exists():
  426. return response.json(500, res={'msg': '用户数据不存在'})
  427. user = user_qs[0]
  428. userID = user.userID
  429. region_code = user.region_code
  430. logger.info('搜索设备的用户id:{},地区:{}'.format(userID, region_code))
  431. # AlexaAuthModel表数据
  432. alexAuth = AlexaAuthModel.objects.filter(token=access_token, skill_name=skill_name).order_by('-addTime')
  433. if alexAuth.exists():
  434. auth_res = alexAuth.values()
  435. event_access_token = auth_res[0]['access_token']
  436. event_refresh_token = auth_res[0]['refresh_token']
  437. event_token = auth_res[0]['token']
  438. event_expiresTime = auth_res[0]['expiresTime']
  439. event_addTime = auth_res[0]['addTime']
  440. event_updTime = auth_res[0]['updTime']
  441. event_alexa_region = auth_res[0]['alexa_region']
  442. AlexaAuthModel.objects.filter(userID=userID).delete()
  443. alexAuth.delete()
  444. AlexaAuthModel.objects.create(
  445. userID=userID,
  446. access_token=event_access_token,
  447. refresh_token=event_refresh_token,
  448. token=event_token,
  449. expiresTime=event_expiresTime,
  450. addTime=event_addTime,
  451. updTime=event_updTime,
  452. alexa_region=event_alexa_region,
  453. skill_name=skill_name,
  454. )
  455. # 请求搜索设备
  456. domain_name = SERVER_PREFIX_EU if region_code == 'EU' else SERVER_PREFIX
  457. auth_request_url = '{}/oalexa/discoveryuid'.format(domain_name)
  458. requests_data = {'sid': 'admin', 'sst': 'admin', 'alexa_user_id': userID}
  459. res = requests.post(url=auth_request_url, data=requests_data)
  460. if res.status_code != 200:
  461. return response.json(500, res={'msg': 'discover device error'})
  462. res_json = res.json()
  463. logger.info('{}服务器搜索设备响应:{}'.format(region_code, res_json))
  464. if res_json['result_code'] != 0:
  465. return response.json(500, res={'msg': 'discover device error'})
  466. uid_arr = res_json['result']['uid_arr']
  467. rtko = tkObject(rank=1)
  468. now_time = int(time.time())
  469. user.uid_rtsp.clear()
  470. res_json = []
  471. uid_rtsp_id_list = []
  472. for uid_a in uid_arr:
  473. uid = uid_a['uid']
  474. serial_number = uid_a['serial_number']
  475. nick = uid_a['nick']
  476. rtsp_url = rtko.encrypt(data=uid)
  477. try:
  478. uid_rtsp_qs = UidRtspModel.objects.get(uid=uid)
  479. except UidRtspModel.DoesNotExist:
  480. uid_rtsp_qs = UidRtspModel.objects.create(
  481. user_id=userID, uid=uid, serial_number=serial_number, nick=nick, region=region_code,
  482. password=uid_a['password'], rtsp_url=rtsp_url, addTime=now_time, updTime=now_time)
  483. else:
  484. uid_rtsp_qs.nick = nick
  485. uid_rtsp_qs.region = region_code
  486. uid_rtsp_qs.password = uid_a['password']
  487. uid_rtsp_qs.user_id = userID
  488. uid_rtsp_qs.serial_number = serial_number
  489. uid_rtsp_qs.updTime = now_time
  490. uid_rtsp_qs.save()
  491. uid_rtsp_id_list.append(uid_rtsp_qs.id)
  492. rtsp_domain = RESP_SERVER_DOMAIN_DATA[region_code]
  493. rtsp_uri = '{}://{}:443/{}'.format(RTSP_PREFIX, rtsp_domain, rtsp_url)
  494. ur_data = {
  495. 'uri': rtsp_uri,
  496. 'endpointId': uid,
  497. 'friendlyName': nick,
  498. 'manufacturerName': skill_name,
  499. 'description': 'Camera connected via {}'.format(skill_name),
  500. 'protocols': ['RTSP'],
  501. 'audioCodecs': ['G711'],
  502. 'videoCodecs': ['H264'],
  503. 'authorizationTypes': ['NONE'],
  504. 'resolutions': [{'width': 640, 'height': 360}],
  505. }
  506. res_json.append(ur_data)
  507. user.uid_rtsp.add(*uid_rtsp_id_list)
  508. logger.info('搜索设备返回值: {}'.format(res_json))
  509. return JsonResponse(res_json, safe=False)
  510. class oa2DiscoverySwitch(TemplateView):
  511. def post(self, request, *args, **kwargs):
  512. request.encoding = 'utf-8'
  513. request_dict = request.POST
  514. return self.validate(request_dict)
  515. def get(self, request, *args, **kwargs):
  516. request.encoding = 'utf-8'
  517. request_dict = request.GET
  518. return self.validate(request_dict)
  519. def validate(self, request_dict):
  520. # 增加对code和client_id的校验代码,返回access_token和refresh_token
  521. skill_name = request_dict.get("skill_name", 'loocam')
  522. access_token = request_dict.get("access_token", None)
  523. logger = logging.getLogger('django')
  524. logger.info('--------{} 开始搜索设备--------'.format(skill_name))
  525. user_qs = UserModel.objects.filter(access_token=access_token)
  526. if not user_qs.exists():
  527. return JsonResponse({'错误': '用户数据不存在'})
  528. response = ResObject()
  529. user = user_qs.first()
  530. userID = user.userID
  531. region_code = user.region_code
  532. logger.info('userID: {},region_code:{}'.format(userID, region_code))
  533. # 更新事件网关接口
  534. alexAuth = AlexaAuthModel.objects.filter(token=access_token, skill_name=skill_name).order_by('-addTime')
  535. if alexAuth.exists():
  536. auth_res = alexAuth.values()
  537. event_access_token = auth_res[0]['access_token']
  538. event_refresh_token = auth_res[0]['refresh_token']
  539. event_token = auth_res[0]['token']
  540. event_expiresTime = auth_res[0]['expiresTime']
  541. event_addTime = auth_res[0]['addTime']
  542. event_updTime = auth_res[0]['updTime']
  543. event_alexa_region = auth_res[0]['alexa_region']
  544. AlexaAuthModel.objects.filter(userID=userID).delete()
  545. alexAuth.delete()
  546. logger.info('update_event_access_token')
  547. logger.info(event_token)
  548. AlexaAuthModel.objects.create(
  549. userID=userID,
  550. access_token=event_access_token,
  551. refresh_token=event_refresh_token,
  552. token=event_token,
  553. expiresTime=event_expiresTime,
  554. addTime=event_addTime,
  555. updTime=event_updTime,
  556. alexa_region=event_alexa_region,
  557. skill_name=skill_name,
  558. )
  559. domain_name = SERVER_PREFIX_EU if region_code == 'EU' else SERVER_PREFIX
  560. auth_request_url = '{}/oalexa/discoveryswitch'.format(domain_name)
  561. requests_data = {'sid': 'admin', 'sst': 'admin', 'alexa_user_id': userID}
  562. res = requests.post(url=auth_request_url, data=requests_data)
  563. res_json = res.json()
  564. logger.info('{}正式服务器响应: {}'.format(region_code, res_json))
  565. # 添加测试服务器测试
  566. if res_json['result_code'] != 0:
  567. auth_request_url = '{}/oalexa/discoveryswitch'.format(SERVER_PREFIX_TEST)
  568. res = requests.post(url=auth_request_url, data=requests_data)
  569. res_json = res.json()
  570. logger.info('请求服务器url: {}'.format(auth_request_url))
  571. logger.info('服务器响应: {}'.format(res_json))
  572. if res_json['result_code'] != 0:
  573. return response.json(0, res={'msg': 'error'})
  574. switch_list = res_json['result']['switch_list']
  575. now_time = int(time.time())
  576. res_json = []
  577. for switch in switch_list:
  578. serial_number = switch['uid']
  579. nick = switch['nick']
  580. switch_info_qs = SwitchModel.objects.filter(serial_number=serial_number, userID=userID)
  581. if switch_info_qs.exists():
  582. switch_info_qs.update(nick=nick, region=switch['region'], updTime=now_time)
  583. else:
  584. SwitchModel.objects.create(serial_number=serial_number, nick=nick, region=switch['region'],
  585. addTime=now_time, updTime=now_time, userID=userID)
  586. ur_data = {
  587. 'endpointId': serial_number,
  588. 'friendlyName': nick,
  589. 'manufacturerName': skill_name,
  590. 'description': 'Plug connected via {}'.format(skill_name),
  591. 'modelName': 'P1425-LE',
  592. 'manufacturerId': 'zosi-ACCC8E5E7513',
  593. }
  594. res_json.append(ur_data)
  595. logger.info('搜索设备返回值: {}'.format(res_json))
  596. return JsonResponse(res_json, safe=False)
  597. class powerController(TemplateView):
  598. def post(self, request, *args, **kwargs):
  599. request.encoding = 'utf-8'
  600. request_dict = request.POST
  601. return self.power_controller(request_dict)
  602. def get(self, request, *args, **kwargs):
  603. request.encoding = 'utf-8'
  604. request_dict = request.GET
  605. return self.power_controller(request_dict)
  606. def power_controller(self, request_dict):
  607. serial_number = request_dict.get('serial_number', '')
  608. access_token = request_dict.get('access_token', '')
  609. skill_name = request_dict.get('skill_name', 'loocam')
  610. power_controller = request_dict.get('power_controller', '')
  611. if not all([serial_number, access_token, skill_name]):
  612. return JsonResponse({'result_code': '500', '错误': '缺少参数'})
  613. user_qs = UserModel.objects.filter(access_token=access_token)
  614. if not user_qs.exists():
  615. return JsonResponse({'result_code': '500', '错误': '用户数据不存在'})
  616. userID = user_qs.first().userID
  617. switch_qs = SwitchModel.objects.filter(serial_number=serial_number, userID=userID).values('nick', 'region')
  618. if not switch_qs.exists():
  619. return JsonResponse({'result_code': '500', '错误': 'serial_number数据不存在'})
  620. logger = logging.getLogger('django')
  621. logger.info('{} 控制插座 {}'.format(skill_name, serial_number))
  622. # 请求MQTT发布消息
  623. region = switch_qs[0]['region']
  624. if region == 'EU':
  625. domain_name = SERVER_PREFIX_EU
  626. elif region == 'CN':
  627. domain_name = SERVER_PREFIX_TEST
  628. else:
  629. domain_name = SERVER_PREFIX
  630. url = '{}/api/loocam/open/socket/alexa-socket-switch'.format(domain_name)
  631. requests_data = {'serial_number': serial_number, 'power_controller': power_controller} # TurnOn, TurnOff
  632. r = requests.post(url, requests_data)
  633. if r.status_code != 200:
  634. return JsonResponse({'result_code': '500', '错误': '请求服务器响应异常'})
  635. res = r.json()
  636. logger.info('{}服务器返回状态: {}'.format(region, res))
  637. if res['result_code'] != 0:
  638. return JsonResponse({'result_code': '500', '错误': '请求MQTT下发指令失败'})
  639. else:
  640. logger.info('请求MQTT下发指令成功')
  641. return JsonResponse({'result_code': '0'})
  642. class RtcController(TemplateView):
  643. def get(self, request, *args, **kwargs):
  644. request.encoding = 'utf-8'
  645. operation = kwargs.get('operation')
  646. return self.validation(request.GET, operation)
  647. def post(self, request, *args, **kwargs):
  648. request.encoding = 'utf-8'
  649. operation = kwargs.get('operation')
  650. return self.validation(request.POST, operation)
  651. def validation(self, request_dict, operation):
  652. if operation == 'rtc':
  653. return self.rtc(request_dict)
  654. @classmethod
  655. def rtc(cls, request_dict):
  656. uid = request_dict.get("uid")
  657. access_token = request_dict.get("access_token")
  658. skill_name = request_dict.get("skill_name")
  659. offer_sdp = request_dict.get("offer_sdp")
  660. if not all([uid, access_token, skill_name, offer_sdp]):
  661. return JsonResponse({'错误': '缺少参数'})
  662. user_qs = UserModel.objects.filter(access_token=access_token)
  663. if not user_qs.exists():
  664. return JsonResponse({'错误': '用户数据不存在'})
  665. ur_qs = UidRtspModel.objects.filter(uid=uid).values('user_id', 'uid', 'nick', 'rtsp_url', 'password')
  666. if not ur_qs.exists():
  667. return JsonResponse({'错误': 'uid数据不存在'})
  668. logger = logging.getLogger('django')
  669. try:
  670. # 请求go2rtc创建流
  671. rtsp = GO2RTC_RTSP.format(uid)
  672. rtc_url = GO2RTC_API.format('api/streams')
  673. params = {
  674. 'src': rtsp,
  675. 'name': uid,
  676. }
  677. r = requests.put(url=rtc_url, data=params, timeout=30)
  678. assert r.status_code == 200
  679. # mqtt下发推流指令
  680. logger.info('------rtc开始向设备下发推流指令:{}------'.format(uid))
  681. # 确认设备的用户地区
  682. region = 'US'
  683. user_id = ur_qs[0]['user_id']
  684. user_qs = UserModel.objects.filter(userID=user_id).values('region_code')
  685. if user_qs.exists():
  686. region = user_qs[0]['region_code']
  687. # 根据用户地区确认域名
  688. domain_name = SERVER_PREFIX_EU if region == 'EU' else SERVER_PREFIX
  689. # 请求MQTT发布消息
  690. url = '{}/iot/requestPublishMessage'.format(domain_name)
  691. requests_data = {'UID': uid, 'rtsp': rtsp, 'enable': '1'} # 1: 开始推流,0: 停止推流; channel: 推流通道
  692. r = requests.post(url, requests_data)
  693. assert r.status_code == 200
  694. res = r.json()
  695. logger.info('rtc请求MQTT发布消息参数:{},result_code: {}'.format(requests_data, res['result_code']))
  696. if res['result_code'] == 0:
  697. logger.info('rtc请求MQTT下发指令成功,地区:{}'.format(region))
  698. elif res['result_code'] == 10044:
  699. url = '{}/iot/requestPublishMessage'.format(SERVER_PREFIX_TEST) # 测试服务器
  700. r = requests.post(url, requests_data)
  701. assert r.status_code == 200
  702. res = r.json()
  703. if res['result_code'] == 0:
  704. logger.info('请求MQTT下发指令成功---测试服')
  705. # 获取SDP
  706. sdp = cls.handle_alexa_offer(offer_sdp, rtsp)
  707. # params = {
  708. # 'src': uid,
  709. # }
  710. # r = requests.get(url=rtc_url, params=params, timeout=30)
  711. # assert r.status_code == 200
  712. # res = r.json()
  713. # # 遍历producers数组,查找包含sdp字段的对象
  714. # sdp = ''
  715. # for producer in res['producers']:
  716. # if 'sdp' in producer:
  717. # sdp = producer['sdp']
  718. # break
  719. res_json = {
  720. 'SDP': sdp
  721. }
  722. logger.info('------------rtc响应数据---------------: {}'.format(res_json))
  723. return JsonResponse(res_json, safe=False)
  724. except Exception as e:
  725. logger.info('rtc异常,error_line:{},error_msg:{}'.format(e.__traceback__.tb_lineno, repr(e)))
  726. return JsonResponse({'错误': 'rtc异常'})
  727. class VesseTest(TemplateView):
  728. def get(self, request, *args, **kwargs):
  729. request.encoding = 'utf-8'
  730. operation = kwargs.get('operation')
  731. return self.validation(request.GET, operation, request)
  732. def post(self, request, *args, **kwargs):
  733. request.encoding = 'utf-8'
  734. operation = kwargs.get('operation')
  735. return self.validation(request.POST, operation, request)
  736. def validation(self, request_dict, operation, request):
  737. response = ResponseObject()
  738. if operation == 'get-token': # 获取token
  739. return self.get_token(response)
  740. elif operation == 'get-user-id': # 获取userid
  741. return self.get_user_id(request_dict, response, request)
  742. elif operation == 'get-user-info': # 获取用户信息
  743. return self.get_user_info(request_dict, response, request)
  744. elif operation == 'rtc': # rtc测试
  745. return self.rtc(request_dict, response, request)
  746. def get_token(self, response):
  747. data = {
  748. 'corpid': 'ww467ec1685e8262e6',
  749. 'corpsecret': 'IeUoaQ-0hEhEduCQq1zyfVXjfeZpMsThK1nklszRzUY'
  750. }
  751. try:
  752. token_url = 'https://qyapi.weixin.qq.com/cgi-bin/gettoken'
  753. token_response = requests.get(token_url, params=data)
  754. if token_response.status_code == 200:
  755. data = token_response.json()
  756. access_token = data.get("access_token")
  757. if access_token:
  758. return response.json(0, {'access_token': access_token})
  759. else:
  760. return response.json(0, "Failed to get AccessToken.")
  761. else:
  762. code = token_response.status_code
  763. content = token_response.content
  764. return response.json(code, content)
  765. except requests.exceptions.RequestException as e:
  766. print(e)
  767. return response.json(500, repr(e))
  768. def get_user_id(self, request_dict, response, request):
  769. access_token = request_dict.get('access_token', None)
  770. code = request_dict.get('code', None)
  771. if not all([access_token, code]):
  772. return response.json(444, 'error: access_token, code')
  773. data = {
  774. 'access_token': access_token,
  775. 'code': code
  776. }
  777. try:
  778. token_url = 'https://qyapi.weixin.qq.com/cgi-bin/auth/getuserinfo'
  779. token_response = requests.get(token_url, params=data)
  780. if token_response.status_code == 200:
  781. data = token_response.json()
  782. userid = data.get("userid")
  783. if userid:
  784. return response.json(0, {'userid': userid})
  785. else:
  786. code = token_response.status_code
  787. content = token_response.content
  788. return response.json(code, content)
  789. except Exception as e:
  790. print(e)
  791. return response.json(500, repr(e))
  792. def get_user_info(self, request_dict, response, request):
  793. access_token = request_dict.get('access_token', None)
  794. userid = request_dict.get('userid', None)
  795. if not all([access_token, userid]):
  796. return response.json(444, 'error: access_token, userid')
  797. data = {
  798. 'access_token': access_token,
  799. 'userid': userid
  800. }
  801. try:
  802. token_url = 'https://qyapi.weixin.qq.com/cgi-bin/user/get'
  803. token_response = requests.get(token_url, params=data)
  804. if token_response.status_code == 200:
  805. data = token_response.json()
  806. userid = data.get("userid")
  807. name = data.get('name')
  808. position = data.get('position')
  809. status = data.get('status')
  810. if userid:
  811. url_data = {
  812. 'userid': userid,
  813. 'name': name,
  814. 'position': position,
  815. 'status': status
  816. }
  817. return response.json(0, {'data': url_data})
  818. else:
  819. code = token_response.status_code
  820. content = token_response.content
  821. return response.json(code, content)
  822. except Exception as e:
  823. print(e)
  824. return response.json(500, repr(e))
  825. @staticmethod
  826. def rtc(request_dict, response, request):
  827. uid = request_dict.get('uid', None)
  828. try:
  829. # 请求go2rtc创建流
  830. # rtsp = GO2RTC_RTSP.format(uid)
  831. rtc_url = GO2RTC_API.format('api/streams')
  832. # params = {
  833. # 'src': rtsp,
  834. # 'name': uid,
  835. # }
  836. # r = requests.put(url=rtc_url, params=params, timeout=30)
  837. # assert r.status_code == 200
  838. # 获取SDP
  839. params = {
  840. 'src': uid,
  841. }
  842. r = requests.get(url=rtc_url, params=params, timeout=30)
  843. assert r.status_code == 200
  844. res = r.json()
  845. # 遍历producers数组,查找包含sdp字段的对象
  846. sdp = ''
  847. for producer in res['producers']:
  848. if 'sdp' in producer:
  849. sdp = producer['sdp']
  850. break
  851. res_json = {
  852. 'SDP': sdp
  853. }
  854. return response.json(0)
  855. except Exception as e:
  856. print(e)
  857. return response.json(500, repr(e))
  858. logger = logging.getLogger('django')
  859. def get_uuid():
  860. return str(uuid.uuid4())