from flask import jsonify
from sqlalchemy import func
from datetime import date
from app.stats import stats
from app import db
from app.classes.delivery import Delivery_Delivery


@stats.route("/gallons/total/<int:driver_id>", methods=["GET"])
def total_gallons_delivered_driver(driver_id):
    gallons_list = []
    
    total_gallons = db.session\
                    .query(Delivery_Delivery)\
                    .filter(Delivery_Delivery.driver_employee_id ==  driver_id)\
                    .all()
                      
    for f in total_gallons:
        gallons_list.append(f.gallons_delivered)
    sum_of_gallons = (sum(gallons_list))
  
    return jsonify({"ok": True,
                    'data': sum_of_gallons,
                    }), 200

@stats.route("/delivery/total/<int:driver_id>", methods=["GET"])
def total_deliveries_driver(driver_id):
    print(driver_id)
    total_stops = (db.session
                        .query(Delivery_Delivery)
                        .filter(Delivery_Delivery.driver_employee_id == driver_id)
                        .count())
    print(total_stops)
    return jsonify({"ok": True,
                    'data': total_stops,
                    }), 200