diff --git a/hiboo/account/forms.py b/hiboo/account/forms.py index 11ddbc9cc4e778050832b5b1e54d1ac686015cdb..83d1cf860a3c0741eeabed6aad1ce39f9c2a16fe 100644 --- a/hiboo/account/forms.py +++ b/hiboo/account/forms.py @@ -4,10 +4,10 @@ from flask_babel import lazy_gettext as _ import flask_wtf - class LoginForm(flask_wtf.FlaskForm): username = fields.StringField(_('Username'), [validators.DataRequired()]) password = fields.PasswordField(_('Password'), [validators.DataRequired()]) + remember_me = fields.BooleanField("Remember me", default=False) submit = fields.SubmitField(_('Sign in')) diff --git a/hiboo/account/login.py b/hiboo/account/login.py index 0b40e7d0e4b7ae31844aa94da800f2d066197919..37fefe095e65b0c9340d53c2cb3b8ac17faa9251 100644 --- a/hiboo/account/login.py +++ b/hiboo/account/login.py @@ -1,11 +1,12 @@ from hiboo import models, utils, security from hiboo.account import blueprint, forms from flask_babel import lazy_gettext as _ +from flask import session import datetime import flask_login import flask - +import sys @blueprint.route("/signin", methods=["GET", "POST"]) def signin(): @@ -14,6 +15,8 @@ def signin(): user = models.User.login(form.username.data, form.password.data) if user: flask_login.login_user(user) + if form.remember_me.data == True: + session.permanent = True return flask.redirect(utils.url_or_intent(".home")) else: flask.flash(_("Wrong credentials"), "danger")