Add invoice generator for fuel station transactions

- Read CSV files from input/ directory
- Generate static HTML invoices grouped by customer and card
- Filter transactions to only include fleet customers
- Compact print-friendly layout with 2 decimal precision
This commit is contained in:
2026-03-23 10:36:14 +01:00
parent d3b8828260
commit 650435e00c
6 changed files with 609 additions and 0 deletions

10
Cargo.toml Normal file
View File

@@ -0,0 +1,10 @@
[package]
name = "invoice-generator"
version = "0.1.0"
edition = "2021"
[dependencies]
askama = "0.15.5"
chrono = "0.4.44"
csv = "1.4.0"
serde = "1.0.228"

126
src/invoice_generator.rs Normal file
View File

@@ -0,0 +1,126 @@
use chrono::NaiveDateTime;
use csv::ReaderBuilder;
use std::collections::BTreeMap;
use std::fs;
use std::path::Path;
#[derive(Debug, Clone)]
pub struct Transaction {
pub date: NaiveDateTime,
pub batch_number: String,
pub amount: f64,
pub volume: f64,
pub price: f64,
pub quality: i32,
pub quality_name: String,
pub card_number: String,
pub card_type: String,
pub customer_number: String,
pub station: String,
pub terminal: String,
pub pump: String,
pub receipt: String,
pub card_report_group_number: String,
pub control_number: String,
}
#[derive(Debug, Clone)]
pub struct Batch {
pub filename: String,
pub transactions: Vec<Transaction>,
}
#[derive(Debug, Clone)]
pub struct Customer {
pub customer_number: String,
pub cards: BTreeMap<String, Vec<Transaction>>,
}
fn get_field(record: &csv::StringRecord, index: usize) -> &str {
record.get(index).unwrap_or("")
}
impl Transaction {
pub fn from_record(record: &csv::StringRecord) -> Option<Self> {
let date_str = get_field(record, 0);
let date = NaiveDateTime::parse_from_str(date_str, "%Y-%m-%d %H:%M:%S").ok()?;
Some(Transaction {
date,
batch_number: get_field(record, 1).to_string(),
amount: get_field(record, 2).parse().unwrap_or(0.0),
volume: get_field(record, 3).parse().unwrap_or(0.0),
price: get_field(record, 4).parse().unwrap_or(0.0),
quality: get_field(record, 5).parse().unwrap_or(0),
quality_name: get_field(record, 6).to_string(),
card_number: get_field(record, 7).to_string(),
card_type: get_field(record, 8).to_string(),
customer_number: get_field(record, 9).to_string(),
station: get_field(record, 10).to_string(),
terminal: get_field(record, 11).to_string(),
pump: get_field(record, 12).to_string(),
receipt: get_field(record, 13).to_string(),
card_report_group_number: get_field(record, 14).to_string(),
control_number: get_field(record, 15).to_string(),
})
}
}
pub fn read_csv_file(path: &Path) -> Result<Batch, Box<dyn std::error::Error>> {
let filename = path
.file_name()
.and_then(|n| n.to_str())
.unwrap_or("unknown")
.to_string();
let file = fs::File::open(path)?;
let mut rdr = ReaderBuilder::new()
.delimiter(b'\t')
.has_headers(true)
.flexible(true)
.from_reader(file);
let mut transactions = Vec::new();
for result in rdr.records() {
let record = result?;
if let Some(tx) = Transaction::from_record(&record) {
if tx.amount > 0.0 && !tx.customer_number.is_empty() {
transactions.push(tx);
}
}
}
transactions.sort_by(|a, b| a.date.cmp(&b.date));
Ok(Batch {
filename,
transactions,
})
}
pub fn group_by_customer(batches: &[Batch]) -> BTreeMap<String, Customer> {
let mut customers: BTreeMap<String, Customer> = BTreeMap::new();
for batch in batches {
for tx in &batch.transactions {
let customer = customers
.entry(tx.customer_number.clone())
.or_insert_with(|| Customer {
customer_number: tx.customer_number.clone(),
cards: BTreeMap::new(),
});
let card_txs = customer.cards.entry(tx.card_number.clone()).or_default();
card_txs.push(tx.clone());
}
}
for customer in customers.values_mut() {
for card_txs in customer.cards.values_mut() {
card_txs.sort_by(|a, b| a.date.cmp(&b.date));
}
}
customers
}

175
src/main.rs Normal file
View File

