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

Island; Remove disused IslandMode mongo document

This commit is contained in:
Mike Salvatore 2022-07-01 12:44:35 -04:00
parent f78fa73563
commit 410401a319
2 changed files with 1 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
monkey/monkey_island/cc/models/island_mode_model.py
View File

@ -1,7 +0,0 @@
from mongoengine import Document, StringField
class IslandMode(Document):
COLLECTION_NAME = "island_mode"
mode = StringField()

3
monkey/monkey_island/cc/services/database.py
View File

@ -6,7 +6,6 @@ from monkey_island.cc.database import mongo
from monkey_island.cc.models import Config from monkey_island.cc.models import Config
from monkey_island.cc.models.agent_controls import AgentControls from monkey_island.cc.models.agent_controls import AgentControls
from monkey_island.cc.models.attack.attack_mitigations import AttackMitigations from monkey_island.cc.models.attack.attack_mitigations import AttackMitigations
from monkey_island.cc.models.island_mode_model import IslandMode
from monkey_island.cc.services.config import ConfigService from monkey_island.cc.services.config import ConfigService
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
@ -33,7 +32,7 @@ class Database(object):
@staticmethod @staticmethod
def _should_drop(collection: str, drop_config: bool) -> bool: def _should_drop(collection: str, drop_config: bool) -> bool:
if not drop_config: if not drop_config:
if collection == IslandMode.COLLECTION_NAME or collection == Config.COLLECTION_NAME: if collection == Config.COLLECTION_NAME:
return False return False
return ( return (
not collection.startswith("system.") not collection.startswith("system.")