kedong/scripts/cbd_msg.py

import os
import sys
import time
import django
import json
from datetime import datetime, timedelta
local_path = os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
print(local_path)
if local_path not in sys.path:
    sys.path.append(local_path)
os.environ.setdefault("DJANGO_SETTINGS_MODULE", "kedong.settings")
django.setup()
from django.conf import settings
from smartfarming.models.sim_card import MongoMsg_Conf
from smartfarming.models.worm_forecast import MongoCBDphoto
from smartfarming.models.pest_count import MongoCBDPestWarning
from smartfarming.api.views.forecast.all_dict import insect_dict



# 获取所有的预警配置
msg_conf = MongoMsg_Conf.objects.all()
for msg in msg_conf:
    # 获取device_id  conf
    device_id = msg_conf.device_id
    conf = json.load(msg_conf.conf) 
    # 获取该device_id 下前一天的害虫情况
    now = datetime.now()
    previous_day = now - timedelta(days=1)
    start_time = datetime(previous_day.year, previous_day.month, previous_day.day, 0, 0, 0).timestamp()
    end_time = datetime(now.year, now.month, now.day, 23, 59, 59).timestamp()
    print("起始时间:", start_time)
    print("结束时间:", end_time)
    photo_data = MongoCBDphoto.objects.filter(addtime__range = [start_time, end_time]).values_list("indentify_result", flat=True)
    # 把当天的害虫统计出来
    indentify_result = {}
    for p in photo_data:
        tmp = p.split("#")
        if tmp[0] in indentify_result.keys():
            indentify_result[tmp[0]] += int(tmp[1])
        else:
            indentify_result[tmp[0]] = int(tmp[1])

    # 定义预警变量
    warning_result = ""
    # 指定害虫数量预警
    appointPest = conf.get("appointPest")
    if appointPest == "on":
        appointPestName = conf.get("appointPestName")
        appointPestNum = conf.get("appointPestNum")
        pest_cg = appointPestName.split("#")
        pest_nu = appointPestName.split("#")
        for i in pest_cg:
            if i in indentify_result.keys():
                if indentify_result[i] > int(pest_nu[pest_cg.index(i)]):
                    zh = insect_dict.get(i, "未知")
                    warning_result += f"害虫{zh}数量{str(indentify_result[i])},"

    # 害虫种类
    pestCategory = conf.get("pestCategory")
    if pestCategory == "on":
        pestCategoryNum = conf.get("pestCategoryNum") 
        pest_cg_count = len(indentify_result.keys())
        if pest_cg_count > int(pestCategoryNum):
            warning_result += f"害虫种类{pest_cg_count},"

    # 害虫总数
    pestTotal = conf.get("pestTotal")
    if pestTotal == "on":
        pestTotalNum = conf.get("pestTotalNum")
        ct = 0
        for m, n in indentify_result.items():
            ct += int(n)
        if ct > int(pestTotalNum):
            warning_result += f"害虫总数{str(ct)},"

    # 综合预警
    pestWarn = conf.get("pestWarn")
    if pestWarn == "on":
        for m, n in indentify_result.items():
            warning_result += f"{insect_dict.get(m)},数量{str(n)}," 
    if warning_result:
        warning_result += "请注意防范"
    # 写入数据库
    MongoCBDPestWarning.objects.create(
        
    )