"""Quirk for _TZE204_lawxy9e2 TS0601."""
from zigpy.quirks.v2 import EntityType
import zigpy.types as t

from zhaquirks.tuya.builder import TuyaQuirkBuilder

class TuyaSpeedFan(t.enum8):
  level_1 = 0x00
  level_2 = 0x01
  level_3 = 0x02
  level_4 = 0x03
  level_5 = 0x04

class TuyaLightMode(t.enum8):
  none = 0x00
  relay = 0x01
  pos = 0x02

(
  TuyaQuirkBuilder("_TZE204_lawxy9e2", "TS0601")
  .tuya_switch(
    dp_id=1,
    attribute_name="on_off_1",
    entity_type=EntityType.STANDARD,
    translation_key="on_off_1",
    fallback_name="Switch 1",
  )
  .tuya_switch(
    dp_id=5,
    attribute_name="on_off_2",
    entity_type=EntityType.STANDARD,
    translation_key="on_off_2",
    fallback_name="Switch 2",
  )
  .tuya_enum(
    dp_id=3,
    attribute_name="Speed_Fan",
    enum_class=TuyaSpeedFan,
    translation_key="Speed_Fan",
    fallback_name="Speed fan",
  )
  .tuya_enum(
    dp_id=12,
    attribute_name="Ligth_Mode",
    enum_class=TuyaLightMode,
    translation_key="Ligth_Mode",
    fallback_name="Ligth mode",
  )
  .skip_configuration()
  .add_to_registry()
)