def register_post(): # Create the user object to store our data user = User() user.first_name = request.form.get('first_name') user.last_name = request.form.get('last_name') user.email = request.form.get('email') user.phone = request.form.get('phone') user.password = request.form.get('password') user.role = request.form.get('role') existing = db_session.query(User).filter_by(email=user.email).first() if existing: flash("Email already in use", "error") return redirect(url_for("user_login_get")) # Add the user object to the database db_session.add(user) # Save the user in the database db_session.commit() # # Redirect user to landing page return redirect(url_for("traveler_view_trip"))
def authenticate(): if request.form['btn'] == "login": form_login = forms.LoginForm(request.form) if not form_login.validate(): flash("Incorrect username or password") return render_template("create_info") email = form_login.email.data password = form_login.password.data user = User.query.filter_by(email=email).first() if not user or not user.authenticate(password): flash("Incorrect username or password") return render_template("login.html") login_user(user) session["user_id"] = user.id return redirect(request.args.get("next", url_for("browse"))) elif request.form['btn'] == "register": password = request.form.get("password") email = request.form.get("email") first_name=request.form.get("first_name") last_name=request.form.get("last_name") new_user = User(email=email) new_user.set_password(password=password) new_user.first_name=first_name new_user.last_name=last_name model.session.add(new_user) model.session.commit() return redirect(url_for("create_info")) elif request.form['btn'] == "fb_login": fb_name = request.form.get("name") if current_user.is_anonymous == False: current_user.fb_id = fb_id current_user.fb_img_url = "https://graph.facebook.com/%s/picture?type=small" % fb_id print current_user.fb_img_url model.session.commit() model.session.refresh() return redirect(url_for("browse"))
def register_submit(): """ Register new user """ firstname = request.form.get('firstname') lastname = request.form.get('lastname') email = request.form.get('newemail') password1 = request.form.get('password1') if User.query.filter_by(email=email).first(): flash("A user already exists with this email") return render_template("register.html") else: new_user = User(email, password1) db.session.add(new_user) new_user.first_name = firstname new_user.last_name = lastname db.session.commit() if not session.get('new_user.user_id'): session['user_id'] = new_user.user_id flash("You have been registered successfully.") return redirect("/dashboard")