예제 #1
0
def signup():
    form = UserForm()
    if request.method == "POST":
        if form.validate():
            try:
                new_user = User(email=form.email.data,
                                username=form.username.data,
                                name=form.name.data,
                                address=form.address.data,
                                city=form.city.data,
                                state=form.state.data,
                                zipcode=form.zipcode.data,
                                password=form.password.data)
                if form.image_url.data:
                    new_user.image_url = form.image_url.data
                db.session.add(new_user)
                db.session.commit()
                login_user(new_user)
            except IntegrityError as e:
                flash({
                    'text': "Username or email already taken",
                    'status': 'danger'
                })
                return render_template('users/signup.html', form=form)
            return redirect(url_for('root'))
    return render_template('users/signup.html', form=form)
예제 #2
0
def signup():
    form = UserForm()
    if request.method == 'POST':
        if form.validate():
            try:
                new_user = User(username=form.username.data,
                                email=form.email.data,
                                password=form.password.data)
                if form.image_url.data:
                    new_user.image_url = form.image_url.data
                db.session.add(new_user)
                db.session.commit()
                login_user(new_user)
            except IntegrityError as e:
                flash({'text': 'Username already taken', 'status': 'danger'})
                return render_template('users/signup.html', form=form)
            return redirect(url_for('root'))
    return render_template('users/signup.html', form=form)