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:
10
Cargo.toml
Normal file
10
Cargo.toml
Normal 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
126
src/invoice_generator.rs
Normal 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
175
src/main.rs
Normal 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
51
src/templates.rs
Normal 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
160
templates/customer.html
Normal 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
87
templates/index.html
Normal 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>
|
||||
Reference in New Issue
Block a user