def test_booked_dates(self):
b1 = Booking(pet_id=48038022,
renter_id=self.rid1,
startTime="2020-5-25",
endTime="2020-5-28",
rating=2)
b2 = Booking(pet_id=48038022,
renter_id=self.rid1,
startTime="2020-6-5",
endTime="2020-6-6",
rating=4)
db.session.add(b1)
db.session.add(b2)
db.session.commit()
book_dates = Booking.booked_dates(id=48038022)
self.assertEqual(len(book_dates), 6)
def pet(pet_id):
"return pet page"
form = PetForm()
rating = Booking.avg_rating(id=pet_id)
if not g.renter:
flash("Sign in", "danger")
return render_template("pet.html", id=pet_id)
booked_dates = Booking.booked_dates(id=pet_id)
print("_____BOOKED", booked_dates)
if form.validate_on_submit():
isAvaliable = Booking.is_avaliable(id=pet_id,
start=form.start_date.data,
end=form.end_date.data)
print("ISAVALIABLE", isAvaliable)
if isAvaliable:
booking = Booking(pet_id=pet_id,
renter_id=g.renter.id,
startTime=form.start_date.data,
endTime=form.end_date.data)
db.session.add(booking)
db.session.commit()
print("____red")
return redirect(f"/renter")
else:
flash("already booked pick another date", "danger")
return render_template("pet.html",
id=pet_id,
form=form,
rating=rating,
booked=booked_dates)
else:
return render_template("pet.html",
id=pet_id,
form=form,
rating=rating,
booked=booked_dates)
