from pyrosapp.models import *
from .Device import DeviceObj
from enum import Enum

class FilterEnum(Enum):
    t = "T"
    h = "H"
    # TODO: définir les noms des filtres

class FilterWheelObj(DeviceObj):

    def __init__(self):
        super().__init__()

        self.set_msgs = [
            ("FILTER", [1, ], FilterEnum),
        ]

        self.get_msgs = [
            "STATUS",
            "SETUP",
        ]

        self.do_msgs = [
            ("CHANGE_FILTER", [0, ]),
        ]