from fastapi import APIRouter, Request from datetime import datetime from database import session from pyowm import OWM from decimal import Decimal from app.models.auto import Auto_Delivery from app.models.delivery import Delivery from app.script.update_auto import calc_home_factor router = APIRouter( prefix="/confirm", tags=["confirm"], responses={404: {"description": "Not found"}}, ) @router.put("/delivery") async def add_delivery(request: Request): now = datetime.utcnow request_body = await request.json() gallons_delivered = request_body['gallons'] gallons_delivered = Decimal(gallons_delivered) delivery_id = request_body['delivery_id'] get_delivery = (session .query(Delivery) .filter(Delivery.id == delivery_id) .first()) get_auto_delivery = (session .query(Auto_Delivery) .filter(Auto_Delivery.customer_id == get_delivery.customer_id) .first()) gallons_put_in_home = Decimal(gallons_delivered) customer_home_factor = get_auto_delivery.house_factor new_home_factor = calc_home_factor(gallons_put_in_home = gallons_put_in_home, current_house_factor=customer_home_factor) get_auto_delivery.house_factor = new_home_factor get_auto_delivery.tank_height = 'Full' get_auto_delivery.last_fill = now get_auto_delivery.estimated_gallons_left = 240 get_auto_delivery.estimated_gallons_left_prev_day = 240 get_auto_delivery.auto_status = 0 session.add(get_auto_delivery) session.commit() return ({"ok": True}), 200