p34709852
/
monkey
forked from p15670423/monkey
1
0
Fork 1
Code Issues Pull Requests Projects Releases Wiki Activity

Fixed island's config.py default value setting method to handle 3 layers of nested objects

This commit is contained in:
VakarisZ 2020-08-03 11:45:31 +03:00
parent 51c2655e2b
commit d74c3d15ca
1 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
monkey/monkey_island/cc/services/config.py
View File

@ -246,6 +246,16 @@ class ConfigService:
for property3, subschema3 in list(subschema2["properties"].items()): for property3, subschema3 in list(subschema2["properties"].items()):
if "default" in subschema3: if "default" in subschema3:
sub_dict[property3] = subschema3["default"] sub_dict[property3] = subschema3["default"]
elif "properties" in subschema3:
layer_3_dict = {}
for property4, subschema4 in list(subschema3["properties"].items()):
if "properties" in subschema4:
raise ValueError("monkey/monkey_island/cc/services/config.py "
"can't handle 5 level config. "
"Either change back the config or refactor.")
if "default" in subschema4:
layer_3_dict[property4] = subschema4["default"]
sub_dict[property3] = layer_3_dict
main_dict[property2] = sub_dict main_dict[property2] = sub_dict
instance.setdefault(property1, main_dict) instance.setdefault(property1, main_dict)