forked from p15670423/monkey
Island: Remove disused set_server_config.py
This commit is contained in:
parent
13b1904cf7
commit
b4e861cdd6
|
@ -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()
|
Loading…
Reference in New Issue