From f88357a7a0728053453b671dcecfdc11d866e186 Mon Sep 17 00:00:00 2001
From: Enrique Barcelli <kiko@barcelli.net>
Date: Tue, 27 Jan 2026 02:49:25 +0800
Subject: [PATCH] Include GST totals in the header

---
 config-sample.py   |  1 +
 extract_invoice.py | 50 ++++++++++++++++++++++++++++++++++++++++++++--
 2 files changed, 49 insertions(+), 2 deletions(-)

diff --git a/config-sample.py b/config-sample.py
index 986218e..ec5a6af 100644
--- a/config-sample.py
+++ b/config-sample.py
@@ -29,3 +29,4 @@ CustomizationID = '<cbc:CustomizationID>urn:peppol:pint:billing-1@sg-1</cbc:Cust
 ProfileID = '<cbc:ProfileID>urn:peppol:pint:billing-1@sg-1</cbc:ProfileID>'
 InvoiceTypeCode = '<cbc:InvoiceTypeCode>380</cbc:InvoiceTypeCode>'
 PeppolID = 'SGUEN200212345Z'
+CountryID = 'SG'
diff --git a/extract_invoice.py b/extract_invoice.py
index 2e63b85..5fcf46e 100644
--- a/extract_invoice.py
+++ b/extract_invoice.py
@@ -32,6 +32,8 @@ def extract_invoice(document):
 
 
 	businessPartner=invoice['response']['data'][0]['businessPartner']
+	summedLineAmount=invoice['response']['data'][0]['summedLineAmount']
+	totalTaxAmount=invoice['response']['data'][0]['grandTotalAmount']-summedLineAmount
 
 
 	url = ob_api_url+"InvoiceLine?_where=invoice='"+invoice['response']['data'][0]['id']+"'&_noActiveFilter=false&_sortBy=lineNo"
@@ -72,10 +74,54 @@ def extract_invoice(document):
 	ubloutput = ubloutput+indentation+'<cac:AccountingSupplierParty>'+'\n'
 	ubloutput = ubloutput+indentation+indentation+'<cac:Party>'+'\n'
 	ubloutput = ubloutput+indentation+indentation+indentation+'<cbc:EndpointID schemeID="0195">'+PeppolID+'</cbc:EndpointID>'+'\n'
-	
-	
+	ubloutput = ubloutput+indentation+indentation+indentation+'<cac:PartyIdentification>'+'\n'
+	ubloutput = ubloutput+indentation+indentation+indentation+indentation+'<cbc:ID schemeID="0035">5790000436071</cbc:ID>'+'\n'
+	ubloutput = ubloutput+indentation+indentation+indentation+'</cac:PartyIdentification>'+'\n'
+	ubloutput = ubloutput+indentation+indentation+indentation+'<cac:PostalAddress>'+'\n'
+	ubloutput = ubloutput+indentation+indentation+indentation+indentation+'<cac:Country>'+'\n'
+	ubloutput = ubloutput+indentation+indentation+indentation+indentation+indentation+'<cbc:IdentificationCode>'+CountryID+'</cbc:IdentificationCode>'+'\n'
+	ubloutput = ubloutput+indentation+indentation+indentation+indentation+'</cac:Country>'+'\n'
+	ubloutput = ubloutput+indentation+indentation+indentation+'</cac:PostalAddress>'+'\n'
+	ubloutput = ubloutput+indentation+indentation+indentation+'<cac:PartyLegalEntity>'+'\n'
+	ubloutput = ubloutput+indentation+indentation+indentation+indentation+'<cbc:RegistrationName>'+invoice['response']['data'][0]['client$_identifier']+'</cbc:RegistrationName>'+'\n'
+	ubloutput = ubloutput+indentation+indentation+indentation+'</cac:PartyLegalEntity>'+'\n'
 	ubloutput = ubloutput+indentation+indentation+'</cac:Party>'+'\n'
 	ubloutput = ubloutput+indentation+'</cac:AccountingSupplierParty>'+'\n'
