254 lines
No EOL
10 KiB
Python
254 lines
No EOL
10 KiB
Python
import json
|
|
from fpdf import FPDF
|
|
import os
|
|
|
|
class Generator:
|
|
def __init__(self, form_data, template_path="Templates/proposition_commercial.json"):
|
|
self.form_data = form_data
|
|
self.template_path = template_path
|
|
|
|
def load_template(self):
|
|
with open(self.template_path, 'r', encoding='utf-8') as file:
|
|
template = json.load(file)
|
|
return template
|
|
|
|
def replace_variables(self, content, visited=None):
|
|
if visited is None:
|
|
visited = set()
|
|
|
|
# Avoid infinite recursion by tracking visited objects
|
|
if id(content) in visited:
|
|
return content
|
|
visited.add(id(content))
|
|
|
|
if isinstance(content, str):
|
|
# Replace variables in a string
|
|
for key, value in self.form_data.items():
|
|
if key == "client" and isinstance(value, dict):
|
|
# Handle client dictionary directly
|
|
for sub_key, sub_value in value.items():
|
|
placeholder = f"{{{sub_key}}}"
|
|
content = content.replace(placeholder, str(sub_value))
|
|
elif isinstance(value, dict):
|
|
# Flatten nested dictionaries like 'client'
|
|
for sub_key, sub_value in value.items():
|
|
placeholder = f"{{{key}_{sub_key}}}"
|
|
content = content.replace(placeholder, str(sub_value))
|
|
elif isinstance(value, list):
|
|
# Handle lists (e.g., features or services)
|
|
if key == "features":
|
|
formatted_features = "\n".join(f"- {item['description']}" for item in value if isinstance(item, dict) and 'description' in item)
|
|
content = content.replace(f"{{{key}}}", formatted_features)
|
|
elif key == "services":
|
|
formatted_services = "\n".join(
|
|
f"- {item['description']} : {item['prix_unitaire']} euros"
|
|
for item in value if isinstance(item, dict) and 'description' in item and 'prix_unitaire' in item
|
|
)
|
|
content = content.replace(f"{{{key}}}", formatted_services)
|
|
else:
|
|
content = content.replace(f"{{{key}}}", str(value))
|
|
return content
|
|
elif isinstance(content, list):
|
|
# Replace variables in each item of a list
|
|
return [self.replace_variables(item, visited) for item in content]
|
|
elif isinstance(content, dict):
|
|
# Replace variables in a dictionary
|
|
return {k: self.replace_variables(v, visited) for k, v in content.items()}
|
|
return content
|
|
|
|
def generate_content(self):
|
|
template = self.load_template()
|
|
return self.replace_variables(template)
|
|
|
|
def generate_pdf(self, output_path=None):
|
|
content = self.generate_content()
|
|
|
|
# Format the output file name based on the client name
|
|
client_name = self.form_data.get("client_name", "client").replace(" ", "_").lower()
|
|
if output_path is None:
|
|
output_path = f"output/{client_name}_proposition_commercial.pdf"
|
|
else:
|
|
output_path = f"output/{output_path}/{client_name}_proposition_commercial.pdf"
|
|
|
|
pdf = FPDF()
|
|
pdf.add_page()
|
|
|
|
# Ajouter un fond (image de fond)
|
|
background_path = "Data/BG.png" # Chemin de l'image de fond
|
|
if os.path.exists(background_path):
|
|
pdf.image(background_path, x=0, y=0, w=210, h=297) # Dimensions pour une page A4
|
|
|
|
# Title
|
|
title = content.get("title", "Proposition Commerciale")
|
|
pdf.set_font("Helvetica", "B", 16)
|
|
pdf.cell(0, 10, title, ln=True, align="C")
|
|
pdf.ln(10)
|
|
|
|
# Main content
|
|
pdf.set_font("Helvetica", "", 12)
|
|
for section in content.get("sections", []):
|
|
# Section header
|
|
pdf.set_font("Helvetica", "B", 14)
|
|
pdf.cell(0, 10, section.get("header", ""), ln=True)
|
|
pdf.ln(5)
|
|
|
|
# Section content
|
|
pdf.set_font("Helvetica", "", 12)
|
|
section_content = section.get("content", [])
|
|
if isinstance(section_content, list):
|
|
for item in section_content:
|
|
pdf.multi_cell(0, 10, str(item))
|
|
pdf.ln(2)
|
|
elif isinstance(section_content, str):
|
|
pdf.multi_cell(0, 10, section_content)
|
|
pdf.ln(5)
|
|
|
|
# Additional fields outside sections
|
|
for key, value in content.items():
|
|
if key not in ["sections", "title"]: # Exclure "sections" et "title"
|
|
pdf.set_font("Helvetica", "B", 12)
|
|
pdf.cell(0, 10, f"{key} :", ln=True)
|
|
pdf.set_font("Helvetica", "", 12)
|
|
|
|
# Format the client data specially
|
|
if key == "client" and isinstance(value, dict):
|
|
client_str = "\n".join(f"{k.capitalize()}: {v}" for k, v in value.items())
|
|
pdf.multi_cell(0, 10, client_str)
|
|
else:
|
|
pdf.multi_cell(0, 10, str(value))
|
|
|
|
pdf.ln(5)
|
|
|
|
# Ensure output folder exists
|
|
os.makedirs(os.path.dirname(output_path), exist_ok=True)
|
|
|
|
pdf.output(output_path)
|
|
print(f"PDF saved to {output_path}")
|
|
|
|
def generate_facture(self, output_path=None):
|
|
content = self.generate_content()
|
|
|
|
# Format the output file name based on the client name
|
|
# On récupère le nom du client dans le dictionnaire 'client'
|
|
client = content.get("client", {})
|
|
if isinstance(client, dict):
|
|
client_name = client.get("nom", "client").replace(" ", "_").lower()
|
|
else:
|
|
# Fallback to a default name if 'client' is not a dictionary
|
|
client_name = "client"
|
|
print(f"Client name: {client_name}")
|
|
|
|
if output_path is None:
|
|
output_path = f"output/{client_name}_devis.pdf"
|
|
else:
|
|
output_path = f"output/{output_path}/{client_name}_devis.pdf"
|
|
|
|
# Create PDF with invoice-like styling
|
|
pdf = FPDF()
|
|
pdf.add_page()
|
|
|
|
# Add background if exists
|
|
background_path = "Data/BG.png"
|
|
if os.path.exists(background_path):
|
|
pdf.image(background_path, x=0, y=0, w=210, h=297)
|
|
|
|
# Document Title
|
|
pdf.set_font("Helvetica", "B", 18)
|
|
pdf.cell(0, 15, content.get("title", "Devis de service"), ln=True, align="C")
|
|
|
|
# Add a horizontal line
|
|
pdf.line(10, pdf.get_y(), 200, pdf.get_y())
|
|
pdf.ln(10)
|
|
|
|
# Header Info (Devis number and date)
|
|
pdf.set_font("Helvetica", "B", 10)
|
|
pdf.cell(95, 10, f"DEVIS N° : {content.get('numero', '')}", 0, 0)
|
|
pdf.cell(95, 10, f"DATE : {content.get('date', '')}", 0, 1, 'R')
|
|
pdf.ln(5)
|
|
|
|
# Client Information Box
|
|
pdf.set_fill_color(240, 240, 240) # Light gray background
|
|
pdf.rect(10, pdf.get_y(), 190, 30, style='F')
|
|
|
|
pdf.set_font("Helvetica", "B", 12)
|
|
pdf.cell(0, 10, "INFORMATIONS CLIENT", 0, 1, 'L')
|
|
|
|
pdf.set_font("Helvetica", "", 10)
|
|
client_data = content.get("client", {})
|
|
if isinstance(client_data, dict):
|
|
pdf.cell(40, 5, f"Nom:", 0, 0)
|
|
pdf.cell(150, 5, f"{client_data.get('nom', '')}", 0, 1)
|
|
|
|
pdf.cell(40, 5, f"Email:", 0, 0)
|
|
pdf.cell(150, 5, f"{client_data.get('email', '')}", 0, 1)
|
|
|
|
pdf.cell(40, 5, f"Téléphone:", 0, 0)
|
|
pdf.cell(150, 5, f"{client_data.get('telephone', '')}", 0, 1)
|
|
|
|
pdf.cell(40, 5, f"Adresse:", 0, 0)
|
|
pdf.cell(150, 5, f"{client_data.get('adresse', '')}", 0, 1)
|
|
|
|
pdf.ln(10)
|
|
|
|
# Services Table Header
|
|
pdf.set_fill_color(60, 60, 60) # Dark gray for header
|
|
pdf.set_text_color(255, 255, 255) # White text
|
|
pdf.set_font("Helvetica", "B", 10)
|
|
|
|
# Table header
|
|
pdf.cell(120, 10, "DESCRIPTION", 1, 0, 'C', 1)
|
|
pdf.cell(35, 10, "PRIX UNITAIRE", 1, 0, 'C', 1)
|
|
pdf.cell(35, 10, "MONTANT", 1, 1, 'C', 1)
|
|
|
|
# Reset color for table content
|
|
pdf.set_text_color(0, 0, 0)
|
|
pdf.set_font("Helvetica", "", 10)
|
|
|
|
# Services Table Content
|
|
services = self.form_data.get("services", [])
|
|
total = 0
|
|
|
|
if isinstance(services, list):
|
|
for service in services:
|
|
if isinstance(service, dict):
|
|
description = service.get("description", "")
|
|
prix = service.get("prix_unitaire", 0)
|
|
total += float(prix)
|
|
|
|
pdf.cell(120, 8, description, 1, 0)
|
|
pdf.cell(35, 8, f"{prix} E", 1, 0, 'R')
|
|
pdf.cell(35, 8, f"{prix} E", 1, 1, 'R')
|
|
|
|
# Total Amount
|
|
pdf.set_font("Helvetica", "B", 10)
|
|
pdf.cell(155, 10, "TOTAL", 1, 0, 'R')
|
|
pdf.cell(35, 10, f"{total} E", 1, 1, 'R')
|
|
|
|
# Payment Terms and Notes
|
|
pdf.ln(10)
|
|
pdf.set_font("Helvetica", "B", 10)
|
|
pdf.cell(0, 10, "CONDITIONS DE PAIEMENT", 0, 1)
|
|
|
|
pdf.set_font("Helvetica", "", 10)
|
|
payment_terms = self.form_data.get("payment_terms", "50% à la signature, 50% à la livraison")
|
|
pdf.multi_cell(0, 5, payment_terms)
|
|
|
|
pdf.ln(5)
|
|
pdf.set_font("Helvetica", "B", 10)
|
|
pdf.cell(0, 10, "NOTES", 0, 1)
|
|
|
|
pdf.set_font("Helvetica", "", 10)
|
|
pdf.multi_cell(0, 5, "Ce devis est valable 30 jours à compter de la date d'émission.")
|
|
|
|
# Footer
|
|
pdf.ln(10)
|
|
pdf.set_font("Helvetica", "I", 8)
|
|
pdf.cell(0, 10, "Merci de votre confiance.", 0, 1, 'C')
|
|
pdf.cell(0, 10, "Pour accepter ce devis, veuillez le signer et me le retourner par email.", 0, 1, 'C')
|
|
|
|
# Ensure output folder exists
|
|
os.makedirs(os.path.dirname(output_path), exist_ok=True)
|
|
|
|
# Save the PDF
|
|
pdf.output(output_path)
|
|
print(f"PDF saved to {output_path}") |