Flask-wtform区分大小写的问题

时间:2019-07-29 22:43:03

标签: flask flask-sqlalchemy flask-wtforms

我有一个带有用户登录系统的flask应用程序。我现在注意到一些与用于注册和登录该站点的电子邮件地址的区分大小写有关的问题。 TL; DR问题:如何将Flask Form的小写形式用于评估用户是否存在于数据库中?

示例

假设用户John Smith在该网站上注册了一个帐户。他在John.Smith@gmail.com上注册,但随后在John.smith@gmail.com上登录。截至目前,该网站将其视为两个不同的用户。我希望这被视为同一用户。

在Python(我的专长)中,我只拿str(form.email.data).lower()作为注册表格,然后将其写入数据库中的用户表。然后,当有人尝试登录时,我会做同样的事情。换句话说,我将始终考虑注册和登录的小写字母。

但是,尽管它与Python相似,但我仍在Jinja中努力实现它。

register.html的电子邮件部分:

                <div class="form-group">
                    {{ form.email.label(class="form-control-label") }}
                    {% if form.email.errors %}
                        {{ form.email(class="form-control form-control-lg is-invalid") }}
                        <div class="invalid-feedback">
                            {% for error in form.email.errors %}
                                <span>{{ error }}</span>
                            {% endfor %}
                        </div>
                    {% else %}
                        {{ form.email(class="form-control form-control-lg") }}
                    {% endif %}
                </div>

routes.py中的注册码:

@app.route("/register", methods=['GET', 'POST'])
def register():
    if current_user.is_authenticated:
        return redirect(url_for('home'))

    form = RegistrationForm()

    admins = ['admin@admin.com',
              'admin@admin.com']


    if form.validate_on_submit():

        new_user_email = form.email.data.lower() ###<-----NOTE THAT THIS IS LOWER CASE
        admin_test = True if new_user_email in admins else False
        invited_test = Invites.query.filter(Invites.invited_email == new_user_email).count()
        print(admin_test, invited_test)

        if (invited_test == 0) & (admin_test == False): # Not invited and not and admin --> REJECT
            flash(
                'Sorry but it looks like you were not invited. Please contact a current member for an invitation.',
                'warning')
        elif (invited_test > 0) | (admin_test == True): ## if they have been invited OR if they are admin
            hashed_password = bcrypt.generate_password_hash(form.password.data).decode('utf-8')

            #create User object
            user = User(username=form.username.data.lower(), ###<----NOTE LOWER CASE
                        email=form.email.data.lower(), ###<----NOTE LOWER CASE
                        password=hashed_password)

            #write to db
            db.session.add(user)
            db.session.commit()

            #alert user
            flash('Welcome to. Enjoy the platform.', 'success')
            return redirect(url_for('login'))

        else:
            #Not invited condition
            flash('Sorry but it looks like you were not invited to __. Please contact a current member for an invitation.', 'warning')

    return render_template('register.html', title='Register', form=form)

login.html的电子邮件部分:

            <div class="form-group">
                {{ form.email.label(class="form-control-label") }}
                {% if form.email.errors %}
                    {{ form.email(class="form-control form-control-lg is-invalid") }}
                    <div class="invalid-feedback">
                        {% for error in form.email.errors %}
                            <span>{{ error }}</span>
                        {% endfor %}
                    </div>
                {% else %}
                    {{ form.email(class="form-control form-control-lg") }}
                {% endif %}
            </div>

1 个答案:

答案 0 :(得分:0)

您似乎在这里正确的道路上。可以帮助您区分大小写的两件事:

Jinja2

您可以将过滤器lowerupper传递到模板(请注意,您也可以注册自定义过滤器并在jinja2上使用它)。

#lower
{% user.email|lower %} 

#upper
{% user.email|upper %} 

SQLAlchemy

您可以传递==来代替使用ilike进行直接比较,从而使其本质上不区分大小写:

Invites.query.filter(Invites.invited_email.ilike(new_user_email)).count()