密码保护

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=True,autoincrement=True)
    username=db.Column(db.String(20),nullable=False)
    _password=db.Column(db.String(200),nullable=False)#内部使用
    nickname=db.Column(db.String(50))

 

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)

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

 

3.密码验证方法：

    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 \'password error.\'
        else:
            return \'username is not existed.\'