zfbox/web/API/channel.py

import json
from quart import jsonify, request
from . import api
from web.common.utils import login_required
from core.DBManager import mDBM
from myutils.ReManager import mReM
from core.ModelManager import mMM
from myutils.ConfigManager import myCongif
import asyncio
import time

@api.route('/channel/tree',methods=['GET'])
@login_required
async def channel_tree(): #获取通道树
    strsql = ("select t2.area_name ,t1.ID ,t1.channel_name from channel t1 left join area t2 on t1.area_id = t2.id;")
    data = mDBM.do_select(strsql)
    channel_tree = [{"area_name": channel[0], "ID": channel[1],"channel_name":channel[2]} for channel in data]
    return jsonify(channel_tree)

@api.route('/channel/list',methods=['GET'])
async def channel_list(): #获取通道列表 --分页查询，支持区域和通道名称关键字查询
    strsql = ("select t2.area_name,t1.ID,t1.channel_name,t1.ulr,t1.'type',t1.status,t1.element_id,t4.model_name "
              "from channel t1 left join area t2 on t1.area_id = t2.id "
              "left JOIN channel2model t3 on t1.ID = t3.channel_id "
              "left JOIN model t4 on t3.model_id = t4.ID "
              "where t1.is_work=1 order by area_name desc;")
    data = mDBM.do_select(strsql)
    channel_list = [{"area_name": channel[0], "ID": channel[1], "channel_name": channel[2], "ulr": channel[3],
                     "type": channel[4], "status": channel[5],
                     "element_id": channel[6],"model_name":channel[7]} for channel in data]
    return jsonify(channel_list)

@api.route('/channel/info',methods=['GET'])
@login_required
async def channel_info(): #获取通道信息   ---- list已获取详情
    return jsonify(1)

@api.route('/channel/add',methods=['POST'])
@login_required
async def channel_add(): #新增通道 -- 2024-8-1修改为与修改通道用一个接口
    json_data = await request.get_json()
    area = json_data.get('area')
    cName = json_data.get('cName')
    Rtsp = json_data.get('Rtsp')
    cid = int(json_data.get('cid'))

    if mReM.is_valid_rtsp_url(Rtsp) is not True:
        reStatus = 0
        reMsg = 'rtsp地址不合法'
        return jsonify({'status': reStatus, 'msg': reMsg})
    strsql = f"select id from area where area_name = '{area}';"
    area = mDBM.do_select(strsql,1)
    if area:
        area_id = area[0]
        strsql = f"select ID from channel where area_id={area_id} and channel_name='{cName}';"
        data = mDBM.do_select(strsql, 1)
        if data and data[0] != cid: #有值--代表重复 或者是它自己（只修改了RTSP地址）
            reStatus = 0
            reMsg = "同一区域内的通道名称不能相同！"
        else:
            if cid == -1:
                max_count = myCongif.get_data("max_channel_num")
                strsql = "select count(*) from channel;"
                data = mDBM.do_select(strsql,1)
                if data[0] >= max_count:
                    reStatus = 0
                    reMsg = f"已达到最多通道数量-{max_count}通道，不能再添加"
                    return jsonify({'status': reStatus, 'msg': reMsg})
                strsql = (f"INSERT INTO channel (area_id,channel_name,ulr,'type',status) values "
                          f"({area_id},'{cName}','{Rtsp}',1,0);")
            else:
                strsql = (f"UPDATE channel SET area_id={area_id},channel_name='{cName}'"
                          f",ulr='{Rtsp}' where ID={cid};")
            ret = mDBM.do_sql(strsql)
            if ret == True:
                reStatus = 1
                if cid == -1:
                    reMsg = '添加通道成功'
                    #await asyncio.sleep(0.5)  # 休眠0.5，待数据更新完成
                    # 对新增的视频通道进行视频采集和
                    strsql = f"select ID from channel where area_id={area_id} and channel_name='{cName}';"
                    data = mDBM.do_select(strsql, 1)
                    if data:
                        cid = data[0]
                        mMM.start_work(cid)
                    else:
                        print(f"这里不应该没有值！！！！area_id-{area_id},cName--{cName}")
                        await asyncio.sleep(1)  #休眠一秒
                        strsql = f"select ID from channel where area_id={area_id} and channel_name='{cName}';"
                        data = mDBM.do_select(strsql, 1)
                        if data:
                            cid = data[0]
                            mMM.start_work(cid)
                        else:
                            print(f"这里真的不应该没有值了！！！！area_id-{area_id},cName--{cName}")

                elif cid >0: #cid 不会是0，以防万一
                    reMsg = '修改通道成功'
                    #需要先停再启动---不区分有没有修改URL了，就当有修改使用
                    mMM.stop_work(cid)
                    start_time = time.time()
                    mMM.start_work(cid)
                    end_time = time.time()
                    execution_time = end_time - start_time
                    print(f"开始一个通道线程，花费了: {execution_time} seconds")
            else:
                reStatus = 0
                if cid == -1:
                    reMsg = '添加通道失败，请联系技术支持！'
                else:
                    reMsg = '修改通道信息失败，请联系技术支持！'
    else:
        reStatus = 0
        reMsg = '错误的区域ID，请修改'
    return jsonify({'status':reStatus,'msg':reMsg})

