密码保护

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)

 

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

 

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:
        username = request.form.get(‘username‘)  
        password = request.form.get(‘password‘)
        nickname = request.form.get(‘nickname‘)
        user = User.query.filter(User.username == username).first()
        # id = User.query.filter(User.id == id).first()
        if user:
            if user.check_password(password):
                session[‘user‘] = username
                # session[‘id‘] = id
                session.permanent=True
                return redirect(url_for(‘index‘)) 
            else:
                return u‘password error‘
        else:
            return u‘error username or password‘