def before_request():
g.user = current_user
uri_pattern = request.url_rule
if current_user.is_authenticated and (uri_pattern == '/'
or uri_pattern == '/login'
or uri_pattern == '/register'):
redirect_to('home')
def register():
form = RegisterForm()
if request.method == 'POST':
if form.validate_on_submit():
if User.query.filter(User.email == form.email.data).count() > 0:
form.email.errors.append("Email address already in use.")
return view('register', form)
user = User(
name=form.name.data,
email=form.email.data,
password=form.password.data,
phone_number="+{0}{1}".format(form.country_code.data, form.phone_number.data),
area_code=str(form.phone_number.data)[0:3])
db.session.add(user)
db.session.commit()
login_user(user, remember=True)
return redirect_to('home')
else:
return view('register', form)
return view('register', form)
def new_reservation(property_id):
vacation_property = None
form = ReservationForm()
form.property_id.data = property_id
if request.method == 'POST':
if form.validate_on_submit():
guest = User.query.get(current_user.get_id())
vacation_property = VacationProperty.query.get(
form.property_id.data)
reservation = Reservation(form.message.data, vacation_property,
guest)
db.session.add(reservation)
db.session.commit()
reservation.notify_host()
return redirect_to('properties')
if property_id is not None:
vacation_property = VacationProperty.query.get(property_id)
return view_with_params('reservation',
vacation_property=vacation_property,
form=form)
def new_property():
form = VacationPropertyForm()
if request.method == 'POST':
if form.validate_on_submit():
host = User.query.get(current_user.get_id())
property = VacationProperty(form.description.data, form.image_url.data, host)
db.session.add(property)
db.session.commit()
return redirect_to('properties')
return view('property_new', form)
def new_property():
form = VacationPropertyForm()
if request.method == "POST":
if form.validate_on_submit():
host = User.query.get(current_user.get_id())
property = VacationProperty(form.description.data, form.image_url.data, host)
db.session.add(property)
db.session.commit()
return redirect_to("properties")
return view("property_new", form)
def login():
form = LoginForm()
if request.method == "POST":
if form.validate_on_submit():
candidate_user = User.query.filter(User.email == form.email.data).first()
if candidate_user is None or not bcrypt.check_password_hash(candidate_user.password, form.password.data):
form.password.errors.append("Invalid credentials.")
return view("login", form)
login_user(candidate_user, remember=True)
return redirect_to("home")
return view("login", form)
def login():
form = LoginForm()
if request.method == 'POST':
if form.validate_on_submit():
candidate_user = User.query.filter(User.email == form.email.data).first()
if candidate_user is None or not bcrypt.check_password_hash(candidate_user.password,
form.password.data):
form.password.errors.append("Invalid credentials.")
return view('login', form)
login_user(candidate_user, remember=True)
return redirect_to('home')
return view('login', form)
def new_reservation(property_id):
vacation_property = None
form = ReservationForm()
form.property_id.data = property_id
if request.method == "POST":
if form.validate_on_submit():
guest = User.query.get(current_user.get_id())
vacation_property = VacationProperty.query.get(form.property_id.data)
reservation = Reservation(form.message.data, vacation_property, guest)
db.session.add(reservation)
db.session.commit()
reservation.notify_host()
return redirect_to("properties")
if property_id is not None:
vacation_property = VacationProperty.query.get(property_id)
return view_with_params("reservation", vacation_property=vacation_property, form=form)
def register():
form = RegisterForm()
if request.method == "POST":
if form.validate_on_submit():
if User.query.filter(User.email == form.email.data).count() > 0:
form.email.errors.append("Email address already in use.")
return view("register", form)
user = User(
name=form.name.data,
email=form.email.data,
password=form.password.data,
phone_number="+{0}{1}".format(form.country_code.data, form.phone_number.data),
)
db.session.add(user)
db.session.commit()
login_user(user, remember=True)
return redirect_to("home")
else:
return view("register", form)
return view("register", form)
def logout():
logout_user()
return redirect_to('home')
def logout():
logout_user()
return redirect_to("home")
def before_request():
g.user = current_user
uri_pattern = request.url_rule
if current_user.is_authenticated and (uri_pattern == "/" or uri_pattern == "/login" or uri_pattern == "/register"):
redirect_to("home")
def test_redirect_to_redirects_to_same_location_of_redirect(self):
# assert
assert redirect_to('home').location == redirect(url_for('home')).location
