Skip to content

Commit

Permalink
Refactoring names
Browse files Browse the repository at this point in the history
  • Loading branch information
akadlec committed Dec 4, 2021
1 parent 68bf6f0 commit 57bb437
Show file tree
Hide file tree
Showing 3 changed files with 41 additions and 14 deletions.
18 changes: 9 additions & 9 deletions fb_bus_connector_plugin/api/v1parser.py
Original file line number Diff line number Diff line change
Expand Up @@ -137,16 +137,16 @@ def parse_message( # pylint: disable=too-many-branches,too-many-return-statemen
client_id=client_id,
)

if V1Validator.validate_get_device_state(payload=payload) and address is not None:
return self.parse_get_device_state(
if V1Validator.validate_read_device_state(payload=payload) and address is not None:
return self.parse_read_device_state(
payload=payload,
length=length,
address=address,
client_id=client_id,
)

if V1Validator.validate_set_device_state(payload=payload) and address is not None:
return self.parse_set_device_state(
if V1Validator.validate_write_device_state(payload=payload) and address is not None:
return self.parse_write_device_state(
payload=payload,
length=length,
address=address,
Expand All @@ -169,8 +169,8 @@ def parse_message( # pylint: disable=too-many-branches,too-many-return-statemen
client_id=client_id,
)

if V1Validator.validate_pub_sub_broadcast(payload=payload) and address is not None:
return self.parse_pub_sub_broadcast(
if V1Validator.validate_pub_sub_broadcast_register_value(payload=payload) and address is not None:
return self.parse_pub_sub_broadcast_register_value(
payload=payload,
length=length,
address=address,
Expand Down Expand Up @@ -451,7 +451,7 @@ def parse_report_single_register_value(

# -----------------------------------------------------------------------------

def parse_get_device_state(
def parse_read_device_state(
self,
payload: bytearray,
length: int,
Expand Down Expand Up @@ -479,7 +479,7 @@ def parse_get_device_state(

# -----------------------------------------------------------------------------

def parse_set_device_state(
def parse_write_device_state(
self,
payload: bytearray,
length: int,
Expand Down Expand Up @@ -574,7 +574,7 @@ def parse_pub_sub_write_register_key(
# -----------------------------------------------------------------------------

@staticmethod
def parse_pub_sub_broadcast(
def parse_pub_sub_broadcast_register_value(
payload: bytearray,
length: int,
address: int,
Expand Down
35 changes: 31 additions & 4 deletions fb_bus_connector_plugin/api/v1validator.py
Original file line number Diff line number Diff line change
Expand Up @@ -35,7 +35,34 @@ class V1Validator:
@classmethod
def validate(cls, payload: bytearray) -> bool:
"""Validate topic against sets of regular expressions"""
return True
if not cls.validate_version(payload=payload):
return False

if cls.validate_discover_device(payload=payload):
if (
cls.validate_pair_command_search_devices(payload=payload)
or cls.validate_pair_command_write_address(payload=payload)
or cls.validate_pair_command_provide_register_structure(payload=payload)
or cls.validate_pair_command_pairing_finished(payload=payload)
):
return True

if (
cls.validate_read_single_register(payload=payload)
or cls.validate_read_multiple_registers(payload=payload)
or cls.validate_write_single_register(payload=payload)
or cls.validate_write_multiple_registers(payload=payload)
or cls.validate_report_single_register(payload=payload)
or cls.validate_read_device_state(payload=payload)
or cls.validate_write_device_state(payload=payload)
or cls.validate_report_device_state(payload=payload)
or cls.validate_pub_sub_write_register_key(payload=payload)
or cls.validate_pub_sub_broadcast_register_value(payload=payload)
or cls.validate_pong_response(payload=payload)
):
return True

return False

# -----------------------------------------------------------------------------

Expand Down Expand Up @@ -82,14 +109,14 @@ def validate_report_single_register(cls, payload: bytearray) -> bool:
# -----------------------------------------------------------------------------

@classmethod
def validate_get_device_state(cls, payload: bytearray) -> bool:
def validate_read_device_state(cls, payload: bytearray) -> bool:
"""Validate payload against get device state packet structure"""
return Packet(int(payload[1])) == Packet.READ_STATE

# -----------------------------------------------------------------------------

@classmethod
def validate_set_device_state(cls, payload: bytearray) -> bool:
def validate_write_device_state(cls, payload: bytearray) -> bool:
"""Validate payload against set device state packet structure"""
return Packet(int(payload[1])) == Packet.WRITE_STATE

Expand All @@ -110,7 +137,7 @@ def validate_pub_sub_write_register_key(cls, payload: bytearray) -> bool:
# -----------------------------------------------------------------------------

@classmethod
def validate_pub_sub_broadcast(cls, payload: bytearray) -> bool:
def validate_pub_sub_broadcast_register_value(cls, payload: bytearray) -> bool:
"""Validate payload against pub/sub broadcast packet structure"""
return Packet(int(payload[1])) == Packet.PUB_SUB_BROADCAST_REGISTER_VALUE

Expand Down
2 changes: 1 addition & 1 deletion fb_bus_connector_plugin/connector.py
Original file line number Diff line number Diff line change
Expand Up @@ -196,4 +196,4 @@ def run(self) -> None:

time.sleep(0.001)

self.__logger.info("FB BUS connector was closed")
self.__logger.info("Connector FB BUS was closed")

0 comments on commit 57bb437

Please sign in to comment.