密码保护

Posted 2020-10-18 刘书签

1.更新User对象，设置对内的_password

class User(db.Model):

    __tablename__ = ‘user‘ 

    _password = db.Column(db.String(200), nullable=False) #内部使用

 

class User(db.Model):
    __tablename__ = ‘user‘
    id = db.Column(db.Integer, primary_key=TabError, autoincrement=True)
    username = db.Column(db.String(20), nullable=False)
    _password = db.Column(db.String(200), nullable=False)#内部使用
    nickname = db.Column(db.String(20))

 

2.编写对外的password

from werkzeug.security import generate_password_hash, check_password_hash

    @property

    def password(self):  #外部使用，取值

        return self._password

    @password.setter

    def password(self, row_password):#外部使用，赋值

        self._password = generate_password_hash(row_password)

 

    @property
    def password(self):#外部使用
        return self._password

    @password.setter
    def password(self,row_password):
        self._password=generate_password_hash(row_password)

 

3.密码验证方法：

    def check_password(self, row_password): #密码验证

        result = check_password_hash(self._password,row_password)

        return result

 

    def check_password(self,row_password):
        result=check_password_hash(self._password,row_password)
        return result

 

4.登录验证：

        password1 = request.form.get(‘password‘)

        user = User.query.filter(User.username == username).first()

        if user:

            if user.check_password(password1):

 

# 登录
@app.route(‘/login/‘, methods=[‘GET‘, ‘POST‘])
def login():
    if request.method == ‘GET‘:
        return render_template(‘login.html‘)
    else:
        usern = request.form.get(‘username‘)
        passw = request.form.get(‘password‘)
        user = User.query.filter(User.username == usern).first()
        if user:
            if user.check_password(passw):
                session[‘user‘] = usern
                session[‘userid‘]=user.id
                session.permanent = True
                return redirect(url_for(‘index‘))
            else:
                return u‘password is error‘
        else:
            return u‘用户名不存在‘