import cv2
import win32com.client

class CAM_UTIL:
    def __init__(self, cam1_name='USB Camera', cam2_name="PC Camera",):
        self.webcam_list = None
        self.cam1 = cam1_name
        self.cam2 = cam2_name
        self.list_webcams()




    # 获取摄像头列表
    def list_webcams(self):
        # 创建一个WMI客户端实例
        wmi = win32com.client.GetObject("winmgmts:")
        webcams = wmi.InstancesOf("Win32_PnPEntity")
        # 创建摄像头和索引字典
        webcam_dict = {}
        index = 0
        for device in webcams:
            name = getattr(device, 'Name', None)
            pnp_class = getattr(device, 'PNPClass', None)
            if pnp_class == 'Camera':
                # 将设备名字和索引添加到字典中
                print(f"{name},{pnp_class}")
            if name is not None and self.cam1 in name:

                # 将设备名字和索引添加到字典中
                webcam_dict[self.cam1] = index
                index += 1
            elif name is not None and self.cam2 in name:

                # 将设备名字和索引添加到字典中
                webcam_dict[self.cam2] = index
                index += 1
            self.webcam_list = webcam_dict

        print(webcam_dict)
        # 按名称重新排序
        webcam_dict = dict(sorted(webcam_dict.items(), key=lambda x: x[0]))
        print(webcam_dict)
        return webcam_dict