+	ubloutput = ubloutput+indentation+'<cac:AccountingCustomerParty>'+'\n'
+	ubloutput = ubloutput+indentation+indentation+'<cac:Party>'+'\n'
+	ubloutput = ubloutput+indentation+indentation+indentation+'<cbc:EndpointID schemeID="0195">'+'XXXXXXXXXXXXXXXXX'+'</cbc:EndpointID>'+'\n'
+	ubloutput = ubloutput+indentation+indentation+indentation+'<cac:PartyIdentification>'+'\n'
+	ubloutput = ubloutput+indentation+indentation+indentation+indentation+'<cbc:ID schemeID="0035">5790000436071</cbc:ID>'+'\n'
+	ubloutput = ubloutput+indentation+indentation+indentation+'</cac:PartyIdentification>'+'\n'
+	ubloutput = ubloutput+indentation+indentation+indentation+'<cac:PostalAddress>'+'\n'
+	ubloutput = ubloutput+indentation+indentation+indentation+indentation+'<cac:Country>'+'\n'
+	ubloutput = ubloutput+indentation+indentation+indentation+indentation+indentation+'<cbc:IdentificationCode>'+CountryID+'</cbc:IdentificationCode>'+'\n'
+	ubloutput = ubloutput+indentation+indentation+indentation+indentation+'</cac:Country>'+'\n'
+	ubloutput = ubloutput+indentation+indentation+indentation+'</cac:PostalAddress>'+'\n'
+	ubloutput = ubloutput+indentation+indentation+indentation+'<cac:PartyLegalEntity>'+'\n'
+	ubloutput = ubloutput+indentation+indentation+indentation+indentation+'<cbc:RegistrationName>'+invoice['response']['data'][0]['businessPartner$_identifier']+'</cbc:RegistrationName>'+'\n'
+	ubloutput = ubloutput+indentation+indentation+indentation+'</cac:PartyLegalEntity>'+'\n'
+	ubloutput = ubloutput+indentation+indentation+'</cac:Party>'+'\n'
+	ubloutput = ubloutput+indentation+'</cac:AccountingCustomerParty>'+'\n'
+	ubloutput = ubloutput+indentation+'<cac:TaxTotal>'+'\n'
+	ubloutput = ubloutput+indentation+indentation+'<cbc:TaxAmount currencyID="SGD">'+f"{totalTaxAmount:.2f}"+'</cbc:TaxAmount>'+'\n'
+	ubloutput = ubloutput+indentation+indentation+'<cac:TaxSubtotal>'+'\n'
+	ubloutput = ubloutput+indentation+indentation+indentation+'<cbc:TaxableAmount currencyID="SGD">'+f"{summedLineAmount:.2f}"+'</cbc:TaxableAmount>'+'\n'
+	ubloutput = ubloutput+indentation+indentation+indentation+'<cbc:TaxAmount currencyID="SGD">'+f"{totalTaxAmount:.2f}"+'</cbc:TaxAmount>'+'\n'
+	ubloutput = ubloutput+indentation+indentation+indentation+'<cac:TaxCategory>'+'\n'
+	ubloutput = ubloutput+indentation+indentation+indentation+indentation+'<cbc:ID>SR</cbc:ID>'+'\n'
+	ubloutput = ubloutput+indentation+indentation+indentation+indentation+'<cbc:Percent>'+gst_rate+'</cbc:Percent>'+'\n'
+	ubloutput = ubloutput+indentation+indentation+indentation+indentation+'<cac:TaxScheme>'+'\n'
+	ubloutput = ubloutput+indentation+indentation+indentation+indentation+indentation+'<cbc:ID>GST</cbc:ID>'+'\n'
+	ubloutput = ubloutput+indentation+indentation+indentation+indentation+'</cac:TaxScheme>'+'\n'
+	ubloutput = ubloutput+indentation+indentation+indentation+'</cac:TaxCategory>'+'\n'
+	ubloutput = ubloutput+indentation+indentation+'</cac:TaxSubtotal>'+'\n'
+	ubloutput = ubloutput+indentation+'</cac:TaxTotal>'+'\n'
+	
+	
+	
+	
+	
 	
 	print(ubloutput)