WTForms: две формы на одной странице?

у меня есть динамическая веб-страница, которая должна обрабатывать две формы: a форма входа и форма регистрации. Я использую WTForms для обработки двух форм, но у меня возникли проблемы с его работой, так как обе формы отображаются на одной странице.

следующий код логин мой сайт:

PYTHON:
class Login(Form):
    login_user = TextField('Username', [validators.Required()])
    login_pass = PasswordField('Password', [validators.Required()])

@application.route('/index', methods=('GET', 'POST'))
def index():
    l_form = Login(request.form, prefix="login-form")
    if request.method == 'POST' and l_form.validate():
        check_login = cursor.execute("SELECT * FROM users WHERE username = '%s' AND pwd = '%s'"
        % (l_form.login_user.data, hashlib.sha1(l_form.login_pass.data).hexdigest()))
        if check_login == True:
            conn.commit()
            return redirect(url_for('me'))
    return render_template('index.html', lform=l_form)


HTML:
<form name="lform" method="post" action="/index">
    {{ lform.login_user }}
    {{ lform.login_pass }}
    <input type="submit" value="Login" />
</form>

следующий код зарегистрироваться форма моего страницы:

PYTHON:
class Register(Form):
    username = TextField('Username', [validators.Length(min=1, max = 12)])
    password = PasswordField('Password', [
        validators.Required(),
        validators.EqualTo('confirm_password', message='Passwords do not match')
    ])
    confirm_password = PasswordField('Confirm Password')
    email = TextField('Email', [validators.Length(min=6, max=35)])

@application.route('/index', methods=('GET','POST'))
def register():
    r_form = Register(request.form, prefix="register-form")
    if request.method == 'POST' and r_form.validate():
        check_reg = cursor.execute("SELECT * FROM users WHERE username = '%s' OR `e-mail` = '%s'"
        % (r_form.username.data, r_form.email.data))

        if check_reg == False:
            cursor.execute("INSERT into users (username, pwd, `e-mail`) VALUES ('%s','%s','%s')"
            % (r_form.username.data, hashlib.sha1(r_form.password.data).hexdigest(), check_email(r_form.email.data)))
            conn.commit()
            return redirect(url_for('index'))
    return render_template('index.html', rform=r_form)


HTML:
<form name="rform" method="post" action="/index">
    {{ rform.username }}
    {{ rform.email }}
    {{ rform.password }}
    {{ rform.confirm_password }}
    <input type="submit" value="Register />
</form>

Я получаю следующую ошибку, когда я иду вперед и загружаю веб-страницу:

    Traceback (most recent call last):
  File "C:UsersHTVal_000DesktopinnoCMSvirtualenvlibsite-packagesflaskapp.py", line 1836, in __call__
    return self.wsgi_app(environ, start_response)
  File "C:UsersHTVal_000DesktopinnoCMSvirtualenvlibsite-packagesflaskapp.py", line 1820, in wsgi_app
    response = self.make_response(self.handle_exception(e))
  File "C:UsersHTVal_000DesktopinnoCMSvirtualenvlibsite-packagesflaskapp.py", line 1403, in handle_exception
    reraise(exc_type, exc_value, tb)
  File "C:UsersHTVal_000DesktopinnoCMSvirtualenvlibsite-packagesflaskapp.py", line 1817, in wsgi_app
    response = self.full_dispatch_request()
  File "C:UsersHTVal_000DesktopinnoCMSvirtualenvlibsite-packagesflaskapp.py", line 1477, in full_dispatch_request
    rv = self.handle_user_exception(e)
  File "C:UsersHTVal_000DesktopinnoCMSvirtualenvlibsite-packagesflaskapp.py", line 1381, in handle_user_exception
    reraise(exc_type, exc_value, tb)
  File "C:UsersHTVal_000DesktopinnoCMSvirtualenvlibsite-packagesflaskapp.py", line 1475, in full_dispatch_request
    rv = self.dispatch_request()
  File "C:UsersHTVal_000DesktopinnoCMSvirtualenvlibsite-packagesflaskapp.py", line 1461, in dispatch_request
    return self.view_functions[rule.endpoint](**req.view_args)
  File "C:UsersHTVal_000DesktopinnoCMSmain.py", line 36, in index
    return render_template('index.html', lform=l_form)
  File "C:UsersHTVal_000DesktopinnoCMSvirtualenvlibsite-packagesflasktemplating.py", line 128, in render_template
    context, ctx.app)
  File "C:UsersHTVal_000DesktopinnoCMSvirtualenvlibsite-packagesflasktemplating.py", line 110, in _render
    rv = template.render(context)
  File "C:UsersHTVal_000DesktopinnoCMSvirtualenvlibsite-packagesjinja2environment.py", line 969, in render
    return self.environment.handle_exception(exc_info, True)
  File "C:UsersHTVal_000DesktopinnoCMSvirtualenvlibsite-packagesjinja2environment.py", line 742, in handle_exception
    reraise(exc_type, exc_value, tb)
  File "C:UsersHTVal_000DesktopinnoCMStemplatesdefaultindex.html", line 52, in top-level template code
    {{ rform.username }}
  File "C:UsersHTVal_000DesktopinnoCMSvirtualenvlibsite-packagesjinja2environment.py", line 397, in getattr
    return getattr(obj, attribute)
