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())