From 623426374679e3a520d8bdf752e851e48d011949 Mon Sep 17 00:00:00 2001 From: Shreya Malviya Date: Tue, 30 Aug 2022 23:39:35 +0530 Subject: [PATCH] Island: Simplify configuration updating logic in PBAFileUpload --- monkey/monkey_island/cc/resources/pba_file_upload.py | 11 ++--------- 1 file changed, 2 insertions(+), 9 deletions(-) diff --git a/monkey/monkey_island/cc/resources/pba_file_upload.py b/monkey/monkey_island/cc/resources/pba_file_upload.py index d76b8665a..ef57b2429 100644 --- a/monkey/monkey_island/cc/resources/pba_file_upload.py +++ b/monkey/monkey_island/cc/resources/pba_file_upload.py @@ -101,16 +101,9 @@ class PBAFileUpload(AbstractResource): agent_configuration = self._agent_configuration_repository.get_configuration() if target_os == LINUX_PBA_TYPE: - custom_pbas = agent_configuration.custom_pbas.copy( - update={"linux_filename": safe_filename} - ) + agent_configuration.custom_pbas.linux_filename = safe_filename else: - custom_pbas = agent_configuration.custom_pbas.copy( - update={"windows_filename": safe_filename} - ) - - updated_agent_configuration = agent_configuration.copy(update={"custom_pbas": custom_pbas}) - self._agent_configuration_repository.store_configuration(updated_agent_configuration) + agent_configuration.custom_pbas.windows_filename = safe_filename @jwt_required def delete(self, target_os):