@api.route('/channel/img',methods=['POST'])
@login_required
async def get_channel_img():
    '''获取一帧图片'''
    json_data = await request.get_json()
    cid = int(json_data.get('cid'))
    img_base64 = mMM.verify_list.cm_get_cap_frame(cid)
    if img_base64:
        # 返回JSON响应
        return jsonify({"image": img_base64})
    else:
        return jsonify({"error": "Failed to capture frame"}), 404

@api.route('/channel/del',methods=['POST'])
@login_required
async def channel_del(): #删除通道
    json_data = await request.get_json()
    cid = int(json_data.get('cid'))
    reStatus = 0
    if cid > 0:
        mMM.stop_work(cid)
        #删除该通道和算法的关联信息：布防时间，算法模型数据----使用外键级联删除会方便很多，只要一个删除就可以
        ret = mDBM.delchannel(cid)
        if ret == True:
            reStatus = 1
            reMsg = '删除通道信息成功'
        else:
            reMsg = '删除通道信息失败，请联系技术支持！'
    else:
        reMsg = '通道ID参数错误，请联系技术支持！'
    return jsonify({'status': reStatus, 'msg': reMsg})

@api.route('/channel/check',methods=['GET'])
@login_required
async def channel_check(): #检查通道视频的在线情况--可以获取全部 ID-0,全部，1*具体通道ID--10分钟会更新一次在线状态-- 要验证
    ID = request.args.get('ID')
    if ID:
        ID = int(ID)
        if ID==0:
            strsql = "select ID,status from channel;"
        elif ID > 0:
            strsql = f"select ID,status from channel where ID={ID};"
        else:
            reStatus = 0
            reMsg = "参数错误"
            return jsonify({'status': reStatus, 'msg': reMsg})
        datas = mDBM.do_select(strsql)
        reStatus = 1
        reMsg = [{'ID': data[0], 'status': data[1]} for data in datas]
        return jsonify({'status': reStatus, 'msg': reMsg})
    else:
        reStatus = 0
        reMsg = "参数错误"
        return jsonify({'status': reStatus, 'msg': reMsg})

