代码改变世界

dada

2017-12-22 21:54  019王泽  阅读(170)  评论(0)    收藏  举报
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(20))

    @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

@app.route('/login/', methods=['GET', 'POST'])
def login():
    if request.method == 'GET':
        return render_template("login.html")
    else:
        usern = request.form.get('username')
        password1 = request.form.get('password')
        user = User.query.filter(User.username == usern).first()
        if user:
            if user.check_password(password1):
                session['user'] = usern
                session['id'] = user.id
                session.permanent = True
                return redirect(url_for('index'))
            else:
                return '密码错误'
        else:
            return '用户名不存在'