from werkzeug.security import generate_password_hash,check_password_hash
app=Flask(__name__)
app.config.from_object(config)
db=SQLAlchemy(app)
# 创建用户模型
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))
@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:
username = request.form.get('username') # 获取form中的数据
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' #输出错误
1.更新User对象,设置对内的_password
2.编写对外的password
3.密码验证:
4.登录验证: