182 lines
6.8 KiB
Python
182 lines
6.8 KiB
Python
from fastapi import APIRouter, Request, HTTPException
|
|
from datetime import date
|
|
from database import session
|
|
from decimal import Decimal
|
|
|
|
# Import your models
|
|
from app.models.auto import Auto_Delivery, Tickets_Auto_Delivery
|
|
from app.models.pricing import Pricing_Oil_Oil
|
|
|
|
# Import the new estimator class
|
|
from app.script.fuel_estimator import FuelEstimator
|
|
|
|
router = APIRouter(
|
|
prefix="/confirm",
|
|
tags=["confirm"],
|
|
responses={404: {"description": "Not found"}},
|
|
)
|
|
|
|
|
|
@router.put("/auto/update/{autoid}")
|
|
async def update_auto(autoid: int, request: Request):
|
|
"""
|
|
Confirms a delivery, updates the ticket, and triggers the self-correcting
|
|
K-Factor (house_factor) refinement.
|
|
"""
|
|
try:
|
|
request_body = await request.json()
|
|
gallons_delivered = Decimal(request_body['gallons_delivered'])
|
|
|
|
auto_customer = session.query(Auto_Delivery).filter(Auto_Delivery.id == autoid).first()
|
|
if not auto_customer:
|
|
raise HTTPException(status_code=404, detail="Auto Delivery profile not found")
|
|
|
|
if auto_customer.open_ticket_id is None:
|
|
raise HTTPException(status_code=400, detail="No open ticket found for this auto-delivery. Cannot confirm delivery.")
|
|
|
|
ticket = session.query(Tickets_Auto_Delivery).filter(Tickets_Auto_Delivery.id == auto_customer.open_ticket_id).first()
|
|
if not ticket:
|
|
raise HTTPException(status_code=404, detail=f"Open ticket with ID {auto_customer.open_ticket_id} not found.")
|
|
|
|
ticket.gallons_delivered = gallons_delivered
|
|
ticket.total_amount_customer = ticket.price_per_gallon * gallons_delivered
|
|
|
|
estimator = FuelEstimator(session=session)
|
|
estimator.refine_factor_after_delivery(ticket=ticket)
|
|
|
|
session.add(ticket)
|
|
session.add(auto_customer)
|
|
|
|
session.commit()
|
|
|
|
return {"ok": True, "message": "Delivery confirmed and customer factor refined successfully."}
|
|
|
|
except Exception as e:
|
|
session.rollback()
|
|
raise HTTPException(status_code=500, detail=f"An internal error occurred: {str(e)}")
|
|
|
|
|
|
@router.post("/auto/create/{autoid}")
|
|
async def create_auto_ticket(autoid: int, request: Request):
|
|
"""
|
|
Creates a new delivery ticket and links it to the auto-delivery customer profile.
|
|
"""
|
|
request_body = await request.json()
|
|
gallons_delivered = request_body['gallons_delivered'] # Estimated gallons for pre-auth
|
|
gallons_delivered = Decimal(gallons_delivered)
|
|
payment_type = request_body.get('payment_type')
|
|
payment_card_id = request_body.get('payment_card_id')
|
|
payment_status = request_body.get('payment_status')
|
|
|
|
get_auto_delivery = session.query(Auto_Delivery).filter(Auto_Delivery.id == autoid).first()
|
|
get_todays_price = session.query(Pricing_Oil_Oil).order_by(Pricing_Oil_Oil.id.desc()).first()
|
|
|
|
todays_price = Decimal(get_todays_price.price_for_customer)
|
|
total_amount = gallons_delivered * todays_price
|
|
|
|
create_new_ticket = Tickets_Auto_Delivery(
|
|
customer_id=get_auto_delivery.customer_id,
|
|
account_number=get_auto_delivery.account_number,
|
|
customer_town=get_auto_delivery.customer_town,
|
|
customer_state=get_auto_delivery.customer_state,
|
|
customer_address=get_auto_delivery.customer_address,
|
|
customer_zip=get_auto_delivery.customer_zip,
|
|
customer_full_name=get_auto_delivery.customer_full_name,
|
|
oil_prices_id=get_todays_price.id,
|
|
gallons_delivered=gallons_delivered,
|
|
price_per_gallon=get_todays_price.price_for_customer,
|
|
total_amount_customer=total_amount,
|
|
fill_date=date.today(),
|
|
payment_type=payment_type,
|
|
payment_card_id=payment_card_id,
|
|
payment_status=payment_status,
|
|
)
|
|
|
|
session.add(create_new_ticket)
|
|
session.flush()
|
|
|
|
get_auto_delivery.open_ticket_id = create_new_ticket.id
|
|
session.add(get_auto_delivery)
|
|
|
|
session.commit()
|
|
|
|
return ({"ok": True, "auto_ticket_id":create_new_ticket.id}), 200
|
|
|
|
|
|
@router.put("/auto/close_ticket/{ticket_id}")
|
|
async def close_ticket(ticket_id: int):
|
|
"""
|
|
Closes an auto ticket by updating payment_status and unlinking from the auto delivery profile.
|
|
"""
|
|
ticket = session.query(Tickets_Auto_Delivery).filter(Tickets_Auto_Delivery.id == ticket_id).first()
|
|
if ticket:
|
|
ticket.payment_status = 3 # Assuming 3 means closed/paid
|
|
session.add(ticket)
|
|
|
|
delivery = session.query(Auto_Delivery).filter(Auto_Delivery.open_ticket_id == ticket_id).first()
|
|
if delivery:
|
|
delivery.open_ticket_id = None
|
|
session.add(delivery)
|
|
|
|
session.commit()
|
|
return {"ok": True}, 200
|
|
|
|
|
|
@router.delete("/auto/delete_ticket/{ticket_id}")
|
|
async def delete_ticket(ticket_id: int):
|
|
"""
|
|
Deletes an auto ticket and unlinks its reference, used for cleanup (e.g., failed payment).
|
|
"""
|
|
ticket = session.query(Tickets_Auto_Delivery).filter(Tickets_Auto_Delivery.id == ticket_id).first()
|
|
if ticket:
|
|
session.delete(ticket)
|
|
|
|
delivery = session.query(Auto_Delivery).filter(Auto_Delivery.open_ticket_id == ticket_id).first()
|
|
if delivery:
|
|
delivery.open_ticket_id = None
|
|
session.add(delivery)
|
|
|
|
session.commit()
|
|
return {"ok": True}, 200
|
|
|
|
|
|
@router.post("/auto/create/nopreauth/{autoid}")
|
|
async def create_auto_ticket_no_preauth(autoid: int, request: Request):
|
|
"""
|
|
Creates a new delivery ticket without pre-authorization details.
|
|
"""
|
|
request_body = await request.json()
|
|
gallons_delivered = request_body['gallons_delivered']
|
|
gallons_delivered = Decimal(gallons_delivered)
|
|
payment_type = 1
|
|
payment_card_id = 0
|
|
payment_status = 3
|
|
|
|
get_auto_delivery = session.query(Auto_Delivery).filter(Auto_Delivery.id == autoid).first()
|
|
get_todays_price = session.query(Pricing_Oil_Oil).order_by(Pricing_Oil_Oil.id.desc()).first()
|
|
|
|
todays_price = Decimal(get_todays_price.price_for_customer)
|
|
total_amount = gallons_delivered * todays_price
|
|
|
|
create_new_ticket = Tickets_Auto_Delivery(
|
|
customer_id=get_auto_delivery.customer_id,
|
|
account_number=get_auto_delivery.account_number,
|
|
customer_town=get_auto_delivery.customer_town,
|
|
customer_state=get_auto_delivery.customer_state,
|
|
customer_address=get_auto_delivery.customer_address,
|
|
customer_zip=get_auto_delivery.customer_zip,
|
|
customer_full_name=get_auto_delivery.customer_full_name,
|
|
oil_prices_id=get_todays_price.id,
|
|
gallons_delivered=gallons_delivered,
|
|
price_per_gallon=get_todays_price.price_for_customer,
|
|
total_amount_customer=total_amount,
|
|
fill_date=date.today(),
|
|
payment_type=payment_type,
|
|
payment_card_id=payment_card_id,
|
|
payment_status=payment_status,
|
|
)
|
|
|
|
session.add(create_new_ticket)
|
|
session.commit()
|
|
|
|
return ({"ok": True, "auto_ticket_id":create_new_ticket.id}), 200 |