pbdkf2.py

#!/usr/bin/env python

from flask import current_app
from werkzeug.security import (
    generate_password_hash,
    check_password_hash,
    DEFAULT_PBKDF2_ITERATIONS,
)


class PBDKF2(object):
    def __init__(self, app=None):
        if app is not None:  # pragma: no cover
            self.init_app(app)

    def init_app(self, app):
        # Configure default app settings
        app.config.setdefault('PBDKF2_METHOD', 'sha1')
        app.config.setdefault('PBKDF2_ITERATIONS', DEFAULT_PBKDF2_ITERATIONS)
        app.config.setdefault('PBDKF2_SALT_LENGTH', 8)

    def generate_password_hash(self, password):
        method = 'pbkdf2:{}:{}'.format(current_app.config['PBDKF2_METHOD'],
                                       current_app.config['PBKDF2_ITERATIONS'])
        return generate_password_hash(password, method,
                                      current_app.config['PBDKF2_SALT_LENGTH'])

    def check_password_hash(self, pwhash, password):
        return check_password_hash(pwhash, password)