from flask import request, jsonify
from app.delivery_data import delivery_data
from app import db
from app.classes.customer import Customer_Customer
from app.classes.delivery import Delivery_Delivery
from app.classes.employee import Employee_Employee
from app.classes.cards import Card_Card
from app.classes.pricing import Pricing_Oil_Oil
from app.classes.stats_employee import Stats_Employee_Oil




@delivery_data.route("/finalize/<int:delivery_id>", methods=["PUT"])
def office_finalize_delivery(delivery_id):
    """
    This will make a delivery finalized from the driver
    """

    """
    Finalizes a delivery from office
    """
    get_delivery = db.session \
        .query(Delivery_Delivery) \
        .filter(Delivery_Delivery.id == delivery_id) \
        .first()

    get_today_price = db.session \
        .query(Pricing_Oil_Oil) \
        .order_by(Pricing_Oil_Oil.id.desc()) \
        .first()

    get_customer = db.session \
        .query(Customer_Customer) \
        .filter(Customer_Customer.id == get_delivery.customer_id) \
        .first()


    gallons_delivered = request.json["gallons_delivered"]
    card_payment = request.json["credit"]
    cash_payment = request.json["cash"]
    if request.json["credit_card_id"]:
        card_payment_id = request.json["credit_card_id"]
    else:
        card_payment_id = None


    if card_payment_id is not None:
        get_card = (db.session
                    .query(Card_Card)
                    .filter(Card_Card.id == card_payment_id)
                    .filter(Card_Card.user_id == get_customer.id)
                    .first())
        card_id_from_customer = get_card.id
    else:
        card_id_from_customer = None

    if cash_payment is True and card_payment is False:
        delivery_payment_method = 0
    elif card_payment is True and cash_payment is False:
        delivery_payment_method = 1

    elif card_payment is True and cash_payment is True:
        delivery_payment_method = 2
    else:
        delivery_payment_method = 3


    same_day_info = request.json["same_day"]
    if same_day_info is True:
        same_day_info = 1
    else:
        same_day_info = 0

    prime_info = request.json["prime"]
    if prime_info is True:
        prime_info = 1
    else:
        prime_info = 0

    get_delivery.gallons_delivered = gallons_delivered
    get_delivery.prime = prime_info
    get_delivery.same_day = same_day_info
    get_delivery.payment_type = delivery_payment_method
    get_delivery.payment_card_id = card_id_from_customer

    get_delivery.delivery_status = 10
    
    db.session.add(get_delivery)
    db.session.commit()

    return jsonify({
        "ok": True,
        'delivery': {
            'id': get_delivery.id,
        },
    }), 200



@delivery_data.route("/driver/finalize/<int:delivery_id>", methods=["PUT"])
def driver_finalize_delivery(delivery_id):
    """
    This will make a delivery finalized from the driver
    """

    """
    Finalizes a delivery from office
    """
    get_delivery = (db.session 
        .query(Delivery_Delivery) 
        .filter(Delivery_Delivery.id == delivery_id) 
        .first())

    get_today_price = (db.session 
        .query(Pricing_Oil_Oil) 
        .order_by(Pricing_Oil_Oil.id.desc()) 
        .first())

    get_stats = (db.session
                 .query(Stats_Employee_Oil)
                 .filter(Stats_Employee_Oil == get_delivery.driver_employee_id)
                 .first())

    gallons_delivered = request.json["gallons_delivered"]

    if request.json["driver_employee_id"]:
        delivery_driver_id = request.json["delivery_driver_id"]
        get_driver = (db.session
                        .query(Employee_Employee)
                        .filter(Employee_Employee.id == delivery_driver_id)
                        .first())
        same_driver = False
    else:
        same_driver = True
        
    customer_filled = request.json["customer_filled"]
    if customer_filled is True:
        customer_filled = 1
    else:
        customer_filled = 0
        
    if request.json["cash_amount"]:
        cash_amount = request.json["cash_amount"]
    else:
        cash_amount = None

    prime_info = request.json["prime"]
    if prime_info is True:
        prime_info = 1
        current_primes = get_stats.total_primes
        newprimes = current_primes + 1
        get_stats.total_primes = newprimes
    else:
        prime_info = 0

    get_delivery.gallons_delivered = gallons_delivered
    
    get_delivery.prime = prime_info
    get_delivery.customer_filled = customer_filled
    get_delivery.cash_recieved = cash_amount
    
    if same_driver == False:
        get_delivery.driver_last_name = get_driver.employee_last_name
        get_delivery.driver_first_name = get_driver.employee_first_name
        get_delivery.driver_employee_id = get_driver.id
        
        
    # update stats
    current_deliveres = get_stats.total_deliveries + 1
    get_stats.total_deliveries = current_deliveres
    
    current_gallons_delivered = get_stats.total_gallons_delivered + gallons_delivered
    get_stats.total_gallons_delivered = current_gallons_delivered
        
        
        
    # update delivery status to delivered
    if get_delivery.payment_type == 0:
        get_delivery.delivery_status = 10
    else:
        #office needs to confirm credit card
        get_delivery.delivery_status = 1
  
    db.session.add(get_stats)
    db.session.add(get_delivery)
    db.session.commit()

    return jsonify({
        "ok": True,
        'delivery': {
            'id': get_delivery.id,
        },
    }), 200