Updated claude big changes

app/delivery_data/views.py

@@ -1,9 +1,10 @@
 import logging
-from flask import request, jsonify
+from flask import request
 from datetime import datetime
 from decimal import Decimal
 from app.delivery_data import delivery_data
 from app import db
+from app.common.responses import success_response
 from app.classes.customer import Customer_Customer, Customer_Description
 from app.classes.delivery import Delivery_Delivery
 from app.classes.employee import Employee_Employee
@@ -54,12 +55,9 @@ def office_finalize_delivery(delivery_id):
             .filter(Customer_Description.customer_id == get_delivery.customer_id) \
             .first()
    
-    #TODO hardcode for now
-    delivery_driver_id = 2
-   
     get_driver = (db.session
                     .query(Employee_Employee)
-                    .filter(Employee_Employee.id == delivery_driver_id)
+                    .filter(Employee_Employee.id == get_delivery.driver_employee_id)
                     .first())
     
     get_stats_employee = (db.session
@@ -156,9 +154,8 @@ def office_finalize_delivery(delivery_id):
     db.session.commit()
 
 
-    return jsonify({
-        "ok": True,
+    return success_response({
         'delivery': {
             'id': get_delivery.id,
         },
-    }), 200
+    })