@@ -0,0 +1,175 @@
use askama::Template;
use chrono::Utc;
use std::fs;
use std::path::Path;
mod invoice_generator;
use invoice_generator::{group_by_customer, read_csv_file, Batch, Customer};
fn fmt(v: f64) -> String {
format!("{:.2}", v)
}
#[derive(Clone)]
struct CardData {
card_number: String,
transactions: Vec<FormattedTransaction>,
total_amount: String,
total_volume: String,
}
#[derive(Clone)]
struct FormattedTransaction {
date: String,
quality_name: String,
price: String,
volume: String,
amount: String,
receipt: String,
}
#[derive(Clone)]
struct PreparedCustomer {
customer_number: String,
cards: Vec<CardData>,
grand_total: String,
}
impl PreparedCustomer {
fn from_customer(customer: Customer) -> Self {
let cards: Vec<CardData> = customer
.cards
.into_iter()
.map(|(card_number, transactions)| {
let formatted_txs: Vec<FormattedTransaction> = transactions
.into_iter()
.map(|t| FormattedTransaction {
date: t.date.format("%Y-%m-%d %H:%M").to_string(),
quality_name: t.quality_name,
price: fmt(t.price),
volume: fmt(t.volume),
amount: fmt(t.amount),
receipt: t.receipt,
})
.collect();
let total_amount: f64 = formatted_txs
.iter()
.map(|t| t.amount.parse::<f64>().unwrap())
.sum();
let total_volume: f64 = formatted_txs
.iter()
.map(|t| t.volume.parse::<f64>().unwrap())
.sum();
CardData {
card_number,
transactions: formatted_txs,
total_amount: fmt(total_amount),
total_volume: fmt(total_volume),
}
})
.collect();
let grand_total: f64 = cards
.iter()
.map(|c| c.total_amount.parse::<f64>().unwrap())
.sum();
PreparedCustomer {
customer_number: customer.customer_number,
cards,
grand_total: fmt(grand_total),
}
}
}
#[derive(Template)]
#[template(path = "index.html")]
struct IndexTemplate {
customers: Vec<(String, usize)>,
batches: Vec<String>,
}
#[derive(Template)]
#[template(path = "customer.html")]
struct CustomerTemplate {
customer: PreparedCustomer,
batches: Vec<String>,
generated_date: String,
}
fn main() -> Result<(), Box<dyn std::error::Error>> {
let data_dir = Path::new("input");
let output_dir = Path::new("output");
if !output_dir.exists() {
fs::create_dir_all(output_dir)?;
}
let mut batches: Vec<Batch> = Vec::new();
for entry in fs::read_dir(data_dir)? {
let entry = entry?;
let path = entry.path();
if path.extension().and_then(|s| s.to_str()) == Some("csv") {
match read_csv_file(&path) {
Ok(batch) => {
println!(
"Loaded {} transactions from {}",
batch.transactions.len(),
batch.filename
);
batches.push(batch);
}
Err(e) => {
eprintln!("Error reading {:?}: {}", path, e);
}
}
}
}
batches.sort_by(|a, b| a.filename.cmp(&b.filename));
let batch_filenames: Vec<String> = batches.iter().map(|b| b.filename.clone()).collect();
let customers = group_by_customer(&batches);
let index_customers: Vec<(String, usize)> = customers
.iter()
.map(|(num, c)| (num.clone(), c.cards.len()))
.collect();
let html = IndexTemplate {
customers: index_customers,
batches: batch_filenames.clone(),
}
.render()
.unwrap();
fs::write(output_dir.join("index.html"), html)?;
let generated_date = Utc::now().format("%Y-%m-%d %H:%M").to_string();
let customer_count = customers.len();
for (customer_num, customer) in customers {
let prepared = PreparedCustomer::from_customer(customer);
let customer_html = CustomerTemplate {
customer: prepared,
batches: batch_filenames.clone(),
generated_date: generated_date.clone(),
}
.render()
.unwrap();
let filename = format!("customer_{}.html", customer_num);
fs::write(output_dir.join(&filename), customer_html)?;
println!("Generated {}", filename);
}
println!(
"\nGenerated {} customer invoices in output/",
customer_count
);
Ok(())
}

51
src/templates.rs Normal file
View File

@@ -0,0 +1,51 @@
use askama::Template;
use askama_derive::Filter;
use chrono::Utc;
use std::collections::BTreeMap;
use invoice_generator::{group_by_customer, read_csv_file, Customer, Transaction};
#[derive(Template)]
#[template(path = "index.html")]
pub struct IndexTemplate {
pub customers: Vec<(String, Customer)>,
pub batches: Vec<String>,
}
#[derive(Template)]
#[template(path = "customer.html")]
pub struct CustomerTemplate {
pub customer: &Customer,
pub batches: Vec<String>,
}
pub fn calculate_card_totals(transactions: &[Transaction]) -> (f64, f64) {
let total_amount: f64 = transactions.iter().map(|t| t.amount).sum();
let total_volume: f64 = transactions.iter().map(|t| t.volume).sum();
(total_amount, total_volume)
}
pub fn calculate_grand_total(customer: &Customer) -> f64 {
customer
.cards
.values()
.flat_map(|t| t.iter())
.map(|t| t.amount)
.sum()
}
pub fn format_amount(amount: f64) -> String {
format!("{:.2}", amount)
}
pub fn format_volume(volume: f64) -> String {
format!("{:.2}", volume)
}
pub fn format_date(date: &chrono::NaiveDateTime) -> String {
date.format("%Y-%m-%d %H:%M").to_string()
}
pub fn now_string() -> String {
Utc::now().format("%Y-%m-%d %H:%M").to_string()
}

160
templates/customer.html Normal file
View File

