You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

799 lines
26 KiB
Python

# HA Spec -- http://www.zigbee.org/zigbee-for-developers/applicationstandards/zigbeehomeautomation/
# ZCL Spec -- http://www.zigbee.org/download/standards-zigbee-cluster-library/
# Zigbee & Zigbee Pro -- http://www.zigbee.org/zigbee-for-developers/network-specifications/zigbeepro/
# ZLL -- http://www.zigbee.org/zigbee-for-developers/applicationstandards/zigbee-light-link/
# Direct document downloads:
# HA Spec -- http://www.zigbee.org/?wpdmdl=2129
# Zigbee Pro Stack Profile -- http://www.zigbee.org/wp-content/uploads/2014/11/docs-07-4855-05-0csg-zigbee-pro-stack-profile-2.pdf
# Zigbee Spec -- http://www.zigbee.org/wp-content/uploads/2014/11/docs-05-3474-20-0csg-zigbee-specification.pdf
# ZCL Spec -- http://www.zigbee.org/~zigbeeor/wp-content/uploads/2014/10/07-5123-06-zigbee-cluster-library-specification.pdf
# ZLL - http://www.zigbee.org/?wpdmdl=2132
# A ZigBee device is made up of endpoints, each of which corresponds
# to a single piece of functionality on the device. Endpoint IDs are
# 8-bit. Endpoints have an associated 16-bit profile ID, which defines
# the category of functionality on that endpoint (e.g. "Home
# Automation").
#
# Each endpoint contains a set of input and output cluster IDs (also
# 16-bit), which correspond to a specific function in that
# profile. These clusters are defined in the Zigbee Cluster Library.
# The profile defines which clusters are expected to be implemented
# for a given device type.
#
# Input clusters allow you to send a message to the device. For
# example, a light bulb has an endpoint on the ZLL or HA profile with
# a 'level control' input cluster to set brightness. Output clusters
# allow the device to send a message. For examplebutton has an HA
# endpoint with an 'on off' output cluster.
#
# Clusters provide commands and attributes (e.g. the "on/off" cluster
# has "turn on", "turn off" and "toggle" commands and an "on/off"
# attribute).
#
# Devices can send commands from an endpoint to clusters on endpoints
# on other devices. There are two types of commands - profile
# commands and cluster commands. Profile commands are common to all
# clusters and allow you to perform actions such as querying an
# attribute from the specified cluster. Cluster commands are defined
# on each cluster.
#
# For example, to query the state of a bulb, you would send the "read
# attributes" profile command to the "on/off" cluster for the "on/off"
# attribute id. To turn on the bulb, you would send the "on" cluster
# command to the "on/off" cluster.
#
# There is a special "ZDO" endpoint (id=0) on all devices which uses
# the "ZigBee" profile, and this allows general device configuration
# and information. The ZDO clusters correspond to requests and
# responses. The response cluster IDs have the high bit set.
# Essentially each cluster only supports a single command, and there #
# are only cluster commands, so the command id and command type are
# not included in the request.
#
# Some common ZDO requests are:
# "device announce" (broadcast from a device on power on)
# "bind" (allows a device to connect an output cluster from another
# device to an input cluster).
# "active endpoints" (get the list of endpoint IDs)
# t_int = 0
#import enum
import struct
class DataType():
# ZCL Spec -- "2.5.2 Data Types"
NULL = 0x00
BOOLEAN = 0x10
BITMAP8 = 0x18
BITMAP16 = 0x19
BITMAP64 = 0x1f
UINT8 = 0x20
UINT16 = 0x21
UINT64 = 0x27
INT8 = 0x28
INT16 = 0x29
INT64 = 0x2f
ENUM8 = 0x30
ENUM16 = 0x31
CHARACTER_STRING = 0x42
ANALOG_DATATYPES = [
DataType.UINT8, DataType.UINT16, DataType.UINT64, DataType.INT8, DataType.INT16, DataType.INT64
]
class Profile():
ZIGBEE = 0x0000
HOME_AUTOMATION = 0x0104
ZIGBEE_LIGHT_LINK = 0xc05e
def get_profile_by_name(n):
return getattr(Profile, n.upper(), None)
class Endpoint():
ZDO = 0x00
class ZclCommandType():
PROFILE = 0
CLUSTER = 1
class Status():
SUCCESS = 0x00
FAILURE = 0x01
NOT_AUTHORIZED = 0x7E
RESERVED_FIELD_NOT_ZERO = 0x7F
MALFORMED_COMMAND = 0x80
UNSUP_CLUSTER_COMMAND = 0x81
UNSUP_GENERAL_COMMAND = 0x82
UNSUP_MANUF_CLUSTER_COMMAND = 0x83
UNSUP_MANUF_GENERAL_COMMAND = 0x84
INVALID_FIELD = 0x85
UNSUPPORTED_ATTRIBUTE = 0x86
INVALID_VALUE = 0x87
INSUFFICIENT_SPACE = 0x89
DUPLICATE_EXISTS = 0x8A
NOT_FOUND = 0x8B
UNREPORTABLE_ATTRIBUTE = 0x8C
INVALID_DATA_TYPE = 0x8D
INVALID_SELECTOR = 0x8E
WRITE_ONLY = 0x8F
INCONSISTENT_STARTUP_STATE = 0x90
DEFINED_OUT_OF_BAND = 0x91
INCONSISTENT = 0x92
ACTION_DENIED = 0x93
TIMEOUT = 0x94
ABORT = 0x95
INVALID_IMAGE = 0x96
WAIT_FOR_DATA = 0x97
NO_IMAGE_AVAILABLE = 0x98
REQUIRE_MORE_IMAGE = 0x99
NOTIFICATION_PENDING = 0x9A
HARDWARE_FAILURE = 0xC0
SOFTWARE_FAILURE = 0xC1
CALIBRATION_ERROR = 0xC2
UNSUPPORTED_CLUSTER = 0xC3
#ZDO_BY_NAME = {
# # Zigbee Spec -- "2.4.3.1.5 Simple_Desc_req"
# 'simple_desc': (0x0004, ('addr16:uint16', 'endpoint:uint8',),),
# # Zigbee Spec -- "2.4.4.1.5 Simple_Desc_resp"
# 'simple_desc_resp': (0x8004, ('status:enum8:success,invalid_ep,not_active,device_not_found,inv_requesttype,no_descriptor', 'addr16:uint16', 'b_simple_descriptors:uint8', 'simple_descriptors:#simple_descriptor',),),
# # Zigbee Spec -- "2.4.3.1.6 Active_EP_req"
# 'active_ep': (0x0005, ('addr16:uint16',),),
# # Zigbee Spec -- "2.4.4.1.6 Active_EP_resp"
# 'active_ep_resp': (0x8005, ('status:enum8:success,device_not_found,inv_requesttype,no_descriptor', 'addr16:uint16', 'n_active_eps:uint8', 'active_eps:*uint8',),),
# # Zigbee Spec -- "2.4.3.1.7 Match_Desc_req"
# 'match_desc': (0x0006, ('addr16:uint16', 'profile:uint16', 'n_in_clusters:uint8', 'in_clusters:*uint16', 'n_out_clusters:uint8', 'out_clusters:*uint16',),),
# # Zigbee Spec -- "2.4.4.1.7 Match_Desc_resp"
# 'match_desc_resp': (0x8006, ('status:enum8:success,device_not_found,inv_requesttype,no_descriptor', 'addr16:uint16', 'n_match_list:uint8', 'match_list:*uint8',),),
# # Zigbee Spec -- "2.4.3.2.2 Bind_req"
# 'bind': (0x0021, ('src_addr:uint64', 'src_ep:uint8', 'cluster:uint16', 'dst_addr_mode:enum8:_,addr16,_,addr64', 'dst_addr:uint64', 'dst_ep:uint8',),),
# # Zigbee Spec -- "2.4.3.2.3 Unbind_req"
# 'unbind': (0x0022, ('src_addr:uint64', 'src_ep:uint8', 'cluster:uint16', 'dst_addr_mode:enum8:_,addr16,_,addr64', 'dst_addr:uint64', 'dst_ep:uint8',),),
# # Zigbee Spec -- "2.4.4.2.2 Bind_resp"
# 'bind_resp': (0x8021, ('status:enum8:success,not_supported,invalid_ep,table_full,not_authorized',),),
# # Zigbee Spec -- "2.4.4.2.3 Unbind_resp"
# 'unbind_resp': (0x8022, ('status:enum8:success,not_supported,invalid_ep,table_full,not_authorized',),),
# # Spec -- "2.4.3.1.11 Device_annce"
# 'device_annce': (0x0013, ('addr16:uint16', 'addr64:uint64', 'capability:uint8'),), # See Figure 2.17
# # Zigbee Spec -- "2.4.4.3.9 Mgmt_NWK_Update_notify"
# 'mgmt_nwk_update_notify': (0x8038, ('status:uint8', 'scanned_channels:uint32', 'total_transmissions:uint16', 'transmisson_failures:uint16', 'n_energy_values:uint8', 'energy_values:*uint8',),),
#}
#ZDO_BY_ID = {
# cluster: (name, args) for name, (cluster, args) in ZDO_BY_NAME.items()
#}
def _decode_helper(args, data, i=0):
kwargs = {}
n = 1
b = 0
for arg in args:
arg = arg.split(':')
name, datatype = arg[0], arg[1],
v = None
decode, encode = STRUCT_TYPES[datatype.strip('*#%')]
if not callable(decode):
fmt, nbytes, = decode, encode
decode = lambda dd, ii, _: (struct.unpack(fmt, dd[ii:ii + nbytes])[0], ii + nbytes,)
if datatype.startswith('*'):
# Repeat n times
v = []
for _i in range(n):
x, i = decode(data, i, kwargs)
v.append(x)
elif datatype.startswith('#'):
# There are b bytes of records
v = []
ii = i + b
while i < ii:
x, i = decode(data, i, kwargs)
v.append(x)
elif datatype.startswith('%'):
# Keep reading records until end of frame
v = []
ii = len(data)
while i < ii:
x, i = decode(data, i, kwargs)
v.append(x)
else:
v, i = decode(data, i, kwargs)
stop = False
if name.startswith('s_'):
name = name[2:]
if v != 'SUCCESS':
stop = True
if name.startswith('n_'):
n = v
elif name.startswith('b_'):
b = v
else:
kwargs[name] = v
n = 1
b = 0
if stop:
break
return kwargs, i
def _decode_simple_descriptor(data, i, obj):
return _decode_helper(('endpoint:uint8', 'profile:uint16', 'device_identifier:uint16', 'device_version:uint8', 'n_in_clusters:uint8', 'in_clusters:*uint16', 'n_out_clusters:uint8', 'out_clusters:*uint16',), data, i)
#def _encode_simple_descriptor():
# pass
def _decode_read_attr_status(data, i, obj):
return _decode_helper(('attribute:uint16', 's_status:status8', 'datatype:uint8', 'value:datatype',), data, i)
#def _encode_read_attr_status():
# pass
def _decode_datatype(data, i, obj):
if 'datatype' not in obj:
raise ValueError('Object needs datatype field.')
datatype = obj['datatype']
if datatype == DataType.NULL:
return None, i,
if datatype not in DATATYPE_STRUCT_TYPES:
raise ValueError('Unknown struct type')
decode, encode = STRUCT_TYPES[DATATYPE_STRUCT_TYPES[datatype]]
if not callable(decode):
fmt, nbytes, = decode, encode
decode = lambda dd, ii, _: (struct.unpack(fmt, dd[ii:ii + nbytes])[0], ii + nbytes,)
return decode(data, i, obj)
#def _encode_datatype():
# pass
def _decode_attr_reporting_config():
pass
#def _encode_attr_reporting_config(obj):
# data = bytes()
# datatype = DATATYPES_BY_NAME[obj['datatype']]
# # min=1s, max=60s
# data += struct.pack('<BHBHH', 0, obj['attribute'], datatype, obj['minimum'], obj['maximum'])
# if datatype in ANALOG_DATATYPES:
# decode, encode = STRUCT_TYPES[DATATYPE_STRUCT_TYPES[datatype]]
# fmt, _nbytes = decode, encode
# data += struct.pack(fmt, obj.get('delta', 1))
# return data
def _decode_attr_reporting_status(data, i, obj):
# Note that attribute status records are not included for successfully configured attributes, in order to save bandwidth. In the case of successful configuration of all attributes, only a single attribute status record SHALL be included in the command, with the status field set to SUCCESS and the direction and attribute identifier fields omitted.
if data[i] == 0x00 and len(data) - i == 1:
return {
'status': 'SUCCESS',
}, i + 1
return _decode_helper(('status:status8', 'direction:uint8', 'attribute:uint16',), data, i)
#def _encode_attr_reporting_status():
# pass
def _decode_reported_attribute(data, i, obj):
return _decode_helper(('attribute:uint16', 'datatype:uint8', 'value:datatype',), data, i)
#def _encode_reported_attribute():
# pass
def _decode_string(data, i, obj):
nbytes, = struct.unpack('<B', data[i:i+1])
nbytes += 1
return data[i+1:i+nbytes].decode(), i + nbytes
def _encode_string(val):
val = val.encode()
return struct.pack('<B', len(val)) + val
def _decode_status(data, i, obj):
status, = struct.unpack('<B', data[i:i+1])
for s in Status:
if s.value == status:
return s.name, i + 1
raise ValueError('Unknown status {}'.format(status))
def _encode_status(val):
for s in Status:
if s.name == val:
return struct.pack('<B', s.value)
raise ValueError('Unknown status {}'.format(val))
STRUCT_TYPES = {
'uint8': ('<B', 1,),
'uint16': ('<H', 2,),
'uint32': ('<I', 4,),
'uint64': ('<Q', 8,),
'int8': ('<b', 1,),
'int16': ('<h', 2,),
'int32': ('<i', 4,),
'int64': ('<q', 8,),
'enum8': ('<B', 1,),
'enum16': ('<H', 2,),
'status8': (_decode_status, _encode_status,),
'string': (_decode_string, _encode_string,),
#'simple_descriptor': (_decode_simple_descriptor, _encode_simple_descriptor,),
#'read_attr_status': (_decode_read_attr_status, _encode_read_attr_status,),
#'datatype': (_decode_datatype, _encode_datatype,),
#'attr_reporting_config': (_decode_attr_reporting_config, _encode_attr_reporting_config,),
#'attr_reporting_status': (_decode_attr_reporting_status, _encode_attr_reporting_status,),
#'reported_attribute': (_decode_reported_attribute, _encode_reported_attribute,),
}
DATATYPE_STRUCT_TYPES = {
DataType.BOOLEAN: 'uint8',
DataType.BITMAP8: 'uint8',
DataType.BITMAP16: 'uint16',
DataType.BITMAP64: 'uint64',
DataType.UINT8: 'uint8',
DataType.UINT16: 'uint16',
DataType.UINT64: 'uint64',
DataType.INT8: 'int8',
DataType.INT16: 'int16',
DataType.INT64: 'int64',
DataType.ENUM8: 'enum8',
DataType.ENUM16: 'enum16',
DataType.CHARACTER_STRING: 'string',
}
DATATYPES_BY_NAME = {
'bool': DataType.BOOLEAN,
'bitmap8': DataType.BITMAP8,
'bitmap16': DataType.BITMAP16,
'bitmap64': DataType.BITMAP64,
'uint8': DataType.UINT8,
'uint16': DataType.UINT16,
'uint64': DataType.UINT64,
'int8': DataType.INT8,
'int16': DataType.INT16,
'int64': DataType.INT64,
'enum8': DataType.ENUM8,
'enum16': DataType.ENUM16,
'string': DataType.CHARACTER_STRING,
}
#def decode_zdo(cluster, data):
# if cluster not in ZDO_BY_ID:
# raise ValueError('Unknown ZDO 0x{:04x}'.format(cluster))
#print([hex(b) for b in data])
# cluster_name, args = ZDO_BY_ID[cluster]
# seq, = struct.unpack('<B', data[:1])
# data = data[1:]
# kwargs, _nbytes =_decode_helper(args, data)
# return cluster_name, seq, kwargs
#def _encode_helper(args, kwargs):
# data = bytes()
#
# for arg in args:
# arg = arg.split(':')
# name, datatype = arg[0], arg[1],
#
# if name.startswith('n_'):
# name = name[2:]
# values = [len(kwargs[name])]
# else:
# values = [kwargs[name]]
#
# if datatype.startswith('*'):
# datatype = datatype[1:]
# values = kwargs[name]
#
# decode, encode = STRUCT_TYPES[datatype]
#
# for value in values:
# if not callable(decode):
# fmt, _nbytes = decode, encode
# if datatype == 'uint64' and isinstance(value, str):
# value = int(value, 16)
# data += struct.pack(fmt, value)
# else:
# data += encode(value)
#
# return data
#def encode_zdo(cluster_name, seq, **kwargs):
# if cluster_name not in ZDO_BY_NAME:
# raise ValueError('Unknown ZDO "{}"'.format(cluster_name))
# cluster, args = ZDO_BY_NAME[cluster_name]
# data = struct.pack('<B', seq) + _encode_helper(args, kwargs)
# return cluster, data
_PROFILE_COMMANDS_BY_NAME = {
# ZCL Spec -- "2.5 General Command Frames"
'read_attributes': (0x00, ('attributes:*uint16',),),
'read_attributes_response': (0x01, ('attributes:%read_attr_status',),),
'write_attributes': (0x02, ('a:*write_attr',),),
'write_attributes_undivided': (0x03, ('a:*write_attr',),),
#'write_attributes_response': (0x04, ('*a:write_attr_status',),),
#'write_attributes_no_response': (0x05, ('a:*write_attr',),),
#'configure_reporting': (0x06, ('configs:*attr_reporting_config',),),
#'configure_reporting_response': (0x07, ('results:%attr_reporting_status',),),
# 'read_reporting_configuration': (0x08, (),),
# 'read_reporting_configuration_response': (0x09, (),),
'report_attributes': (0x0a, ('attributes:%reported_attribute',),),
'default_response': (0x0b, ('command:uint8', 'status:uint8',),),
# 'discover_attributes': (0x0c, (),),
# 'discover_attributes_response': (0x0d, (),),
# 'read_attributes_structured': (0x0e, (),),
# 'write_attributes_structured': (0x0f, (),),
# 'write_attributes_structured_response': (0x10, (),),
}
PROFILE_COMMANDS_BY_ID = {
command: (command_name, args) for command_name, (command, args) in _PROFILE_COMMANDS_BY_NAME.items()
}
_CLUSTERS_BY_NAME = {
# ZCL Spec -- Chapter 3 -- General
'basic': (0x0000, {
'reset': (0x00, (),),
}, {
},{
'zclversion': (0x0000, 'uint8',),
'application_version': (0x0001, 'uint8',),
'stack_version': (0x0002, 'uint8',),
'hw_version': (0x0003, 'uint8',),
'manufacturer_name': (0x0004, 'string',),
'model_id': (0x0005, 'string',),
'date_code': (0x0006, 'string',),
'power_source': (0x0007, 'enum8:unknown,mains-single,mains-three,battery,dc,emergency-constat,emergency-transfer',),
'location': (0x0010, 'string',),
'physical_environment': (0x0011, 'uint8',),
'device_enabled': (0x0012, 'bool',),
'sw_build_id': (0x4000, 'string',),
},),
'power_configuration': (0x0001, {
}, {
}, {
},),
'identify': (0x0003, {
'identify': (0x00, ('identify_time:uint16',),),
'identify_query': (0x01, (),),
'trigger_effect': (0x40, ('effect_id:uint8', 'effect_variant:uint8',),),
}, {
'identify_query_response': (0x00, ('timeout:uint16'),),
},{
'identify_time': (0x0000, 'uint16',),
},),
'groups': (0x0004, {
'add_group': (0x00, ('id:uint16', 'name:string'),),
'view_group': (0x01, ('id:uint16',),),
'get_group_membership': (0x02, ('n_ids:uint8', 'ids:*uint16'),),
'remove_group': (0x03, ('id:uint16',),),
'remove_all_groups': (0x04, (),),
'add_group_if_identifying': (0x05, ('id:uint16', 'name:string'),),
}, {
'add_group_response': (0x00, ('status:status8', 'id:uint16',),),
'view_group_response': (0x01, ('status:status8', 'id:uint16', 'name:string'),),
'get_group_membership_response': (0x02, ('capacity:uint8', 'n_ids:uint8', 'ids:*uint16',),),
'remove_group_response': (0x03, ('status:status8', 'id:uint16',),),
}, {
'name_support': (0x0000, 'uint8',),
},),
'scenes': (0x0005, {
}, {
}, {
},),
'onoff': (0x0006, {
'off': (0x00, (),),#command
'on': (0x01, (),),
'toggle': (0x02, (),),
'off_with_effect': (0x40, ('effect_id:uint8', 'effect_variant:uint8',),),
'on_with_recall_global_scene': (0x41, (),),
'on_with_timed_off': (0x42, ('control:uint8', 'on_time:uint16', 'off_wait_time:uint16',),),
}, { #middle hash is blank, unknown what this is
}, {
'onoff': (0x0000, 'bool',),#attribute
'global_scene_control': (0x4000, 'bool',),
'on_time': (0x4001, 'uint16',),
'off_wait_time': (0x4002, 'uint16',),
},),
'barrier_control': (0x0103, {
#commands
'go_to_percent': (0x0000, ('percentOpen:uint8'),),
'stop': (0x0001,(),),
}, {#blank middle hash like onoff
}, {#attribute
'moving_state': (0x0001, 'enum8',),
'safety_status': (0x0002,'map16',),
'capabilities': (0x0003, 'map8',),
'open_events': (0x0004, 'uint16',),
'close_events': (0x0005, 'uint16',),
'command_open_events': (0x0006, 'uint16',),
'command_close_events': (0x0007, 'uint16',),
'open_perioad': (0x0008, 'uint16',),
'close_period': (0x0009, 'uint16',),
'barrier_position': (0x000A, 'uint8',),
},),
'onoff_configuration': (0x0007, {
}, {
}, {
},),
'poll_control': (0x0020, {
}, {
}, {
},),
'diagnostics': (0x0b05, {
}, {
}, {
},),
# ZCL Spec -- Chapter 4 -- Measurement and Sensing
'electrical_measurement': (0x0b04, {
}, {
}, {
},),
# ZCL Spec -- Chapter 5 -- Lighting
# 'color': (0x0300, {
# 'move_to_hue': (0x00, ('hue:uint8', 'dir:enum8:shortest,longest,up,down', 'time:uint16',),),
# 'move_hue': (0x01, ('mode:enum8:stop,up,_,down', 'rate:uint8',),),
# 'step_hue': (0x02, ('mode:enum8:_,up,_,down', 'size:uint8', 'time:uint8',),),
# 'move_to_saturation': (0x03, ('saturation:uint8', 'dir:enum8:shortest,longest,up,down', 'time:uint16',),),
# 'move_saturation': (0x04, ('mode:enum8:stop,up,_,down', 'rate:uint8',),),
# 'step_saturation': (0x05, ('mode:enum8:_,up,_,down', 'size:uint8', 'time:uint8',),),
# 'move_to_hue_saturation': (0x06, ('hue:uint8', 'saturation:uint8', 'time:uint16',),),
# 'move_to_color_temperature': (0x0a, ('mireds:uint16', 'time:uint16',),),
# }, {
# }, {
# 'hue': (0x0000, 'uint8',),
# 'saturation': (0x0001, 'uint8',),
# 'remaining_time': (0x0002, 'uint16',),
# 'temperature': (0x0007, 'uint16',), # mireds, e.g. 1e6/K
# },),
# ZCL Spec -- Chapter 13 -- Commissioning
# 'commissioning': (0x0015, {
# }, {
# }, {
# },),
# 'touchlink': (0x1000, {
#}, {
#}, {
#},),
}
CLUSTERS_BY_ID = {}
for cluster_name, (cluster, rx_commands, tx_commands, attributes,) in _CLUSTERS_BY_NAME.items():
CLUSTERS_BY_ID[cluster] = (
cluster_name,
{
command: (command_name, args) for command_name, (command, args) in rx_commands.items()
},
{
command: (command_name, args) for command_name, (command, args) in tx_commands.items()
},
{
attribute: (attribute_name, datatype) for attribute_name, (attribute, datatype) in attributes.items()
}
)
def sequence_number(data):
seq, command = struct.unpack('<BB', data[1:3])
return seq
def decode_zcl(cluster, data):
frame_control, = struct.unpack('<B', data[:1])
frame_type = frame_control & 1
direction = frame_control & (1 << 3)
disable_default_response = frame_control & (1 << 4)
manufacturer_specific = 0 #frame_control & (1 << 2) ??
#print(frame_type, direction, manufacturer_specific)
if manufacturer_specific:
manufacturer_code, seq, command = struct.unpack('<HBB', data[1:5])
data = data[5:]
else:
manufacturer_code = 0
seq, command = struct.unpack('<BB', data[1:3])
data = data[3:]
#print(manufacturer_code, seq, command)
if cluster not in CLUSTERS_BY_ID:
raise ValueError('Unknown cluster {}'.format(cluster))
cluster_name, rx_commands, tx_commands, attributes = CLUSTERS_BY_ID[cluster]
if direction == 0:
commands = rx_commands
else:
commands = tx_commands
if frame_type == 0:
# Profile command
if command not in PROFILE_COMMANDS_BY_ID:
raise ValueError('Unknown profile command {} for cluster "{}"'.format(command, cluster_name))
command_name, args = PROFILE_COMMANDS_BY_ID[command]
kwargs, _nbytes = _decode_helper(args, data)
return cluster_name, seq, ZclCommandType.PROFILE, command_name, not disable_default_response, kwargs
else:
# Cluster command
if command not in commands:
raise ValueError('Unknown cluster command {} for cluster "{}"'.format(command, cluster_name))
command_name, args = commands[command]
kwargs, _nbytes = _decode_helper(args, data)
return cluster_name, seq, ZclCommandType.CLUSTER, command_name, not disable_default_response, kwargs
def return_attributes(cluster, data):
seq, command = struct.unpack('<BB', data[1:3])
data = data[3:]
cluster_name, rx_commands, tx_commands, attributes = CLUSTERS_BY_ID[cluster]
commands = tx_commands
command_name, args = commands[command]
kwargs, _nbytes = _decode_helper(args, data)
return kwargs
def get_cluster_by_name(cluster_name):
if cluster_name not in _CLUSTERS_BY_NAME:
raise ValueError('Unknown cluster "{}"'.format(cluster_name))
cluster, _rx_commands, _tx_commands, _attributes = _CLUSTERS_BY_NAME[cluster_name]
return cluster
def get_cluster_rx_command(cluster_name, command_name):
if cluster_name not in _CLUSTERS_BY_NAME:
raise ValueError('Unknown cluster "{}"'.format(cluster_name))
cluster, rx_commands, _tx_commands, _attributes = _CLUSTERS_BY_NAME[cluster_name]
if command_name not in rx_commands:
raise ValueError('Unknown command "{}"'.format(command_name))
command, args = rx_commands[command_name]
return cluster, command, args
#def encode_cluster_command(cluster_name, command_name, seq, direction=0, default_response=True, manufacturer_code=None, **kwargs):
# cluster, command, args = get_cluster_rx_command(cluster_name, command_name)
#
# # ZCL Spec - "2.4.1.1 Frame Control Field"
# frame_control = 1 # Cluster command (command is specific to this cluster)
# if direction:
# frame_control |= 1 << 3
# if not default_response:
# frame_control |= 1 << 4
#
# if manufacturer_code is not None:
# frame_control |= 1 << 2
# data = struct.pack('<BHBB', frame_control, manufacturer_code, seq, command)
# else:
# data = struct.pack('<BBB', frame_control, seq, command)
#
# data += _encode_helper(args, kwargs)
#
# return cluster, data
#def encode_profile_command(cluster_name, command_name, seq, direction=0, default_response=True, manufacturer_code=None, **kwargs):
# if cluster_name not in CLUSTERS_BY_NAME:
# raise ValueError('Unknown cluster "{}"'.format(cluster_name))
#
# cluster, rx_commands, tx_commands, attributes = CLUSTERS_BY_NAME[cluster_name]
#
# if command_name not in PROFILE_COMMANDS_BY_NAME:
# raise ValueError('Unknown command "{}"'.format(command_name))
#
# command, args = PROFILE_COMMANDS_BY_NAME[command_name]
#
# # ZCL Spec - "2.4.1.1 Frame Control Field"
# frame_control = 0 # Profile command (command applies to all clusters)
# data = struct.pack('<BBB', frame_control, seq, command)
#
# data += _encode_helper(args, kwargs)
#
# return cluster, data
#def get_json():
# return {
# 'profile': [
# {
# 'name': p.name,
# 'profile': p
# } for p in Profile
# ],
# # 'zdo': [
# # {
# # 'cluster_name': cluster_name,
# # 'cluster': cluster,
# # 'args': args
# # } for cluster_name, (cluster, args) in ZDO_BY_NAME.items()
# # ],
# 'status': [s.name for s in Status],
# 'profile_command': [
# {
# 'name': command_name,
# 'command': command,
# 'args': args
# } for command_name, (command, args) in PROFILE_COMMANDS_BY_NAME.items()
# ],
# 'cluster': [
# {
# 'name': cluster_name,
# 'cluster': cluster,
# 'rx_commands': [
# {
# 'name': command_name,
# 'command': command,
# 'args': args
# } for command_name, (command, args) in rx_commands.items()
# ],
# 'tx_commands': [
# {
# 'name': command_name,
# 'command': command,
# 'args': args
# } for command_name, (command, args) in tx_commands.items()
# ],
# 'attributes': [
# {
# 'name': attribute_name,
# 'attribute': attribute,
# 'datatype': datatype
# } for attribute_name, (attribute, datatype) in attributes.items()
# ],
# } for cluster_name, (cluster, rx_commands, tx_commands, attributes) in CLUSTERS_BY_NAME.items()
# ]
# }