@api.route('/channel/C2M',methods=['POST'])
@login_required
async def get_ctm_data():
    '''获取通道关联模型的相关数据'''
    json_data = await request.get_json()
    cid = int(json_data.get('cid'))
    #获取算法数据
    strsql = "select ID,model_name from model;"
    model_datas = mDBM.do_select(strsql)
    m_datas = []
    if model_datas:
        m_datas = [{'ID':row[0],'model_name':row[1]} for row in model_datas]
    #获取c2m数据
    strsql = f"select ID,model_id,check_area,polygon,conf_thres,iou_thres from channel2model where channel_id = {cid};"
    c2m_data = mDBM.do_select(strsql,1)
    #schedule数据
    schedule = []
    if c2m_data:
        c2m_data_json = [{'model_id':c2m_data[1],'check_area':c2m_data[2],'polygon':c2m_data[3],
                          'conf_thres':c2m_data[4],'iou_thres':c2m_data[5]}]
        c2m_id = c2m_data[0]
        strsql = f"select day,hour,status from schedule where channel2model_id ={c2m_id} order by hour asc,day asc;"
        schedule_datas = mDBM.do_select(strsql)
        if schedule_datas:
            schedule = [{'day': row[0], 'hour': row[1], 'status': row[2]} for row in schedule_datas]
        else:
            schedule = []
    else:
        schedule = []
        c2m_data_json = []
    #返回数据
    redata = {"m_datas":m_datas,"c2m_data":c2m_data_json,"schedule":schedule}
    return jsonify(redata)

@api.route('/channel/chanegeC2M',methods=['POST'])
@login_required
async def change_c_t_m():
    '''
    修改和新增与通道关联的算法模块
    :return:
    '''
    json_data = await request.get_json()
    model_name = json_data.get('model_name')
    check_area = json_data.get('check_area')
    polygon_str = json_data.get('polygon_str')
    iou_thres = json_data.get('iou_thres')
    conf_thres = json_data.get('conf_thres')
    schedule = json_data.get('schedule')
    cid = int(json_data.get('cid'))
    #返回数据
    reStatus = 0
    reMsg = "更新失败，请联系技术支持！"

    #开始更新--- 统一先停止原通道工作线程
    if model_name=="请选择":
        strsql = f"select ID from channel2model where channel_id={cid};"
        data = mDBM.do_select(strsql,1)
        if data:
            #需要删除数据
            ret = mDBM.delC2M(cid,1)  #只是删除了数据库数据
            if ret:
                #停止该通道的model线程
                mMM.restartC2M(cid) #没有模型关联数据，模型工作线程不会启动
                reStatus = 1
                reMsg = "更新成功"
        else:   #原本就没数据
            reStatus = 1
            reMsg = "更新成功"
    else: #新增关联或修改关联
        # 验证数据
        if conf_thres <= 0 or conf_thres >= 1 or iou_thres <= 0 or iou_thres >= 1:
            reMsg = "阈值的有效范围是大于0，小于1；请输入正确的阈值"
            return jsonify({'status': reStatus, 'msg': reMsg})
        c2m_id = updateModel(cid,model_name,check_area,polygon_str,conf_thres,iou_thres) #c2m表
        if c2m_id != 0:
            ret = updataSchedule(c2m_id,schedule.replace("'", "\""))   #schedule表
            if ret:
                # 重启下通道的工作线程
                print(f"restartC2M--{cid}")
                mMM.restartC2M(cid) #没有模型关联数据，模型工作线程不会启动
                reStatus = 1
                reMsg = "更新成功"
    return jsonify({'status': reStatus, 'msg': reMsg})


def updataSchedule(c2m_id,schedule):
    # schedule_data_str = ("{'6': [1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1], "
    #                      "'0': [1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1], "
    #                      "'1': [1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1], "
    #                      "'2': [1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1], "
    #                      "'3': [1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1],"
    #                      "'4': [1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1],"
    #                      "'5': [1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1]}")
    #先删
    strsql = f"delete from schedule where channel2model_id={c2m_id};"
    mDBM.do_sql(strsql)
    #再加
    json_object = json.loads(schedule)
    for day, hours in json_object.items():
        for hour, status in enumerate(hours):
            strsql = (
                f"insert into schedule (channel2model_id,day,hour,status) values ({c2m_id},'{day}',{hour},{status})"
                f" on conflict(channel2model_id,day,hour) do update set status=excluded.status;")
            ret = mDBM.do_sql(strsql)
            if not ret: #插入失败后，需要删除所有数据
                strsql = f"delete from schedule where channel2model_id={c2m_id};"
                mDBM.do_sql(strsql)
                return ret
    return 1

