4bb9012f23
Change-Id: Iaf7de63f303931bc754f0b4206b88b791046c603 Signed-off-by: Guntaka Umashankar Reddy <umashankarguntaka.reddy@windriver.com>
130 lines
4.4 KiB
Python
130 lines
4.4 KiB
Python
from typing import Any, Dict, List
|
|
|
|
from keywords.ptp.setup.object.grandmaster_settings import GrandmasterSettings
|
|
from keywords.ptp.setup.object.parent_data_set import ParentDataSet
|
|
from keywords.ptp.setup.object.port_data_set import PortDataSet
|
|
from keywords.ptp.setup.object.time_properties_data_set import TimePropertiesDataSet
|
|
|
|
|
|
class PTP4LExpectedDict:
|
|
"""
|
|
Class models a ptp4l expected dict
|
|
"""
|
|
|
|
def __init__(self, expected_dict: Dict[str, Any]):
|
|
"""
|
|
Constructor.
|
|
|
|
Args:
|
|
expected_dict (Dict[str, Any]): The dictionary read from the JSON setup template file associated with this ptp4l expected dict
|
|
|
|
"""
|
|
self.expected_dict = expected_dict
|
|
|
|
if "name" not in expected_dict:
|
|
raise Exception("Every PTP4L expected dict should have a name.")
|
|
self.name = expected_dict["name"]
|
|
|
|
def __str__(self) -> str:
|
|
"""
|
|
String representation of this object.
|
|
|
|
Returns:
|
|
str: String representation of this object.
|
|
|
|
"""
|
|
return self.get_name()
|
|
|
|
def get_name(self) -> str:
|
|
"""
|
|
Gets the name of this ptp4l expected dict.
|
|
|
|
Returns:
|
|
str: The name of this ptp4l expected dict.
|
|
"""
|
|
return self.name
|
|
|
|
def get_parent_data_set_for_hostname(self, hostname: str) -> ParentDataSet:
|
|
"""
|
|
Gets the of parent data set for hostname.
|
|
|
|
Args:
|
|
hostname (str): The name of the host.
|
|
|
|
Returns:
|
|
ParentDataSet: The of parent data set for hostname.
|
|
"""
|
|
host_specific_data = self.expected_dict.get(hostname)
|
|
|
|
if not host_specific_data:
|
|
raise Exception(f"Expected host specific data : {hostname} is not found for instance: {self.name}")
|
|
|
|
parent_data_set_raw = host_specific_data.get("parent_data_set")
|
|
if not parent_data_set_raw:
|
|
raise Exception(f"Expected parent data set not found for hostname: {hostname}")
|
|
|
|
return ParentDataSet(parent_data_set_raw)
|
|
|
|
def get_time_properties_data_set_for_hostname(self, hostname: str) -> TimePropertiesDataSet:
|
|
"""
|
|
Gets the of time properties data set for hostname.
|
|
|
|
Args:
|
|
hostname (str): The name of the host.
|
|
|
|
Returns:
|
|
TimePropertiesDataSet: The of time properties data set for hostname.
|
|
"""
|
|
host_specific_data = self.expected_dict.get(hostname)
|
|
|
|
if not host_specific_data:
|
|
raise Exception(f"Expected host specific data : {hostname} is not found for instance: {self.name}")
|
|
|
|
time_properties_data_set_raw = host_specific_data.get("time_properties_data_set")
|
|
if not time_properties_data_set_raw:
|
|
raise Exception(f"Expected time properties data set not found for hostname: {hostname}")
|
|
|
|
return TimePropertiesDataSet(time_properties_data_set_raw)
|
|
|
|
def get_grandmaster_settings_for_hostname(self, hostname: str) -> GrandmasterSettings:
|
|
"""
|
|
Gets the of grandmaster settings for hostname.
|
|
|
|
Args:
|
|
hostname (str): The name of the host.
|
|
|
|
Returns:
|
|
GrandmasterSettings: The of grandmaster settings for hostname.
|
|
"""
|
|
host_specific_data = self.expected_dict.get(hostname)
|
|
|
|
if not host_specific_data:
|
|
raise Exception(f"Expected host specific data : {hostname} is not found for instance: {self.name}")
|
|
|
|
grandmaster_settings_raw = host_specific_data.get("grandmaster_settings")
|
|
if not grandmaster_settings_raw:
|
|
raise Exception(f"Expected grandmaster settings not found for hostname: {hostname}")
|
|
|
|
return GrandmasterSettings(grandmaster_settings_raw)
|
|
|
|
def get_port_data_set_for_hostname(self, hostname: str) -> List[PortDataSet]:
|
|
"""
|
|
Gets the list of port data set for hostname.
|
|
|
|
Args:
|
|
hostname (str): The name of the host.
|
|
|
|
Returns:
|
|
List[PortDataSet]: The list of port data set for hostname.
|
|
"""
|
|
host_specific_data = self.expected_dict.get(hostname)
|
|
|
|
if not host_specific_data:
|
|
raise Exception(f"Expected host specific data : {hostname} is not found for instance: {self.name}")
|
|
|
|
port_data_set_list_raw = host_specific_data.get("port_data_set")
|
|
if not port_data_set_list_raw:
|
|
raise Exception(f"Expected port data set not found for hostname: {hostname}")
|
|
|
|
return [PortDataSet(port_data_set) for port_data_set in port_data_set_list_raw]
|