"""Custom ZHA quirk for Zemismart/Tuya TS0601 chain curtain motor."""
from __future__ import annotations
from zigpy.profiles import zha
from zigpy.zcl.clusters.general import Basic, Groups, Ota, Scenes, Time
from zhaquirks.const import (
DEVICE_TYPE,
ENDPOINTS,
INPUT_CLUSTERS,
MODELS_INFO,
OUTPUT_CLUSTERS,
PROFILE_ID,
)
from zhaquirks.tuya import (
TuyaManufacturerWindowCover,
TuyaManufCluster,
TuyaWindowCover,
TuyaWindowCoverControl,
)
try:
import zigpy.types as t
from zigpy.zcl import foundation
from zigpy.quirks.v2 import EntityType
from zhaquirks.tuya.builder import TuyaQuirkBuilder
_BUILDER_OK = True
except Exception:
_BUILDER_OK = False
if _BUILDER_OK:
class MotorDirection(t.enum8):
"""Motor direction for the curtain motor."""
forward = 0x00
back = 0x01
class LimitAction(t.enum8):
"""Limit configuration actions exposed by the Tuya MCU."""
set_upper_limit = 0x00
set_lower_limit = 0x01
delete_upper_limit = 0x02
delete_lower_limit = 0x03
delete_all_limits = 0x04
class ClickControl(t.enum8):
"""Short nudge command."""
up = 0x00
down = 0x01
(
TuyaQuirkBuilder("_TZE284_6hrnp30w", "TS0601")
.tuya_enchantment(data_query_spell=True)
.tuya_cover(control_dp=1, position_state_dp=3, position_control_dp=2, invert=True)
.tuya_battery(dp_id=13)
.tuya_enum(
dp_id=5,
attribute_name="motor_direction",
enum_class=MotorDirection,
translation_key="motor_direction",
fallback_name="Motor direction",
)
.tuya_enum(
dp_id=16,
attribute_name="limit_action",
enum_class=LimitAction,
translation_key="limit_action",
fallback_name="Limit action",
)
.tuya_dp_attribute(
dp_id=20,
attribute_name="click_control",
type=ClickControl,
access=foundation.ZCLAttributeAccess.Write,
)
.write_attr_button(
attribute_name="click_control",
attribute_value=0x00,
cluster_id=0xEF00,
unique_id_suffix="nudge_up",
translation_key="nudge_up",
fallback_name="Nudge up",
entity_type=EntityType.STANDARD,
)
.write_attr_button(
attribute_name="click_control",
attribute_value=0x01,
cluster_id=0xEF00,
unique_id_suffix="nudge_down",
translation_key="nudge_down",
fallback_name="Nudge down",
entity_type=EntityType.STANDARD,
)
.prevent_default_entity_creation(
endpoint_id=1,
cluster_id=TuyaWindowCoverControl.cluster_id,
unique_id_suffix="1-258-window_covering_type",
)
.prevent_default_entity_creation(
endpoint_id=1,
cluster_id=Ota.cluster_id,
unique_id_suffix="1-25-firmware_update",
)
.skip_configuration()
.add_to_registry()
)
else:
class ZemismartTs0601Cover(TuyaWindowCover):
"""Fallback quirk when the Tuya builder API is unavailable."""
signature = {
MODELS_INFO: [
("_TZE284_6hrnp30w", "TS0601"),
],
ENDPOINTS: {
1: {
PROFILE_ID: zha.PROFILE_ID,
DEVICE_TYPE: zha.DeviceType.SMART_PLUG,
INPUT_CLUSTERS: [
Basic.cluster_id,
Groups.cluster_id,
Scenes.cluster_id,
0xED00,
TuyaManufCluster.cluster_id,
],
OUTPUT_CLUSTERS: [
Time.cluster_id,
Ota.cluster_id,
],
}
},
}
replacement = {
ENDPOINTS: {
1: {
PROFILE_ID: zha.PROFILE_ID,
DEVICE_TYPE: zha.DeviceType.WINDOW_COVERING_DEVICE,
INPUT_CLUSTERS: [
Basic.cluster_id,
Groups.cluster_id,
Scenes.cluster_id,
TuyaManufacturerWindowCover,
TuyaWindowCoverControl,
],
OUTPUT_CLUSTERS: [
Ota.cluster_id,
],
}
},
}复制代码
发表于 2026-6-16 16:30:23
