コード例 #1
0
    def test_is_avaliable(self):

        b1 = Booking(pet_id=48038022,
                     renter_id=self.rid1,
                     startTime=datetime.date(2020, 5, 3),
                     endTime=datetime.date(2020, 6, 23),
                     rating=2)

        db.session.add(b1)
        db.session.commit()

        isAvaliable = Booking.is_avaliable(id=48038022,
                                           start=datetime.date(2020, 6, 3),
                                           end=datetime.date(2020, 6, 5))
        self.assertFalse(isAvaliable)
コード例 #2
0
ファイル: app.py プロジェクト: davidleecodes/Capstone1
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)