From b4e861cdd66b306a6cfdc30b19891bb34b7322cd Mon Sep 17 00:00:00 2001 From: Mike Salvatore Date: Wed, 1 Sep 2021 08:54:54 -0400 Subject: [PATCH] Island: Remove disused set_server_config.py --- .../cc/environment/set_server_config.py | 66 ------------------- 1 file changed, 66 deletions(-) delete mode 100644 monkey/monkey_island/cc/environment/set_server_config.py diff --git a/monkey/monkey_island/cc/environment/set_server_config.py b/monkey/monkey_island/cc/environment/set_server_config.py deleted file mode 100644 index 490d92479..000000000 --- a/monkey/monkey_island/cc/environment/set_server_config.py +++ /dev/null @@ -1,66 +0,0 @@ -import argparse -import json -import logging -import sys -from pathlib import Path -from shutil import move - - -def add_monkey_dir_to_sys_path(): - monkey_path = Path(sys.path[0]) - monkey_path = monkey_path.parents[2] - sys.path.insert(0, monkey_path.__str__()) - - -add_monkey_dir_to_sys_path() - -from monkey_island.cc.server_utils.consts import DEFAULT_SERVER_CONFIG_PATH # noqa: E402 isort:skip - -SERVER_CONFIG = "server_config" -BACKUP_CONFIG_FILENAME = "./server_config.backup" - -logger = logging.getLogger(__name__) -logger.addHandler(logging.StreamHandler()) -logger.setLevel(logging.DEBUG) - - -def main(): - args = parse_args() - file_path = DEFAULT_SERVER_CONFIG_PATH - - if args.server_config == "restore": - restore_previous_config(file_path) - quit() - - # Read config - with open(file_path) as config_file: - config_data = json.load(config_file) - - # Backup the config - with open(BACKUP_CONFIG_FILENAME, "w") as backup_file: - json.dump(config_data, backup_file, indent=4) - backup_file.write("\n") - - # Edit the config - config_data[SERVER_CONFIG] = args.server_config - - # Write new config - logger.info("Writing the following config: {}".format(json.dumps(config_data, indent=4))) - with open(file_path, "w") as config_file: - json.dump(config_data, config_file, indent=4) - config_file.write("\n") # Have to add newline at end of file, since json.dump does not. - - -def parse_args(): - parser = argparse.ArgumentParser() - parser.add_argument("server_config", choices=["standard", "password", "restore"]) - args = parser.parse_args() - return args - - -def restore_previous_config(config_path): - move(BACKUP_CONFIG_FILENAME, config_path) - - -if __name__ == "__main__": - main()