Update order.py

order.py

@@ -1,14 +1,10 @@
-from flask import Blueprint, render_template, request, session, jsonify, redirect, url_for, send_file
+from flask import Blueprint, render_template, request, session, jsonify  # Added jsonify import
 from salesforce import get_salesforce_connection
-from io import BytesIO
-from reportlab.lib.pagesizes import letter
-from reportlab.pdfgen import canvas
 
 order_blueprint = Blueprint('order', __name__)
 
 # Initialize Salesforce connection
 sf = get_salesforce_connection()
-
 @order_blueprint.route("/order", methods=["GET"])
 def order_summary():
     email = session.get('user_email')  # Fetch logged-in user's email
@@ -34,50 +30,8 @@ def order_summary():
         print(f"Error fetching order details: {str(e)}")
         return render_template("order.html", order=None, error=str(e))
 
-# New route to generate the invoice
-@order_blueprint.route("/generate_invoice/<order_id>", methods=["GET"])
-def generate_invoice(order_id):
-    email = session.get('user_email')  # Fetch logged-in user's email
-    if not email:
-        return jsonify({"success": False, "message": "User not logged in"}), 400
-
-    try:
-        # Fetch order details from Salesforce
-        result = sf.query(f"""
-            SELECT Id, Customer_Name__c, Customer_Email__c, Total_Amount__c, Order_Details__c, Order_Status__c, Discount__c, Total_Bill__c
-            FROM Order__c
-            WHERE Id = '{order_id}'
-        """)
-        order = result.get("records", [])[0] if result.get("records") else None
 
-        if not order:
-            return jsonify({"success": False, "message": "Order not found"}), 404
 
-        # Create PDF invoice
-        buffer = BytesIO()
-        c = canvas.Canvas(buffer, pagesize=letter)
-        c.drawString(100, 750, f"Invoice for Order ID: {order_id}")
-        c.drawString(100, 730, f"Customer Name: {order['Customer_Name__c']}")
-        c.drawString(100, 710, f"Customer Email: {order['Customer_Email__c']}")
-        c.drawString(100, 690, f"Total Amount: ${order['Total_Amount__c']}")
-        c.drawString(100, 670, f"Discount: ${order['Discount__c']}")
-        c.drawString(100, 650, f"Total Bill: ${order['Total_Bill__c']}")
-        
-        # Add order details (split by new lines)
-        order_details = order.get("Order_Details__c", "").split("\n")
-        y_position = 630
-        for line in order_details:
-            c.drawString(100, y_position, line)
-            y_position -= 20
 
-        # Finalize the PDF
-        c.showPage()
-        c.save()
 
-        # Send the PDF as a downloadable file
-        buffer.seek(0)
-        return send_file(buffer, as_attachment=True, download_name=f"invoice_{order_id}.pdf", mimetype="application/pdf")
-    
-    except Exception as e:
-        print(f"Error generating invoice: {str(e)}")
-        return jsonify({"success": False, "message": str(e)}), 500
+