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.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.device_id
    conf = eval(msg.conf) 
    # 获取该device_id 下前一天的害虫情况
    now = datetime.now()
    now_stamp = int(time.time())
    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("#")
        for tp in tmp:
            k = tp.split(",")
            if k[0] in indentify_result.keys():
                indentify_result[k[0]] += int(k[1])
            else:
                indentify_result[k[0]] = int(k[1])

    # 定义预警变量
    warning_result = ""
    # 指定害虫数量预警
    appointPest = conf.get("appointPest")
    if appointPest == "on":
        appointPestName = conf.get("appointPestName")
        appointPestNum = conf.get("appointPestNum")
        pest_cg = appointPestName.split("#")
        pest_nu = appointPestNum.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])},"
        # 写入数据库
        if warning_result:
            MongoCBDPestWarning.objects.create(
                device_id = device_id,
                warning_content = f"{warning_result}请注意防范",
                warning_types = "2",
                upltime = now_stamp
            )        

    # 害虫种类
    pestCategory = conf.get("pestCategory")
    if pestCategory == "on":
        pestCategoryNum = conf.get("pestCategoryNum") 
        pest_cg_count = len(indentify_result.keys())
        if pest_cg_count > int(pestCategoryNum):
            # 写入数据库
            MongoCBDPestWarning.objects.create(
                device_id = device_id,
                warning_content = f"害虫种类{pest_cg_count},请注意防范",
                warning_types = "1",
                upltime = now_stamp
            )  

    # 害虫总数
    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):
            # 写入数据库
            MongoCBDPestWarning.objects.create(
                device_id = device_id,
                warning_content = f"害虫总数{str(ct)},请注意防范",
                warning_types = "3",
                upltime = now_stamp
            )  

    # 综合预警
    total_warning = ""
    pestWarn = conf.get("pestWarn")
    if pestWarn == "on":
        for m, n in indentify_result.items():
            total_warning += f"害虫{insect_dict.get(m)},数量{str(n)};" 
        # 写入数据库
        MongoCBDPestWarning.objects.create(
            device_id = device_id,
            warning_content = f"{total_warning}请注意防范!",
            warning_types = "4",
            upltime = now_stamp
        )