views.py

import uuid
import random

from flask import Blueprint, render_template, session, request, flash, redirect, url_for, make_response
from flask_cors import CORS,cross_origin

from backend.decorators import is_authenticated
from backend.forms import BookCabForm
from backend.models import Booking, User, CabGroup
from backend.algorithm import main

from globals import maps_apikey
from globals import db, maps_apikey, SOURCE


views = Blueprint("views",__name__)

@views.route("/",methods=["GET","POST"])
@is_authenticated
def home():

    form = BookCabForm()

    if request.method == "POST":
        
        if form.validate_on_submit():
            destination = form.data["destination"]
        
            try:
                # booking having status code ongoing
                bookings_ongoing = Booking.query.filter_by(status=0).all()

                # create bookings if there are more than 4 ongoing request in queue
                user = User.query.filter_by(email=session["user"]).first()
                booking_hash = str(uuid.uuid1())
                curr_booking = Booking(
                        destination=destination,
                        user=user.id,
                        status=0,
                        booking_hash=booking_hash
                    )
                db.session.add(curr_booking)
                db.session.commit()

                if len(bookings_ongoing)>=4:
                    print(bookings_ongoing)
                    groups =  main()    

                    for group in groups:
                        name = f"grp{random.randint(1,10000)}:{random.randbytes(10000)}"
                        grp = CabGroup(name=name)
                        db.session.add(grp)
                        db.session.commit()
                        for usrAttr in group:
                        
                            booking_a = Booking.query.filter_by(id=usrAttr.booking_id).all()

                            for booking_ in booking_a:
                                
                                booking_.status = 1
                                booking_.group = grp.id
                                booking_.cost = usrAttr.cost
                                booking_.distance = usrAttr.distance

                            db.session.add_all(booking_a)
                            db.session.commit()

                            flash("Booking added")
                        
                else:
                    flash("request added successfuly")
                    
                        
            except Exception as e:
                print(e,"=========")
                flash("allocation failed try again later!")
        
        else:
            flash(form.errors)

    return render_template(
        "index.html",
        user = session["user"],
        form = form,
        google_maps_apikey = maps_apikey
    )


@views.route("/services",methods=["GET"])
@is_authenticated
def services():

    user = User.query.filter_by(email=session["user"]).first()
    if user == None:
        return redirect(url_for("auth.login"))

    curr_bookings = Booking.query.filter_by(user=user.id).all()

    return render_template(
        "services.html",
        user = session["user"],
        bookings = curr_bookings
    )

@views.route("/contact",methods=["GET"])
@is_authenticated
def contact():

    user = User.query.filter_by(email=session["user"]).first()
    if user == None:
        return redirect(url_for("auth.login"))

    return render_template(
        "contact.html",
        user = session["user"],
        MAPS_APIKEY = maps_apikey,
        SOURCE_LOCATION = SOURCE
    )


@views.route("/detail_group/<pk>",methods=["GET"])
@is_authenticated
def detail_group(pk):

    shared_cab_group = CabGroup.query.filter_by(id=pk).first()
    user = User.query.filter_by(email=session["user"]).first()
    
    my_booking = Booking.query.filter_by(group=pk,user=user.id).first()
    DROP = my_booking.destination

    if shared_cab_group == None:
        return "404 page not found \n group does not exist"

    bookings = shared_cab_group.bookings


    return render_template(
        "detail-group.html",
        user=session["user"],
        grp_bookings = bookings,
        pk=pk,
        MAPS_APIKEY = maps_apikey,
        SOURCE = SOURCE,
        DROP = DROP
    )


@views.route("/chat/<pk>",methods=["GET"])
@is_authenticated
def chat(pk):

    return render_template(
        'chat.html',
        room=pk,
        user = session["user"]
    )


@views.route("/booking-payment/<booking_hash>")
@cross_origin()
def booking_payment(booking_hash):

    booking = Booking.query.filter_by(booking_hash=booking_hash).first()
    if booking==None:
        return make_response("FAILED",404)
        
    if booking.status!=2:

        booking.status = 2
        db.session.add(booking)
        db.session.commit()


    return make_response("SUCCESS",200)