@@ -0,0 +1,160 @@
<!DOCTYPE html>
<html lang="en">
<head>
<meta charset="UTF-8">
<title>Invoice - Customer {{ customer.customer_number }}</title>
<style>
@page {
size: A4;
margin: 15mm;
}
* {
box-sizing: border-box;
}
body {
font-family: Arial, sans-serif;
font-size: 12px;
margin: 0;
padding: 20px;
background: white;
}
@media print {
body {
padding: 0;
}
.card-section {
break-inside: avoid;
}
}
.header {
display: flex;
justify-content: space-between;
align-items: flex-end;
border-bottom: 2px solid #333;
padding-bottom: 10px;
margin-bottom: 15px;
}
.header h1 {
margin: 0;
font-size: 20px;
}
.header .meta {
text-align: right;
font-size: 11px;
color: #666;
}
.card-section {
margin-bottom: 15px;
border: 1px solid #ccc;
}
.card-header {
background: #f0f0f0;
padding: 8px 12px;
font-weight: bold;
border-bottom: 1px solid #ccc;
display: flex;
justify-content: space-between;
}
.card-summary {
padding: 6px 12px;
font-size: 11px;
background: #fafafa;
border-bottom: 1px solid #eee;
}
table {
width: 100%;
border-collapse: collapse;
font-size: 11px;
}
th, td {
padding: 6px 8px;
text-align: left;
}
th {
background: #f5f5f5;
font-weight: bold;
border-bottom: 1px solid #ccc;
}
.amount, .volume, .price {
text-align: right;
}
.card-total {
font-weight: bold;
background: #f9f9f9;
}
.card-total td {
border-top: 1px solid #333;
}
.grand-total {
background: #333;
color: white;
padding: 12px 15px;
font-size: 14px;
font-weight: bold;
text-align: right;
margin-top: 15px;
}
.grand-total span {
font-size: 18px;
margin-left: 20px;
}
</style>
</head>
<body>
<div class="header">
<div>
<h1>Invoice - Customer {{ customer.customer_number }}</h1>
<div>Batches: {{ batches | join(", ") }}</div>
</div>
<div class="meta">
<div>Generated: {{ generated_date }}</div>
</div>
</div>
{% for card in customer.cards %}
<div class="card-section">
<div class="card-header">
<span>Card: {{ card.card_number }}</span>
<span>{{ card.transactions.len() }} transactions</span>
</div>
<div class="card-summary">
Total: {{ card.total_amount }} | Volume: {{ card.total_volume }} L
</div>
<table>
<thead>
<tr>
<th class="date">Date</th>
<th>Product</th>
<th class="price">Price/L</th>
<th class="volume">Volume (L)</th>
<th class="amount">Amount</th>
<th>Receipt</th>
</tr>
</thead>
<tbody>
{% for tx in card.transactions %}
<tr>
<td class="date">{{ tx.date }}</td>
<td>{{ tx.quality_name }}</td>
<td class="price">{{ tx.price }}</td>
<td class="volume">{{ tx.volume }}</td>
<td class="amount">{{ tx.amount }}</td>
<td>{{ tx.receipt }}</td>
</tr>
{% endfor %}
<tr class="card-total">
<td colspan="3">Card Total</td>
<td class="volume">{{ card.total_volume }}</td>
<td class="amount">{{ card.total_amount }}</td>
<td></td>
</tr>
</tbody>
</table>
</div>
{% endfor %}
<div class="grand-total">
Grand Total:<span>{{ customer.grand_total }}</span>
</div>
</body>
</html>

87
templates/index.html Normal file
View File

@@ -0,0 +1,87 @@
<!DOCTYPE html>
<html lang="en">
<head>
<meta charset="UTF-8">
<title>Invoice Overview</title>
<style>
@page {
size: A4;
margin: 15mm;
}
* {
box-sizing: border-box;
}
body {
font-family: Arial, sans-serif;
font-size: 12px;
margin: 0;
padding: 20px;
background: white;
}
@media print {
body { padding: 0; }
}
h1 {
margin: 0 0 15px 0;
font-size: 18px;
border-bottom: 2px solid #333;
padding-bottom: 10px;
}
.batch-info {
margin-bottom: 15px;
font-size: 11px;
color: #666;
}
table {
width: 100%;
border-collapse: collapse;
font-size: 11px;
}
th, td {
padding: 8px 10px;
text-align: left;
border: 1px solid #ccc;
}
th {
background: #f0f0f0;
font-weight: bold;
}
tr:nth-child(even) {
background: #fafafa;
}
a {
color: #000;
text-decoration: none;
}
a:hover {
text-decoration: underline;
}
</style>
</head>
<body>
<h1>Invoice Overview</h1>
<div class="batch-info">
<strong>Processed Batches:</strong> {{ batches | join(", ") }}
</div>
<table>
<thead>
<tr>
<th>Customer Number</th>
<th>Cards</th>
<th>Invoice</th>
</tr>
</thead>
<tbody>
{% for (num, card_count) in customers %}
<tr>
<td>{{ num }}</td>
<td>{{ card_count }}</td>
<td><a href="customer_{{ num }}.html">View</a></td>
</tr>
{% endfor %}
</tbody>
</table>
</body>
</html>