from algo.model_manager import AlgoModelExec
from model_handler.base_model_handler import BaseModelHandler


class LaborEngineModelHandler(BaseModelHandler):

    def __init__(self, model: AlgoModelExec):
        super().__init__(model)
        self.model_names = {
            0: '三脚架',
            1: '三通',
            2: '专用软管',
            3: '人',
            4: '作业信息公示牌',
            5: '切断阀',
            6: '危险告知牌',
            7: '压力测试仪',
            8: '压力表',
            9: '反光衣',
            10: '可燃气体报警控制器',
            11: '呼吸面罩',
            12: '喉箍',
            13: '四合一',
            14: '圆头水枪',
            15: '头',
            16: '安全告知牌',
            17: '安全带',
            18: '安全帽',
            19: '安全标识',
            20: '安全标识牌',
            21: '安全绳',
            22: '对讲机',
            23: '尖头水枪',
            24: '工服',
            25: '开关',
            26: '报警装置',
            27: '接头',
            28: '施工路牌',
            29: '气体检测仪',
            30: '水带',
            31: '水带_矩形',
            32: '流量计',
            33: '消火栓箱',
            34: '灭火器',
            35: '灶台',
            36: '灶眼',
            37: '照明设备',
            38: '熄火保护',
            39: '燃气管道',
            40: '燃气计量器具',
            41: '电线暴露',
            42: '电路图',
            43: '警戒线',
            44: '调压器',
            45: '调长器',
            46: '贴纸',
            47: '跨电线',
            48: '路锥',
            49: '过滤器',
            50: '配电箱内部',
            51: '配电箱外部',
            52: '长柄阀门',
            53: '闪光灯亮',
            54: '闪光灯灭',
            55: '阀门',
            56: '非专用软管',
            57: '风管',
            58: '鼓风机',

        }
        self.model_ids = list(self.model_names.keys())
