from flask import request, jsonify
from flask_login import current_user
from datetime import date, timedelta, time
from app.delivery_status import deliverystatus
from app import db
from datetime import datetime
from app.classes.customer import Customer_Customer
from app.classes.delivery import (Delivery_Delivery,
                                  Delivery_Delivery_schema,
                                  Delivery_Notes_Driver,
                                  Delivery_Payment,
                                  Delivery_Payment_schema,
                                  )
from app.classes.cards import Card_Card
from app.classes.pricing import Pricing_Oil_Oil
from app.classes.auth import Auth_User
from app.classes.pricing import Pricing_Service_General


@deliverystatus.route("/today/driver/<int:user_id>", methods=["GET"])
def get_deliveries_driver_day(user_id):
    """
    Get deliveries for driver that day
    """
    get_delivery = (db.session
        .query(Delivery_Delivery)
        .filter(Delivery_Delivery.driver_employee_id == user_id)
        .filter(Delivery_Delivery.expected_delivery_date == date.today())
        .all())

    delivery_schema = Delivery_Delivery_schema(many=True)
    return jsonify(delivery_schema.dump(get_delivery))

@deliverystatus.route("/tommorrow/driver/<int:user_id>", methods=["GET"])
def get_deliveries_driver_tommorrow(user_id):
    """
    Get deliveries for driver tommrrow
    """
    tomm = datetime.now() + timedelta(days=1)

    get_delivery = (db.session
        .query(Delivery_Delivery)
        .filter(Delivery_Delivery.driver_employee_id == user_id)
        .filter(Delivery_Delivery.expected_delivery_date == tomm)
        .all())

    delivery_schema = Delivery_Delivery_schema(many=True)
    return jsonify(delivery_schema.dump(get_delivery))

@deliverystatus.route("/waiting/driver/<int:user_id>", methods=["GET"])
def get_deliveries_driver_waiting(user_id):
    """
    waiting deliveries scheduled out
    """
    get_delivery = (db.session
        .query(Delivery_Delivery)
        .filter(Delivery_Delivery.driver_employee_id == user_id)
        .filter(Delivery_Delivery.delivery_status == 1)
        .all())

    delivery_schema = Delivery_Delivery_schema(many=True)
    return jsonify(delivery_schema.dump(get_delivery))