def updateModel(channel_id,model_name,check_area,polygon_str,conf_thres,iou_thres):
    '''新增和修改通道关联的模型 ----约束：一个通道只能关联一个算法模型
        return 0 失败， 其他为c2m_id
    '''
    reStatus = 0
    strsql = f"select ID from channel where ID={channel_id};"
    if mDBM.do_select(strsql, 1):  # 通道号在数据库中
        strsql = f"select ID from model where model_name = '{model_name}';"
        data = mDBM.do_select(strsql, 1)
        if data:    #通道和模型都有对应数据
            reStatus = mDBM.updateC2M(channel_id, data[0],check_area,polygon_str,conf_thres,iou_thres)
    return reStatus

@api.route('/channel/area/list',methods=['GET'])
@login_required
async def channel_area_list():
    '''获取所属区域list'''
    strsql = "select * from area;"
    datas = mDBM.do_select(strsql)
    reMsg = [{'id': data[0], 'area_name': data[1]} for data in datas]
    return jsonify(reMsg)

@api.route('/channel/area/change',methods=['POST'])
@login_required
async def channel_area_change(): #修改区域名称
    area_id = (await request.form)['id']
    area_name = (await request.form)['name']
    strsql = f"select id from area where area_name='{area_name}';"
    datas = mDBM.do_select(strsql)
    if datas:
        reStatus = 0
        reMsg = '区域名称重复，请修改！'
    else:
        strsql = f"update area set area_name='{area_name}' where id = {area_id};"
        ret = mDBM.do_sql(strsql)
        if ret == True:
            reStatus = 1
            reMsg = '修改通道名称成功'
        else:
            reStatus = 0
            reMsg = '修改通道名称失败，请联系技术支持！'
    return jsonify({'status': reStatus, 'msg': reMsg})

@api.route('/channel/area/del',methods=['POST'])
@login_required
async def channel_area_del(): #删除区域
    area_id = (await request.form)['id']
    strsql = f"select * from channel where area_id={area_id};"
    datas = mDBM.do_select(strsql)
    if datas:
        reStatus = 0
        reMsg = '该区域下配置了通道，请先删除通道后，再删除区域！'
    else:
        strsql = f"delete from area where id={area_id};"
        ret = mDBM.do_sql(strsql)
        if ret == True:
            reStatus = 1
            reMsg = '删除区域成功'
        else:
            reStatus = 0
            reMsg = '删除区域失败，请联系技术支持！'
    return jsonify({'status': reStatus, 'msg': reMsg})

@api.route('/channel/area/add',methods=['POST'])
@login_required
async def channel_area_add(): #添加区域
    json_data = await request.get_json()
    area_name = json_data.get('name')
    strsql = f"select id from area where area_name='{area_name}';"
    data = mDBM.do_select(strsql,1)
    if data:
        reStatus = 0
        reMsg = '区域名称重复，请修改！'
    else:
        strsql = f"insert into area (area_name) values ('{area_name}');"
        ret = mDBM.do_sql(strsql)
        if ret == True:
            reStatus = 1
            reMsg = '新增区域成功'
        else:
            reStatus = 0
            reMsg = '新增区域失败，请联系技术支持！'
    return jsonify({'status': reStatus, 'msg': reMsg})

@api.route('/channel/model/list',methods=['GET'])
@login_required
async def channel_model_list(): #获取算法列表
    strsql = "select ID,model_name,version from model;"
    datas = mDBM.do_select(strsql)
    reMsg = {}
    if datas:
        reMsg = [{'ID': data[0], 'model_name': data[1],'version':data[2]} for data in datas]
    return jsonify(reMsg)


