oxbrbtx/PrimAITE
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

#2405 remove .lower from _read_application_type, rename _software_manager to _software_request_manager in base.py

Browse Source
This commit is contained in:
Cristian-VM2
2024-03-28 12:01:36 +00:00
parent 9b6135524e
commit 8612842b74
1 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
src/primaite/simulator/network/hardware/base.py
View File

@@ -846,10 +846,10 @@ class Node(SimComponent):
)
rm.add_request("os", RequestType(func=self._os_request_manager, validator=_node_is_on))
self._software_manager = RequestManager()
rm.add_request("software_manager", RequestType(func=self._software_manager, validator=_node_is_on))
self._software_request_manager = RequestManager()
rm.add_request("software_manager", RequestType(func=self._software_request_manager, validator=_node_is_on))
self._application_manager = RequestManager()
self._software_manager.add_request(name="application", request_type=RequestType(func=self._application_manager))
self._software_request_manager.add_request(name="application", request_type=RequestType(func=self._application_manager))
self._application_manager.add_request(
name="install",
@@ -879,17 +879,17 @@ class Node(SimComponent):
def _read_application_type(self, application_class_str: str) -> Type[IOSoftwareClass]:
"""Wrapper that converts the string from the request manager into the appropriate class for the application."""
if application_class_str.lower() == "DoSBot".lower():
if application_class_str == "DoSBot":
from primaite.simulator.system.applications.red_applications.dos_bot import DoSBot
return DoSBot
elif application_class_str.lower() == "DataManipulationBot".lower():
elif application_class_str == "DataManipulationBot":
from primaite.simulator.system.applications.red_applications.data_manipulation_bot import (
DataManipulationBot,
)
return DataManipulationBot
elif application_class_str.lower() == "WebBrowser".lower():
elif application_class_str == "WebBrowser":
from primaite.simulator.system.applications.web_browser import WebBrowser
return WebBrowser