"""IKEA TRADFRI plugs quirk."""

from zigpy.quirks.v2 import QuirkBuilder
from zigpy.zcl.clusters.general import LevelControl

from zhaquirks.ikea import IKEA

# remove LevelControl for plug to not show config options in ZHA
(
    QuirkBuilder(IKEA, "TRADFRI control outlet")
    .removes(LevelControl.cluster_id)
    .add_to_registry()
)