p15670423
/
monkey
1
0
Fork 5
Code Issues Pull Requests 3 Projects Releases Wiki Activity

Add `_expand_path()` to wrap `os.path.expandvars()\' and `os.path.expanduser()\'

This commit is contained in:
Shreya 2021-06-07 19:44:42 +05:30
parent 5ba8effe1a
commit 227039f30c
1 changed files with 10 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
monkey/monkey_island/cc/setup/island_config_options.py
View File

@ -13,8 +13,8 @@ from monkey_island.cc.server_utils.consts import (
class IslandConfigOptions:
def __init__(self, config_contents: dict):
self.data_dir = os.path.expandvars(
os.path.expanduser(config_contents.get("data_dir", DEFAULT_DATA_DIR))
self.data_dir = IslandConfigOptions._expand_path(
config_contents.get("data_dir", DEFAULT_DATA_DIR)
)
self.log_level = config_contents.get("log_level", DEFAULT_LOG_LEVEL)
@ -23,9 +23,13 @@ class IslandConfigOptions:
"mongodb", {"start_mongodb": DEFAULT_START_MONGO_DB}
).get("start_mongodb", DEFAULT_START_MONGO_DB)
self.crt_path = os.path.expandvars(
os.path.expanduser(config_contents.get("cert_path", DEFAULT_CRT_PATH))
self.crt_path = IslandConfigOptions._expand_path(
config_contents.get("cert_path", DEFAULT_CRT_PATH)
)
self.key_path = os.path.expandvars(
os.path.expanduser(config_contents.get("key_path", DEFAULT_KEY_PATH))
self.key_path = IslandConfigOptions._expand_path(
config_contents.get("key_path", DEFAULT_KEY_PATH)
)
@staticmethod
def _expand_path(path: str) -> str:
return os.path.expandvars(os.path.expanduser(path))