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

island: Drop mongo db if registration is required

This commit is contained in:
Shreya Malviya 2021-09-27 14:32:02 +05:30 committed by VakarisZ
parent 194e244080
commit 340dd1f94b
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
monkey/monkey_island/cc/resources/auth/registration.py
View File

@ -7,6 +7,7 @@ from flask import make_response, request
import monkey_island.cc.environment.environment_singleton as env_singleton import monkey_island.cc.environment.environment_singleton as env_singleton
import monkey_island.cc.resources.auth.password_utils as password_utils import monkey_island.cc.resources.auth.password_utils as password_utils
from common.utils.exceptions import InvalidRegistrationCredentialsError, RegistrationNotNeededError from common.utils.exceptions import InvalidRegistrationCredentialsError, RegistrationNotNeededError
from monkey_island.cc.database import mongo
from monkey_island.cc.environment.user_creds import UserCreds from monkey_island.cc.environment.user_creds import UserCreds
from monkey_island.cc.setup.mongo.database_initializer import init_collections from monkey_island.cc.setup.mongo.database_initializer import init_collections
@ -21,6 +22,8 @@ class Registration(flask_restful.Resource):
credentials = _get_user_credentials_from_request(request) credentials = _get_user_credentials_from_request(request)
try: try:
# if new registration is required (credentials are reset), drop previous user's data
_drop_mongo_db()
env_singleton.env.try_add_user(credentials) env_singleton.env.try_add_user(credentials)
init_collections() init_collections()
return make_response({"error": ""}, 200) return make_response({"error": ""}, 200)
@ -42,3 +45,7 @@ def _get_user_credentials_from_request(request):
password_hash = password_utils.hash_password(password) password_hash = password_utils.hash_password(password)
return UserCreds(username, password_hash) return UserCreds(username, password_hash)
def _drop_mongo_db():
mongo.db.command("dropDatabase")