UndefinedError: 'rform' is undefined

из того, что я понимаю, существует конфликт между формами, потому что в соответствии с обратной связью:

return render_template('index.html', lform=l_form)

возвращает следующую ошибку:

UndefinedError: 'rform' is undefined

когда скрипт видит:

{{ rform.username }}
{{ rform.email }}
{{ rform.password }}
{{ rform.confirm_password }}

но он полностью игнорирует:

{{ lform.login_user }}
{{ lform.login_pass }}

это может быть немного запутанным, я тоже запутался, и я надеюсь, что кто-то сталкивался с этой проблемой раньше, чтобы я тоже мог ее решить.

1 ответов


Это немного запутанно, потому что вы визуализируете индекс.html как на index (), так и на register (), и оба регистрируют один и тот же маршрут (@application.route('/index')). При отправке формы /index, только один из них только называются. Вы можете либо

  • поместите всю свою логику в одну индексную функцию и посмотрите, какая форма (если есть) действительна.
  • отделите свою логику и отправьте только соответствующую форму

вообще, вы хотите отделить логику, даже если вы хотите чтобы показать логин и регистрацию на одной странице. Поэтому я постараюсь показать вам в правильном направлении :-)

например, сначала отделите свой логин и зарегистрируйте представления, которые теперь будут проверять только логику для формы, которая их касается:

class Login(Form):
    login_user = TextField('Username', [validators.Required()])
    login_pass = PasswordField('Password', [validators.Required()])

class Register(Form):
    username = TextField('Username', [validators.Length(min=1, max = 12)])
    password = PasswordField('Password', [
        validators.Required(),
        validators.EqualTo('confirm_password', message='Passwords do not match')
    ])
    confirm_password = PasswordField('Confirm Password')
    email = TextField('Email', [validators.Length(min=6, max=35)])

@application.route('/login', methods=['POST'])
def index():
    l_form = Login(request.form, prefix="login-form")
    if l_form.validate():
        check_login = cursor.execute("SELECT * FROM users WHERE username = '%s' AND pwd = '%s'"
            % (l_form.login_user.data, hashlib.sha1(l_form.login_pass.data).hexdigest()))
        if check_login == True:
            conn.commit()
            return redirect(url_for('me'))
    return render_template('index.html', lform=l_form, rform=Register())

@application.route('/register', methods=['POST'])
def register():
    r_form = Register(request.form, prefix="register-form")
    if r_form.validate():
        check_reg = cursor.execute("SELECT * FROM users WHERE username = '%s' OR `e-mail` = '%s'"
            % (r_form.username.data, r_form.email.data))

        if check_reg == False:
            cursor.execute("INSERT into users (username, pwd, `e-mail`) VALUES ('%s','%s','%s')"
                % (r_form.username.data, hashlib.sha1(r_form.password.data).hexdigest(), check_email(r_form.email.data)))
            conn.commit()
            return redirect(url_for('index'))
    return render_template('index.html', lform=Login(), rform=r_form)

@application.route('/index')
def index():
    # If user is logged in, show useful information here, otherwise show login and register
    return render_template('index.html', lform=Login(), rform=Register())

затем создайте индекс.html, который показывает обе формы и отправляет их в правильном направлении.

<form name="lform" method="post" action="{{ url_for('login') }}">
    {{ lform.login_user }}
    {{ lform.login_pass }}
    <input type="submit" value="Login" />
</form>

<form name="rform" method="post" action="{{ url_for('register') }}">
    {{ rform.username }}
    {{ rform.email }}
    {{ rform.password }}
    {{ rform.confirm_password }}
    <input type="submit" value="Register" />
</form>

код не тестировался, поэтому могут быть ошибки, но я надеюсь, что он пошлет вас в правильном направлении. Заметьте, что мы передайте lform и rform во всех вызовах для рендеринга ('index.HTML-код', ...).

дополнительные простые способы улучшения / рефакторинга: используйте функцию для проверки существующего пользователя (ваш SELECT оператор) и использовать Jinja2 включает или макросы для отдельных форм в шаблонах.