Update orderhistory.py

orderhistory.py

@@ -1,7 +1,9 @@
-from flask import Blueprint, render_template, request,redirect, session, jsonify  # Added jsonify import
+from flask import Blueprint, render_template, request, redirect, session, jsonify, send_file, url_for
 from salesforce import get_salesforce_connection
 from datetime import datetime
 import pytz  # Library to handle timezone conversions
+from weasyprint import HTML
+from io import BytesIO
 
 orderhistory_blueprint = Blueprint('orderhistory', __name__)
 
@@ -66,9 +68,53 @@ def order_history():
             order_status = order.get("Order_Status__c", "N/A")  # Default to "N/A" if no status
             order['order_status'] = order_status
 
-
         return render_template("order_history.html", orders=orders)
 
     except Exception as e:
         print(f"Error fetching order history: {str(e)}")
         return render_template("order_history.html", orders=[], error=str(e))
+
+
+@orderhistory_blueprint.route("/generate_invoice/<order_id>", methods=["GET"])
+def generate_invoice(order_id):
+    email = session.get('user_email')  # Get 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, CreatedDate
+            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
+
+        # Format order details for better readability
+        order_details = order.get("Order_Details__c", "")
+        items = order_details.split("\n")
+        formatted_items = []
+
+        for item in items:
+            item_details = item.split(" | ")
+            if len(item_details) > 1:
+                name = item_details[0].strip()
+                quantity = item_details[1].strip()
+                formatted_items.append(f"{name} * {quantity}")
+
+        # Render HTML invoice template
+        rendered_html = render_template('invoice.html', order=order, formatted_items=formatted_items)
+
+        # Convert HTML to PDF using WeasyPrint
+        pdf = HTML(string=rendered_html).write_pdf()
+
+        # Send the PDF as a downloadable file
+        return send_file(BytesIO(pdf), 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