Update pdf_generator.py

pdf_generator.py

@@ -1,85 +1,83 @@
 from reportlab.lib.pagesizes import A5
 from reportlab.lib import colors
-from reportlab.platypus import SimpleDocTemplate, Table, TableStyle, Paragraph, Spacer
 from reportlab.lib.styles import getSampleStyleSheet
+from reportlab.platypus import SimpleDocTemplate, Table, TableStyle, Paragraph, Spacer
+from reportlab.lib.units import mm
+
+def create_invoice(text, file_name):
+    # Nettoyage du texte source
+    text = text.replace('*', '').replace('#', '')
+    text = text.replace('Reçu', '').strip()
+
+    # Extraction des champs
+    lines = [l.strip() for l in text.split("\n") if l.strip()]
+    data_dict = {}
+
+    for line in lines:
+        if ":" in line:
+            key, val = line.split(":", 1)
+            data_dict[key.strip()] = val.strip()
+
+    # Récupération des informations
+    nom = data_dict.get("Nom", "")
+    adresse = data_dict.get("Adresse", "")
+    commande = data_dict.get("Commande", "")
+    prix_piece = data_dict.get("Prix par pièce", "")
+    prix_total = data_dict.get("Prix total", "")
+    date_livraison = data_dict.get("Date de livraison", "")
+
+    # Structure du PDF
+    doc = SimpleDocTemplate(
+        file_name,
+        pagesize=A5,
+        rightMargin=20,
+        leftMargin=20,
+        topMargin=20,
+        bottomMargin=20
+    )
 
-def create_receipt(text, file_name):
-    # Nettoyage basique
-    clean = text.replace('*', '').replace('#', '').replace('Reçu', '')
-
-    # Extraction des champs simples
-    lines = [l.strip() for l in clean.split("\n") if l.strip()]
-    data_map = {}
-    description_lines = []
-
-    for l in lines:
-        if ":" in l:
-            key, val = l.split(":", 1)
-            data_map[key.strip()] = val.strip()
-        else:
-            description_lines.append(l)
-
-    # Données extraites
-    client = data_map.get("Client", "")
-    telephone = data_map.get("Téléphone", "")
-    adresse = data_map.get("Adresse", "")
-    date_commande = data_map.get("Date de commande", "")
-    delai = data_map.get("Délai de livraison estimé", "")
-
-    # Description brute (ex: "3 iPhone (1 rouge, 2 blancs)")
-    description = " ".join(description_lines)
-
-    # Styles
     styles = getSampleStyleSheet()
-    title_style = styles["Title"]
-    normal = styles["Normal"]
-
-    # Création du document
-    doc = SimpleDocTemplate(file_name, pagesize=A5,
-                            leftMargin=25, rightMargin=25, topMargin=25, bottomMargin=25)
-
     elements = []
 
     # Titre
-    elements.append(Paragraph("FACTURE", title_style))
-    elements.append(Spacer(1, 12))
+    titre = Paragraph("<b>FACTURE</b>", styles["Title"])
+    elements.append(titre)
+    elements.append(Spacer(1, 8 * mm))
 
     # Informations client
-    info_block = (
-        f"<b>Client :</b> {client}<br/>"
-        f"<b>Téléphone :</b> {telephone}<br/>"
+    info_client = (
+        f"<b>Client :</b> {nom}<br/>"
         f"<b>Adresse :</b> {adresse}<br/>"
-        f"<b>Date de commande :</b> {date_commande}<br/>"
-        f"<b>Délai de livraison estimé :</b> {delai}<br/>"
+        f"<b>Date de livraison :</b> {date_livraison}<br/>"
     )
-    elements.append(Paragraph(info_block, normal))
-    elements.append(Spacer(1, 18))
+    elements.append(Paragraph(info_client, styles["Normal"]))
+    elements.append(Spacer(1, 8 * mm))
 
-    # Tableau détaillé
+    # Tableau du détail de commande
     table_data = [
-        ["Description", "Quantité"],
-        [description, description.split()[0] if description.split()[0].isdigit() else ""]
+        ["Description", "Qté", "Prix unitaire", "Total"],
+        [commande, "", prix_piece, prix_total]
     ]
 
-    table = Table(table_data, colWidths=[200, 60])
-
-    table.setStyle(
-        TableStyle([
-            ("BACKGROUND", (0, 0), (-1, 0), colors.lightgrey),
-            ("TEXTCOLOR", (0, 0), (-1, 0), colors.black),
-            ("ALIGN", (1, 1), (-1, -1), "CENTER"),
-            ("VALIGN", (0, 0), (-1, -1), "MIDDLE"),
-            ("BOX", (0, 0), (-1, -1), 1, colors.black),
-            ("GRID", (0, 0), (-1, -1), 0.5, colors.grey),
-            ("FONTSIZE", (0, 0), (-1, -1), 10),
-        ])
-    )
+    table = Table(table_data, colWidths=[60*mm, 15*mm, 30*mm, 30*mm])
+    table.setStyle(TableStyle([
+        ("BACKGROUND", (0,0), (-1,0), colors.lightgrey),
+        ("TEXTCOLOR", (0,0), (-1,0), colors.black),
+        ("ALIGN", (1,1), (-1,-1), "CENTER"),
+        ("FONTNAME", (0,0), (-1,0), "Helvetica-Bold"),
+        ("BOTTOMPADDING", (0,0), (-1,0), 8),
+        ("GRID", (0,0), (-1,-1), 0.5, colors.black),
+    ]))
 
     elements.append(table)
-    elements.append(Spacer(1, 24))
+    elements.append(Spacer(1, 12 * mm))
 
-    # Mention finale
-    elements.append(Paragraph("Merci pour votre commande.", normal))
+    # Total général
+    total_paragraph = Paragraph(
+        f"<b>Total dû : {prix_total}</b>",
+        styles["Heading3"]
+    )
+    elements.append(total_paragraph)
 
-    # Génération du PDF
+    # Génération PDF
     doc.build(elements)