Updated claude big changes

app/promo/views.py

@@ -1,9 +1,10 @@
 import logging
-from flask import request, jsonify
+from flask import request
 import decimal
 from datetime import datetime
 from app.promo import promo
 from app import db
+from app.common.responses import success_response
 from app.classes.promo import (
     Promo_Promo,
     Promo_Promo_schema)
@@ -32,7 +33,7 @@ def get_promo(promo_id):
         .filter(Promo_Promo.id == promo_id)
         .first())
     query_schema = Promo_Promo_schema(many=False)
-    return jsonify(query_schema.dump(get_promo_data))
+    return success_response({"promo": query_schema.dump(get_promo_data)})
 
 
 @promo.route("/promoprice/<int:delivery_id>", methods=["GET"])
@@ -47,10 +48,7 @@ def get_promo_price(delivery_id):
 
     price = get_delivery.customer_price - get_delivery.promo_money_discount
     price = convert_to_decimal(price)
-    return jsonify({
-        "ok": True,
-        "price": price,
-    }), 200
+    return success_response({"price": price})
 
 
 
@@ -63,7 +61,7 @@ def get_all_promo():
         .query(Promo_Promo)
         .all())
     query_schema = Promo_Promo_schema(many=True)
-    return jsonify(query_schema.dump(get_promo_data))
+    return success_response({"promos": query_schema.dump(get_promo_data)})
 
 
 @promo.route("/delete/<int:promo_id>", methods=["DELETE"])
@@ -79,10 +77,7 @@ def delete_a_promo(promo_id):
     db.session.delete(get_promo_data)
     db.session.commit()
 
-
-    return jsonify({
-        "ok": True,
-    }), 200
+    return success_response()
 
 @promo.route("/create", methods=["POST"])
 def create_promo():
@@ -108,11 +103,7 @@ def create_promo():
     db.session.add(new_promo)
     db.session.commit()
 
-
-    return jsonify({
-        "ok": True,
-        'promo_id':new_promo.id,
-    }), 200
+    return success_response({'promo_id': new_promo.id})
 
 
 @promo.route("/edit/<int:promo_id>", methods=["PUT"])
@@ -141,11 +132,7 @@ def edit_promo(promo_id):
     db.session.add(get_promo_data)
     db.session.commit()
 
-    return jsonify({
-        "ok": True,
-        'promo_id':get_promo_data.id,
-    }), 200
-
+    return success_response({'promo_id': get_promo_data.id})
 
 
 @promo.route("/on/<int:promo_id>", methods=["PATCH"])
@@ -161,14 +148,10 @@ def turn_on_promo(promo_id):
 
     get_promo_data.active = True
 
-
     db.session.add(get_promo_data)
     db.session.commit()
 
-    return jsonify({
-        "ok": True,
-        'promo_id':get_promo_data.id,
-    }), 200
+    return success_response({'promo_id': get_promo_data.id})
 
 @promo.route("/off/<int:promo_id>", methods=["PATCH"])
 def turn_off_promo(promo_id):
@@ -183,11 +166,7 @@ def turn_off_promo(promo_id):
 
     get_promo_data.active = False
 
-
     db.session.add(get_promo_data)
     db.session.commit()
 
-    return jsonify({
-        "ok": True,
-        'promo_id':get_promo_data.id,
-    }), 200
+    return success_response({'promo_id': get_promo_data.id})