@api.route('/channel/model/linkmodel',methods=['POST'])    #--没调用。。
@login_required
async def channel_model_linkmodel(): #获取算法列表 --关联算法时展示 #？关联算法时需要初始化布防计划，同样删除的需要删除
    channel_id = (await request.form)['channel_id']
    model_list = json.loads((await request.form)['model_list'])
    #? 需要对channel_id和model_list的是否在数据库要进行一个检查
    reStatus = 0
    reMsg = '提交的参数错误，请检查！'
    strsql = f"select ID from channel where ID={channel_id};"
    if mDBM.do_select(strsql,1):    #通道号在数据库中
        for model_id in model_list:
            strsql = f"select ID from model where ID = {model_id};"
            if not mDBM.do_select(strsql,1):
                return jsonify({'status': reStatus, 'msg': reMsg})
        ret =mDBM.updateC2M(channel_id,model_list)
        if ret == True:
            reStatus = 1
            reMsg = '修改关联算法成功'
        else:
            reStatus = 0
            reMsg = '修改关联算法失败，请联系技术支持！'
    return jsonify({'status': reStatus, 'msg': reMsg})

@api.route('/channel/model/getarea',methods=['GET'])
@login_required
async def channel_model_getarea(): #获取算法区域的备注信息 --- 同时获取检测阈值
    ID = request.args.get('ID')
    strsql = f"select check_area,polygon,conf_threshold from channel2model where ID={ID};"
    data = mDBM.do_select(strsql,1)
    if data:
        reMsg = {'ID':ID,'check_area':data[0],'polygon':data[1],'conf_threshold':data[2]}
    else:
        reMsg = {}
    return jsonify(reMsg)

@api.route('/channel/model/changearea',methods=['POST'])
@login_required
async def channel_model_changearea(): #修改算法检测区域信息
    ID = (await request.form)['ID']
    check_area = (await request.form)['check_area']
    check_x = (await request.form)['polygon']
    strsql = (f"update channel2model set check_area={check_area},polygon={check_x} where ID={ID};")
    ret = mDBM.do_sql(strsql)
    if ret == True:
        reStatus = 1
        reMsg = '修改算法检测区域成功'
        #需要重启视频通道的执行程序 --需要cid
        #?
    else:
        reStatus = 0
        reMsg = '新修改算法检测区域失败，请联系技术支持！'
    return jsonify({'status': reStatus, 'msg': reMsg})

@api.route('/channel/model/changethreshold',methods=['POST'])
@login_required
async def channel_model_changthreshold(): #修改算法阈值
    ID = (await request.form)['ID']
    conf_threshold = (await request.form)['conf_threshold']

    strsql = (f"update channel2model set conf_threshold={conf_threshold} where ID={ID};")
    ret = mDBM.do_sql(strsql)
    if ret == True:
        reStatus = 1
        reMsg = '修改算法的阈值成功'
    else:
        reStatus = 0
        reMsg = '新修改算法的阈值失败，请联系技术支持！'
    return jsonify({'status': reStatus, 'msg': reMsg})

@api.route('/channel/model/getschedule',methods=['GET'])
@login_required
async def channel_model_getschedule(): #获取算法的布防时间
    ID = request.args.get('ID')
    strsql = f"select day,hour,status from schedule where channel2model_id={ID};"
    datas = mDBM.do_select(strsql)
    if datas:
        reMsg = [{'day': data[0], 'hour': data[1],'status':data[2]} for data in datas]
    else:
        reMsg = {}
    return jsonify(reMsg)

@api.route('/channel/model/changeschedule',methods=['POST'])
@login_required
async def channel_model_changeschedule(): #修改算法的布防时间
    ID = (await request.form)['ID']
    schedule_data_str = (await request.form)['schedule_data']
    schedule_data = json.loads(schedule_data_str.replace("'", '"'))
    for day,hours in schedule_data.items():
        for hour,status in enumerate(hours):
            strsql = (f"insert into schedule (channel2model_id,day,hour,status) values ({ID},'{day}',{hour},{status})"
                      f" on conflict(channel2model_id,day,hour) do update set status=excluded.status;")
            ret = mDBM.do_sql(strsql)
            if ret == False:
                reStatus = 0
                reMsg = '修改监测区域失败，请联系技术支持！'
                return jsonify({'status': reStatus, 'msg': reMsg})
    reStatus = 1
    reMsg = '修改监测区域成功'
    return jsonify({'status': reStatus, 'msg': reMsg})