#1522: create_random_red_agent -> _create_random_red_agent + converting NodeStateInstructionRed into a dataclass

This commit is contained in:
Czar Echavez
2023-07-03 13:36:14 +01:00
parent befd183b2c
commit a7913487b8
2 changed files with 5 additions and 19 deletions

View File

@@ -1,8 +1,11 @@
# Crown Copyright (C) Dstl 2022. DEFCON 703. Shared in confidence.
"""Defines node behaviour for Green PoL."""
from dataclasses import dataclass
from primaite.common.enums import NodePOLType
@dataclass()
class NodeStateInstructionRed(object):
"""The Node State Instruction class."""
@@ -137,20 +140,3 @@ class NodeStateInstructionRed(object):
The source node service state
"""
return self.source_node_service_state
def __repr__(self):
return (
f"{self.__class__.__name__}("
f"id={self.id}, "
f"start_step={self.start_step}, "
f"end_step={self.end_step}, "
f"target_node_id={self.target_node_id}, "
f"initiator={self.initiator}, "
f"pol_type={self.pol_type}, "
f"service_name={self.service_name}, "
f"state={self.state}, "
f"source_node_id={self.source_node_id}, "
f"source_node_service={self.source_node_service}, "
f"source_node_service_state={self.source_node_service_state}"
f")"
)