COnvert DATA_TYPE to enum. (#57699)

This commit is contained in:
jan iversen 2021-10-15 07:09:59 +02:00 committed by GitHub
parent faf5c2eb40
commit 4417ffb407
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
8 changed files with 122 additions and 155 deletions

View file

@ -53,9 +53,9 @@ from .const import (
CONF_SWAP_WORD_BYTE,
CONF_VERIFY,
CONF_WRITE_TYPE,
DATA_TYPE_STRING,
SIGNAL_START_ENTITY,
SIGNAL_STOP_ENTITY,
DataType,
)
from .modbus import ModbusHub
@ -165,7 +165,7 @@ class BaseStructPlatform(BasePlatform, RestoreEntity):
registers = self._swap_registers(registers)
byte_string = b"".join([x.to_bytes(2, byteorder="big") for x in registers])
if self._data_type == DATA_TYPE_STRING:
if self._data_type == DataType.STRING:
return byte_string.decode()
val = struct.unpack(self._structure, byte_string)