blob: 2253341ed0664fc47def50b686eeb1498d1c8ef0 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
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 'Password updated in DB!\n'
|
