1 files changed, 54 insertions, 0 deletions
diff --git a/cm-container/reset_admin.py b/cm-container/reset_admin.py
new file mode 100644
index 0000000..c7a7672
--- /dev/null
+++ b/cm-container/reset_admin.py
@@ -0,0 +1,54 @@
+########
+# Copyright (c) 2018 Cloudify Platform Ltd. All rights reserved
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#        http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+#    * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+#    * See the License for the specific language governing permissions and
+#    * limitations under the License.
+
+#!/usr/bin/env python
+
+import subprocess
+import json
+import argparse
+from flask_security.utils import encrypt_password
+from manager_rest.flask_utils import setup_flask_app
+
+
+def db_update_password(password):
+    password = encrypt_new_password(password)
+    password = password.replace('$', '\$')
+    sql_command = "\"update users set password='" + password + "' where username='admin'\""
+    cmd = "sudo -u postgres psql cloudify_db -c " + sql_command
+    subprocess.Popen(cmd, stdout=subprocess.PIPE, shell=True)
+
+
+def get_salt():
+    with open('/opt/manager/rest-security.conf') as f:
+        rest_security = json.load(f)
+
+    return rest_security['hash_salt']
+
+
+def encrypt_new_password(password):
+    app = setup_flask_app(hash_salt=get_salt())
+    with app.app_context():
+        password = encrypt_password(password)
+    return password
+
+
+if __name__ == '__main__':
+
+    parser = argparse.ArgumentParser(description=('Reset admin password in DB according to rest-security.conf'))
+    parser.add_argument('-p', '--password', required=True, help='New admin password')
+    args = parser.parse_args()
+
+    db_update_password(args.password)
+    print 'Done!\n'