import re import requests import time import ast import os import datetime import uuid from PIL import Image from io import BytesIO from django.conf import settings from django.db.models import Count from django.core.paginator import Paginator from django.db import transaction from smartfarming.api.views.forecast.all_dict import insect_dict, attract_discern from smartfarming.models.sim_card import MongoMsg_Conf from smartfarming.models.device import MongoDevice from smartfarming.models.camera import MongoCameraPhoto, MongoCameraData, MongoCameraAccount from smartfarming.serializers.device_serializers import DeviceSerializers from smartfarming.utils import bindDeviceLive, getLive, controlMove, modify_live_time, off_on_live, setDeviceSnap, getKitToken, getRecord from kedong.tools import RedisPool from kedong.decoration import kedong_deco, PortError from smartfarming.api.views.camera.cameraserializers import ( MongoCameraPhotoSerializers, MongoDeviceCameraSerializers ) config_dict = settings.CONFIG @kedong_deco(login_required=True) def list_camera(request): # 获取监控设备列表 post_info = request.POST device_id = post_info.get('device_id') page_num = int(post_info.get('page', 1)) page_size = int(post_info.get('page_size', 10)) queryset = MongoDevice.objects.filter(device_type_id=6).order_by("-uptime") if device_id: queryset = queryset.filter(device_id=device_id) paginator = Paginator(queryset, page_size) page_obj = paginator.get_page(page_num) serializers = MongoDeviceCameraSerializers(page_obj, many=True) datas = {"data": serializers.data, "counts": queryset.count()} return datas @kedong_deco(login_required=True) def addr_camera_list(request): """ 获取监控设备播放地址 参数: device_id 必传 (string) 监控设备的序列号 返回值: { "data": { "hls": "http://test.ys7.com:66/openlive/b475942b03544a5aa35b651670d8712c.m3u8", HLS流畅直播地址 "hlsHd": "http://test.ys7.com:66/openlive/b475942b03544a5aa35b651670d8712c.hd.m3u8", HLS高清直播地址 "rtmp": "rtmp://10.82.0.190:1935/openlive/b475942b03544a5aa35b651670d8712c", RTMP流畅直播地址 "rtmpHd": "rtmp://10.82.0.190:1935/openlive/b475942b03544a5aa35b651670d8712c.hd" RTMP高清直播地址 } "errorCode": 0, "message": "", "formError": {} } """ try: if 'HTTP_X_FORWARDED_FOR' in request.META: ip = request.META['HTTP_X_FORWARDED_FOR'] else: ip = request.META['HTTP_X_REAL_IP'] except Exception as KeyError: ip = request.META['REMOTE_ADDR'] device_id = request.POST.get('device_id') try: addr = MongoCameraData.objects.get(device_id=device_id).device_info except: addr = {} return addr @kedong_deco(login_required=True) def ctrl_camera(request): """ 控制监控设备转动 : 参数: device_id 必传 (string) 监控设备的序列号 ctrl 必传 (string) 操作指令 move 控制球机转动 stop 停止球机转动 movenum ctrl=move时必传(string) 0-上,1-下,2-左,3-右,8-放大,9-缩小 返回值: { "data": success, success成功 default失败 "errorCode": 0, "message": "", "formError": {} } """ device_id = request.POST.get('device_id') ctrl = request.POST.get('ctrl') account_id = MongoCameraData.objects.filter(device_id=device_id).first() if not account_id: raise PortError("","未找到此设备") camera_query = MongoCameraAccount.objects.filter(id=int(account_id.account_id)).first() if not camera_query: raise PortError("","未找到此账号") channelNo = "1" if device_id.find("-") != -1: channelNo = device_id.split("-")[1] device_id = device_id.split("-")[0] # 操作命令:0-上,1-下,2-左,3-右,4-左上,5-左下,6-右上,7-右下,8-放大,9-缩小,10-近焦距,11-远焦距 if ctrl == "move": dire = request.POST.get('movenum') # 增加大华相机操作功能 if camera_query.account_type == 1: if dire not in ("0","1","2","3","8","9"): raise PortError("movenum","暂不支持其它指令") appId = camera_query.app_key appSecret = camera_query.app_secret result = controlMove(appId,appSecret,device_id,dire,channelNo) code = result.get("result").get("code") if code == "0": return True else: return {"errorCode": 2, "msg": result.get("result").get("msg")} else: accessToken = camera_query.token ctrl_url = 'https://open.ys7.com/api/lapp/device/ptz/start' speed = '1' data = {'accessToken':accessToken, 'deviceSerial':device_id, 'channelNo':channelNo, 'speed':speed, 'direction':dire} res1 = requests.post(ctrl_url,data=data) elif ctrl == "stop": time.sleep(0.4) # 增加大华相机操作功能 if camera_query.account_type == 1: dire = "10" appId = camera_query.app_key appSecret = camera_query.app_secret result = controlMove(appId,appSecret,device_id,dire,channelNo) code = result.get("result").get("code") if code == "0": return True else: return {"errorCode": 2, "msg": result.get("result").get("msg")} else: accessToken = camera_query.token ctrl_url = 'https://open.ys7.com/api/lapp/device/ptz/stop' data = {'accessToken':accessToken, 'deviceSerial':device_id, 'channelNo':channelNo} res2 = requests.post(ctrl_url,data=data) return True @kedong_deco(login_required=False) def see_camera_account(request): """ 查看当前绑定的萤石云账号信息\按业务员名字搜索\萤石云账号搜索 参数： account 非必传(str) 按萤石云账号名字搜索 salesman 非必传(str) 按业务员名字搜索 说明：两个参数都不传为所有数据，不进行筛选操作 返回值： { "errorCode": 0, "message": "", "formError": {}, "data": [ { "id": 1, 萤石云账号id "account": "hnyf" 萤石云名称 } ], "params": {} } """ account = request.POST.get("account") salesman = request.POST.get("salesman") camera_query = MongoCameraAccount.objects.all().order_by("-id") if account: camera_query = camera_query.filter(account__icontains=account) if salesman: camera_query = camera_query.filter(salesman__icontains=salesman) data = [{'id':i.id,'account':i.account} for i in camera_query] return data @kedong_deco(login_required=True) def add_camera_account(request): """ 添加萤石云账号 管理员可操作 参数： account 必传(str) 萤石云账号 salesman 必传(str) 萤石云账号所相关的业务员名称 app_token 必传(str) 萤石云账号 token app_key 必传(str) 萤石云账号 appkey app_secret 必传(str) 萤石云账号 appsecret 2022-08-19 08:54 csx 增加监控账户区分,在不影响原来的功能上进行修改,增加大华乐橙云账号添加 account_type 非必传(number) 监控设备类型 0代表海康 1代表大华,默认是0 """ account = request.POST.get("account") salesman = request.POST.get("salesman") token = request.POST.get("app_token") app_key = request.POST.get("app_key") app_secret = request.POST.get("app_secret") account_type = request.POST.get("account_type","0") if account_type not in ("0","1"): raise PortError("","无法识别此监控账号类型") if not all([account,salesman,token,app_key,app_secret]): raise PortError("","请把参数填写完整") now_time = int(time.time()) camera_query = MongoCameraAccount.objects.filter(app_key=app_key,account_type=account_type).first() if camera_query: raise PortError("","此appkey已存在,请勿重复添加") try: MongoCameraAccount.objects.create(account=account,salesman=salesman,token=token,addtime=now_time,uptime=now_time, app_key=app_key,app_secret=app_secret,account_type=account_type) except: raise PortError("","添加萤石账号失败") return True @transaction.atomic @kedong_deco(login_required=True) def add_camera(request): """ 添加监控设备接口: 参数： device_id 必传 (string) 监控设备的序列号 camera_channel 必传 (string) 监控通道号 camera_playback 必传 (string) 是否支持回放功能,0不支持 1支持 新加参数 account_id 必传 (string) 萤石云账号id 2022-08-19 08:54 csx 增加监控账户区分,在不影响原来的功能上进行修改,增加大华乐橙云监控绑定 返回值： { "data": null, "errorCode": 0, "message": , "formError": {} } """ device_id = request.POST.get('device_id') camera_channel = request.POST.get("camera_channel","0") camera_playback = request.POST.get("camera_playback","0") account_id = request.POST.get("account_id",1) if not account_id.isdigit(): raise PortError("account_id","无效参数") if not device_id: raise PortError("device_id","未传设备号") if not camera_channel.isdigit(): raise PortError("camera_channel","无效参数") if camera_playback not in ("0","1"): raise PortError("camera_playback","参数超出范围") now_time = int(time.time()) camera_query = MongoCameraAccount.objects.filter(id=int(account_id)).first() if not camera_query: raise PortError("","未找到此账号") device_ids = device_id + "-" + camera_channel # if MongoCameraData.objects.filter(device_id=device_id).exists() or MongoDevice.objects.filter(device_id=device_id).exists(): # raise PortError("","设备已存在，请勿重复添加") # 使用的是大华乐橙云账号监控接入,监控通道号大华的从0开始 coverUrl = "" if camera_query.account_type == 1: appId = camera_query.app_key appSecret = camera_query.app_secret device_live_data = bindDeviceLive(appId, appSecret, device_id,camera_channel) code = device_live_data["result"]["code"] if code == "0": live_path = device_live_data["result"]["data"]["streams"][0]["hls"] elif code == "LV1001": # 如直播地址已存在，则需要查询直播地址列表 live_data = getLive(appId, appSecret, device_id, camera_channel) coverUrl = live_data["result"]["data"]["streams"][0]["coverUrl"] if live_data["result"]["code"] == "0": try: live_path = live_data["result"]["data"]["streams"][0]["hls"] except: raise PortError("","获取直播地址失败") else: raise PortError("",live_data) else: msg = device_live_data["result"]["msg"] raise PortError("", msg) device_info = { "hls": live_path, "hlsHd": live_path, "rtmp": live_path, "rtmpHd": live_path } else: accessToken = camera_query.token url = config_dict.get("camera").get("ys_open") data = {"accessToken":accessToken,"source":"%s:%s"%(device_id,str(camera_channel))} res = requests.post(url,data=data) if ast.literal_eval(res.text)["code"] == "200": url_1 = config_dict.get("camera").get("ys_get") res_1 = requests.post(url_1,data=data) try: addr = ast.literal_eval(res_1.text)["data"][0] device_info = { "hls": addr["hls"], "hlsHd": addr["hlsHd"], "rtmp": addr["rtmp"], "rtmpHd": addr["rtmpHd"] } except Exception as e: raise PortError("device_id", "解析结果异常") else: raise PortError("","Token过期,设备激活失败") try: MongoDevice.objects.create(device_id=device_ids,device_type_id=6,addtime=now_time,device_code=1,uptime=now_time, jk_live_img=coverUrl) MongoCameraData.objects.create(device_id=device_ids,device_info=device_info,addtime=now_time,camera_playback=camera_playback,account_id=account_id) except Exception as e: print(e) raise PortError(" ", "添加设备失败") return True @kedong_deco(login_required=True) def camera_config(request): """ 监控设置 参数: device_name 必传(str) 设备名称 off_on 必传(str) 设备开关 off 关 on 开 start_time 非必传(str) 开始时间 为开时 必传 end_time 非必传(str) 结束时间 为开时 必传 device_id 必传(str) 设备号 """ device_id = request.POST.get("device_id") off_on = request.POST.get("off_on","") start_time = request.POST.get("start_time","") end_time = request.POST.get("end_time","") device_ids = request.POST.get("device_id") device_name = request.POST.get("device_name") redis_pool = RedisPool().get_redis_pool(settings.redis_db["captcha"]) try: device_query = MongoDevice.objects.get(device_id=device_ids) except: raise PortError("","未找出此设备") account_id = MongoCameraData.objects.filter(device_id=device_ids).first() if not account_id: raise PortError("","未找到此设备") camera_query = MongoCameraAccount.objects.filter(id=int(account_id.account_id)).first() if not camera_query: raise PortError("","未找到此账号") channelNo = "1" if device_ids.find("-") != -1: channelNo = device_ids.split("-")[1] device_id = device_ids.split("-")[0] else: device_id = device_ids if camera_query.account_type == 1: appId = camera_query.app_key appSecret = camera_query.app_secret live_token = redis_pool.get(device_ids) if not live_token: live_data = getLive(appId, appSecret, device_id, channelNo) if live_data["result"]["code"] == "0": try: live_token = live_data["result"]["data"]["streams"][0]["liveToken"] redis_pool.set(device_ids,live_token,60*60*23) except: raise PortError("","获取liveToken失败") else: raise PortError("",live_data) else: raise PortError("","其它设备没有此功能") if off_on == "on": if not start_time and not end_time: status = modify_live_time(appId,appSecret,live_token,start_time,end_time,"always") else: # 时间内开启直播 status = modify_live_time(appId,appSecret,live_token,start_time,end_time) elif off_on == "off": status = off_on_live(appId,appSecret,live_token,off_on) else: raise PortError("","参数超出范围") if status["result"]["code"] == "0": device_query.device_name = device_name device_query.save() config = off_on + "-" + start_time + "-" + end_time redis_pool.set(device_ids+"config",config) return True else: return status @kedong_deco(login_required=True) def camera_takephoto(request): # 监控拍照接口 device_id = request.POST.get("device_id") if not device_id: raise PortError("","设备号未传") try: d_id = MongoDevice.objects.get(device_id=device_id).id except: raise PortError("","未找出此设备") account_id = MongoCameraData.objects.filter(device_id=device_id).first() if not account_id: raise PortError("","未找到此设备") camera_query = MongoCameraAccount.objects.filter(id=int(account_id.account_id)).first() if not camera_query: raise PortError("","未找到此账号") channelNo = "0" if device_id.find("-") != -1: channelNo = device_id.split("-")[1] device_id = device_id.split("-")[0] else: device_id = device_id appId = camera_query.app_key appSecret = camera_query.app_secret snap_url = setDeviceSnap(appId, appSecret, device_id, channelNo) print(snap_url) # snap_url = "https://lechangecloud.oss-cn-hangzhou.aliyuncs.com/lechange/snappic/8H09715PHA657EC_img/Alarm/0/9d9913c5112c4587a391b6c454110c5a.jpg?Expires=1690882790&OSSAccessKeyId=LTAIP4igXeEjYBoG&Signature=RK97RSfcLjyA84Fv8aFOtA%2BvGao%3D" # snap_url = "https://lechangecloud.oss-cn-hangzhou.aliyuncs.com/lechange/snappic/8H09715PHA24E37_img/Alarm/0/ab85f10f33a34f068dc55a4a64e51584.jpg?Expires=1690883565&OSSAccessKeyId=LTAIP4igXeEjYBoG&Signature=%2FP1Bxksx0lagn24gkwaXWcS9Dno%3D" time.sleep(5) if snap_url: # 图片保存位置 config = settings.CONFIG save_path = "/jiankong/" + device_id + "/" + time.strftime("%Y%m%d", time.localtime()) camera_path = config.get("media") + save_path os.makedirs(camera_path) if not os.path.exists(camera_path) else None stamp = int(time.time()) unique_id = uuid.uuid4() combined_id = str(stamp) + "-" + str(unique_id) camera_photos = combined_id + ".jpg" img_path = os.path.join(camera_path, camera_photos) # 下载图片 img_query = requests.get(snap_url, timeout=5) if img_query.status_code == 200: img = Image.open(BytesIO(img_query.content)) img.save(img_path) camera_obj = MongoCameraPhoto.objects.create( device_id=d_id, addr=f"/media{save_path}/{camera_photos}", photo_status=1, addtime=int(time.time()) ) if camera_obj: return {"addr":camera_photos} else: raise PortError("","保存图片失败,稍后重试") else: raise PortError("","下载图片失败,稍后重试") else: raise PortError("","拍照失败,稍后重试") @kedong_deco(login_required=True) def camera_photo(request): # 拍照列表 request_data = request.POST device_id = request_data.get("device_id") start_time = request_data.get("start_time","") end_time = request_data.get("end_time","") page_num = int(request_data.get("page")) if request_data.get("page") else 1 page_size = int(request_data.get("pagesize")) if request_data.get("pagesize") else 10 queryset = MongoCameraPhoto.objects.filter(device_id=device_id).order_by("-addtime") if start_time and end_time: queryset = queryset.filter(addtime__gte=start_time, addtime__lte=end_time) paginator = Paginator(queryset, page_size) page_obj = paginator.get_page(page_num) serializers = MongoCameraPhotoSerializers(page_obj, many=True) return {"data": serializers.data, "counts": queryset.count()} @kedong_deco(login_required=True) def get_camera_dh_tokens(request): """ 获取大华相机token 参数: device_id 必传(string) 设备号 """ device_id = request.POST.get("device_id") if not device_id: raise PortError("","参数缺失") camera_query = MongoCameraData.objects.filter(device_id=device_id).first() if not camera_query: raise PortError("","未找到此设备") camera_account_query = MongoCameraAccount.objects.filter(id=camera_query.account_id).first() if not camera_account_query: raise PortError("","此设备未绑定监控账号") if camera_account_query.account_type == 0: raise PortError('', '海康监控暂不支持获取token') if device_id.find("-") != -1: channelNo = device_id.split("-")[1] device_id = device_id.split("-")[0] appId = camera_account_query.app_key appSecret = camera_account_query.app_secret like_token_data = getKitToken(appId, appSecret,device_id,channelNo) return like_token_data @kedong_deco(login_required=True) def get_camera_dh_token(request): """ 获取大华相机token 参数: device_id 必传(string) 设备号 """ device_id = request.POST.get("device_id") if not device_id: raise PortError("","参数缺失") camera_query = MongoCameraData.objects.filter(device_id=device_id).first() if not camera_query: raise PortError("","未找到此设备") camera_account_query = MongoCameraAccount.objects.filter(id=camera_query.account_id).first() if not camera_account_query: raise PortError("","此设备未绑定监控账号") if camera_account_query.account_type == 0: raise PortError('', '海康监控暂不支持获取token') if device_id.find("-") != -1: channelNo = device_id.split("-")[1] device_id = device_id.split("-")[0] appId = camera_account_query.app_key appSecret = camera_account_query.app_secret like_token_data = getRecord(appId, appSecret,device_id,channelNo) return like_token_data