mirror of
https://github.com/pgadmin-org/pgadmin4.git
synced 2024-11-27 19:20:21 -06:00
46 lines
1.3 KiB
Python
46 lines
1.3 KiB
Python
|
|
"""Encrypt the existing user password.
|
|
|
|
Revision ID: f195f9a4923d
|
|
Revises: 3c1e4b6eda55
|
|
Create Date: 2017-06-15 17:18:50.667139
|
|
|
|
"""
|
|
import config
|
|
from flask import current_app
|
|
from flask_security import Security, SQLAlchemyUserDatastore
|
|
from flask_security.utils import verify_and_update_password
|
|
from pgadmin.model import Keys, db, User, Role
|
|
|
|
# revision identifiers, used by Alembic.
|
|
revision = 'f195f9a4923d'
|
|
down_revision = '3c1e4b6eda55'
|
|
branch_labels = None
|
|
depends_on = None
|
|
|
|
|
|
def upgrade():
|
|
app = current_app
|
|
db.session.flush()
|
|
current_salt = Keys.query.filter_by(name = 'SECURITY_PASSWORD_SALT').first().value
|
|
app.config.update(dict(SECURITY_PASSWORD_SALT=current_salt))
|
|
app.config['SECURITY_PASSWORD_HASH'] = config.SECURITY_PASSWORD_HASH
|
|
|
|
if app.extensions.get('security') is None:
|
|
user_datastore = SQLAlchemyUserDatastore(db, User, Role)
|
|
Security(app, user_datastore, register_blueprint=False)
|
|
else:
|
|
app.config['SECURITY_PASSWORD_SALT'] = current_salt
|
|
users = User.query.all()
|
|
# This will upgrade the plaintext password of all the user as per the
|
|
# SECURITY_PASSWORD_HASH.
|
|
for user in users:
|
|
if user.password is not None:
|
|
verify_and_update_password(user.password, user)
|
|
|
|
db.session.commit()
|
|
|
|
|
|
def downgrade():
|
|
pass
|