コード例 #1
0
ファイル: flask_app.py プロジェクト: KirillTeplekov/Website
def registration():
    if 'username' in session:
        pass
        session.pop('username', 0)
        session.pop('user_id', 0)

    check_form = InviteForm()
    if check_form.validate_on_submit():
        if check_form.invite.data in Invite.query.all():
            reg_form = RegistrationForm()
            if reg_form.validate_on_submit():
                # Check for the existence of the user in db
                user = Admin.query.filter_by(login=reg_form.login.data).first()
                if not user:
                    # Create new user
                    new_user = Admin(login=reg_form.login.data,
                                     name=reg_form.name.data,
                                     surname=reg_form.surname.data,
                                     email=reg_form.email.data,
                                     password=reg_form.password.data)
                    db.session.add(new_user)
                    db.session.commit()
                    session['username'] = new_user.login
                    session['user_id'] = new_user.id
                    return redirect('/index')
                else:
                    return render_template(
                        'error_page.html',
                        title='Ошибка',
                        error='Такой пользователь уже существует',
                        link_dict=link_dict,
                        name='')
            return render_template('registration.html',
                                   title='Регистрация',
                                   form=reg_form,
                                   link_dict=link_dict,
                                   name='')
        else:
            return render_template('error_page.html',
                                   title='Ошибка',
                                   error='Неверный код',
                                   link_dict=link_dict,
                                   name='')
    return render_template('invite_check.html',
                           title='Проверка кода-приглашения',
                           form=check_form,
                           link_dict=link_dict,
                           name='')
コード例 #2
0
def invite():
	form = InviteForm()
	if form.validate_on_submit():
		invite_slack_result = invite_to_slack(form.email.data)
		print(invite_slack_result)
		if invite_slack_result["ok"]:
			flash('Invite send at {}'.format(form.email.data, 'success'))
			return(redirect(url_for('welcome')))
		elif invite_slack_result["ok"] is False and invite_slack_result["error"] == "already_in_team":
			flash('There is already a user in the team with the email {}'.format(form.email.data, 'error'))
			return(redirect(url_for('already_in_team_error')))
		elif invite_slack_result["ok"] is False and invite_slack_result["error"] == "already_invited":
			flash('The invitation has already been sent at {}'.format(form.email.data, 'error'))
			return(redirect(url_for('already_invited_error')))
		elif invite_slack_result["ok"] is False and invite_slack_result["error"] == "already_invited":
			flash('The email {} is invalid'.format(form.email.data, 'error'))
			flash('Note that slack does not recognise some email addresses'.format())
			return(redirect(url_for('invalid_email_error')))

	return render_template('invite.html', title='Invite', form=form)