# -*- coding:utf-8 -*- # @Time : 2020/8/25 9.45 上午 # @Author : Creat by cao import operator import json import time import os import requests import datetime import uuid import ast from django.conf import settings 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.worm_forecast import MongoCBDphoto, MongoCBDphoto_B, MongoCBDphoto_C from kedong.decoration import kedong_deco, PortError from .common import * from django.db.models import Q config_dict = settings.CONFIG @kedong_deco(login_required=True) def time_folder(request): """ 测报灯/色诱/吸虫塔/性诱3.0隐藏时间文件夹接口 device_id 必传(str) 设备id identify_model 必传(str) 识别模型AB 返回值： "data":[ { "2021":{ "10":[ "14", "13", ], "09":[ "30", "29", "28", "27", "16", "06" ] } } ], """ device_id = request.POST.get('device_id') if not device_id: raise PortError("device_id", "未传设备号") try: device = MongoDevice.objects.get(device_id=device_id) except: raise PortError("", "设备号不对") myuser_type = request.user_type if device.device_type_id == 3: identify_model = request.POST.get("identify_model","A") if identify_model=="A": models = MongoCBDphoto elif identify_model=="B": models = MongoCBDphoto_B if not myuser_type == 1 and device.device_expire == "1": photo_list = models.objects.filter(device_id=device.id,photo_status=1,addtime__lte=int(device.device_expire_time)).order_by("-id") else: photo_list = models.objects.filter(device_id=device.id,photo_status=1).order_by("-id") else: raise PortError("","暂不支持的设备类型") repeat_data = [] for i in photo_list: othertime = time.strftime("%Y-%m-%d", time.localtime(i.addtime)) year,month,day = othertime.split("-") time_tup = year,month,day repeat_data.append(time_tup) data = list(set(repeat_data)) data.sort(key=repeat_data.index) time_data = [] year_list = [] month_list = [] crowbar = -1 #使用撬棍进行列表下标标识,因为数据时经过去重和排序的,所以可以进行。 for time_i in data: year_time = time_i[0] month_time = time_i[1] day_time = time_i[2] if year_time in year_list: if year_time+month_time in month_list: time_data[crowbar][year_time][month_time].append(day_time) else: time_data[crowbar][year_time][month_time] = [day_time] month_list.append(year_time+month_time) else: year_list.append(year_time) month_list.append(year_time+month_time) month_dict = {month_time:[day_time]} time_data.append({year_time:month_dict}) crowbar+=1 return(time_data) @kedong_deco(login_required=True) def device_photo_details(request): """ 色诱测报灯/测报灯/吸虫塔/性诱3.0/图片详情接口 img_id 必传(str) 图片id cmd 非必传(str) cmd有值并且为sy时,是色诱测报图片列表接口, 为xct时,是吸虫塔图片列表, 为cbd是,是测报灯图片列表, 为xy_three 时, 是性诱3.0图片列表 identify_model 非必传("A","B") cmd为cbd时,A是A模型,B是B模型 返回值： "data": { "device_id": "2631", "addtime": 1632413928, "addr": "http://bigdata-all.oss-accelerate.aliyuncs.com/Basics/cbd/867435059571471/2021/9/24/20210924001848354843.jpg", "ids": 86524, "mark": "", "label": "", "indentify_photo": "http://bigdata-all.oss-accelerate.aliyuncs.com/Result/cbd/867435059571471/2021/9/24/20210924001848354843.jpg", "indentify_result": "158,38#670,26#672,4" }, """ img_id = request.POST.get("img_id") try: imgs = MongoCBDphoto.objects.get(id=img_id) except: raise PortError("img_id", "未找到该图片id") label = imgs.label indentify_photo = imgs.indentify_photo indentify_result = imgs.indentify_result img_url = imgs.addr mark = imgs.mark if mark: mark = json.loads(mark) else: mark = [] data= { "device_id":imgs.device_id, "addtime":imgs.addtime, "addr":img_url, "ids":imgs.id, "mark":mark, "indentify_photo":indentify_photo, "indentify_result":indentify_result, "label":label} return data @kedong_deco(login_required=True) def device_photo_list(request): """ 测报灯图片展示接口 device_id 必传(str) 设备id page 非必传(str) 页数 page_number 非必传(num) 页面显示数据总数,默认10条 time_begin 非必传(时间戳) 开始时间 time_end 非必传(时间戳) 结束时间 返回值： "data": { "disable":0 禁用识别的状态 1 开启识别的状态, { '2021-04-09': [{ 'device_id': '836', 'addtime': 1617936849, 'describe': '', 'addr': 'http://182.92.193.64:8003/Basics/cbd/866262046927462/2021/4/9/20210409105402642179.jpg', 'mark': None, 'id': 5877, 'add_time': '2021-04-09' }], '2021-04-08': [{ 'device_id': '836', 'addtime': 1617867971, 'describe': '', 'addr': 'http://182.92.193.64:8003/Basics/cbd/866262046927462/2021/4/8/20210408154604657028.jpg', 'mark': None, 'id': 5810, 'add_time': '2021-04-08' } } """ deviceas = request.POST device_id = deviceas.get("device_id") page = int(deviceas.get("page",1)) time_begin = deviceas.get("time_begin") time_end = deviceas.get("time_end") page_number = int(deviceas.get("page_number",12)) if not device_id: raise PortError("device_id", "未传设备号") try: device = MongoDevice.objects.get(device_id=device_id) except: raise PortError("", "设备号不对") data = [] image_path = config_dict["image_url"]["image"] photo_list = MongoCBDphoto.objects.filter(device_id=device.id,photo_status=1).order_by("-id") if time_begin and time_begin.isdigit() and time_end.isdigit(): photo_list = photo_list.filter(addtime__range=(int(time_begin),int(time_end))) num = photo_list.count() for i in photo_list[(page_number*(page-1)):(page*page_number)]: # 判断是否有手动标注结果 if i.is_mark == 1: try: mark = ast.literal_eval(i.mark) except: mark = json.loads(i.mark) pest_counts = len(mark) else: indentify_result = "" if i.indentify_result == "0" else i.indentify_result pest_counts = sum([int(i.split(",")[1]) for i in indentify_result.split("#")]) if indentify_result else 0 # TODO 展示图片暂时处理 if str(i.addr).startswith('http'): img_url = i.addr else: img_url = image_path + i.addr if str(i.indentify_photo).startswith('http'): indentify_photo = i.indentify_photo else: indentify_photo = image_path + i.indentify_photo data.append({ "device_id":i.device_id, "addtime":i.addtime, "describe":i.describe, "addr":img_url, "ids":i.id, "mark":i.mark, "pest_counts":pest_counts, "indentify_photo":indentify_photo, "indentify_result":i.indentify_result }) return {"disable":device.disable,"data":data,"num":num} @kedong_deco(login_required=True) def equip_photo_del(request): """ 测报灯/孢子仪/性诱/吸虫塔/性诱3.0图片删除接口 参数 device_id 必传(str) 设备号 addrlist 非必传(str) 图片列表 addrlist没值删除所有的图片,有值就删除里面的值 """ deviceas = request.POST device_id = deviceas.get("device_id") img_list = deviceas.get("addrlist") if not device_id: raise PortError("", "未传设备号") try: device_ids = MongoDevice.objects.get(device_id=device_id) except: raise PortError("", "设备号不存在") img_list = eval(img_list) if device_ids.device_type_id == 3: modules = MongoCBDphoto if img_list: for i in img_list: try: modules.objects.filter(id=i).update(photo_status=4) except: raise PortError("device_id", "删除失败") else: try: modules.objects.filter(device_id=device_ids.id).update(photo_status=4) except: raise PortError("device_id", "删除失败") return True # 测报灯短信配置 @kedong_deco(login_required=True) def cbd_msg_conf(request): """ 测报灯害虫预警短信配置接口： 参数： device_id 必传(str) 设备号 conf 非必传(str) 害虫配置信息 不传conf: 当前短信预警配置 传conf：修改短信预警配置 """ device_id = request.POST.get("device_id") if not device_id: raise PortError("device_id", "参数不对") uptime = int(time.time()) conf = request.POST.get("conf") if conf: try: if MongoMsg_Conf.objects.filter(device_id=device_id).exists(): msgconf = MongoMsg_Conf.objects.get(device_id=device_id) msgconf.conf = conf msgconf.save() else: MongoMsg_Conf.objects.create(device_id=device_id,conf=conf,uptime=uptime) except: raise PortError(" ", "保存失败") return True else: try: msgconf = MongoMsg_Conf.objects.get(device_id=device_id) data = msgconf.conf except: data = "" return data @kedong_deco(login_required=True) def equip_photo_discern(request): # 测报灯、吸虫塔、色诱测报害虫重新图片识别\详情\手动标注接口 img_id = request.POST.get("img_id") ret = request.POST.get("ret") if ret == "again": # 手动标注 # 获取该图片对象 photo_obj = MongoCBDphoto.objects.filter(id=img_id) photo_obj = photo_obj.first() if photo_obj else None date = [] if photo_obj: # 当图片对象存在 addr = photo_obj.addr if addr: # 该图片对应设备的信息 device = MongoDevice.objects.filter(id=photo_obj.device_id) device = device.first() if device else None # 获取图片本地路径 img_content_url = f'{config_dict.get("image_url").get("image")}/media' media = f'{config_dict.get("media")}' if addr.startswith("http"): addr = addr.replace(img_content_url,media) else: addr = addr.replace("/media",media) print(addr, "--------") pass # 重新调用识别接口 identify_url = config_dict.get("image_url").get("identify_url") if device.lng and device.lat: request_data = { "longitude": device.lng, "latitude": device.lat } else: request_data = { "longitude": "126.40548", "latitude": "47.82857" } if os.path.exists(addr): imageFile = { "imageFile": ('pic.jpg', open(addr, "rb"), 'image/jpeg') } res = requests.post(url=identify_url, data=request_data, files=imageFile, timeout=(10, 20)) image_result = json.loads(res.text) try: if image_result.get("returnResult").get("returnStatus") != "0": returnString = image_result["returnResult"]["returnString"] returnImageUrl = image_result["returnResult"]["returnImageUrl"] returnCode = image_result["returnResult"].get("returnCode","") remote_content = requests.get(returnImageUrl).content indentify_photo = photo_obj.indentify_photo # 如果第一次识别图片存在 if photo_obj: if indentify_photo.startswith("http"): indentify_photo = indentify_photo.replace(img_content_url,media) else: indentify_photo = indentify_photo.replace("/media",media) else: # 如果第一次识别图片不存在 inden_path = os.path.join(media, f"result/cbd/{device.device_id}") os.makedirs(inden_path) if not os.path.exists(inden_path) else None stamp = int(datetime.now().timestamp()) unique_id = uuid.uuid4() combined_id = str(stamp) + "-" + str(unique_id) indentify_photo = os.path.join(inden_path, f"{combined_id}.jpg") # 把原本地图片覆盖 with open(indentify_photo, "wb") as f: f.write(remote_content) # 更新结果 cbd_img = indentify_photo.replace(media, img_content_url) photo_obj.indentify_photo = cbd_img photo_obj.indentify_result = returnString photo_obj.label = returnCode photo_obj.save() pest_list = returnString.split("#") for i in pest_list: if i: res = i.split(",") nums = res[1] #数量 number = res[0] #编号 try: result_list = insect_dict[number] except: result_list = attract_discern[number] date.append({"nums":nums,"result_list":result_list,"number":number}) dat = {"image":cbd_img,"result":date, "label": returnCode} print(dat) return dat else: return {"code": 2, "msg": "无识别结果"} except Exception as e: return {"code": 2, "msg": "无识别结果"} else: return {"code": 2, "msg": "参数范围越界"} @kedong_deco(login_required=True) def cbdphoto_receive(request): # 测报灯/色诱/吸虫塔/害虫图片手动添加标注接口： img_id = request.POST.get('img_id') mark = request.POST.get('mark') photo_obj = MongoCBDphoto.objects.get(id=img_id) photo_obj.is_mark = 1 photo_obj.mark = mark photo_obj.save() return json.loads(mark)