146 lines
3.8 KiB
Ruby
146 lines
3.8 KiB
Ruby
# coding: utf-8
|
|
require 'yaml'
|
|
require 'mysql2'
|
|
require 'optparse'
|
|
|
|
load 'config.rb'
|
|
|
|
Options = {}
|
|
OptionParser.new do |opts|
|
|
opts.banner = "Usage: invoice.rb [options]"
|
|
|
|
opts.on("-d", "--draft", "Generate invoices with 'DRAFT' watermark") do |d|
|
|
Options[:draft] = d
|
|
end
|
|
opts.on("-v", "--verbose", "Run verbosely") do |v|
|
|
Options[:verbose] = v
|
|
end
|
|
opts.on("-s", "--silent", "Run silently (overrules -v)") do |s|
|
|
Options[:silent] = true
|
|
Options[:verbose] = false
|
|
end
|
|
end.parse!
|
|
|
|
Transaction = Struct.new(:time, :person, :product, :amount)
|
|
Product = Struct.new(:id, :description, :price) do
|
|
def to_s
|
|
"#{id} => #{description}, #{price}"
|
|
end
|
|
end
|
|
|
|
def read_file(file)
|
|
rows = Array.new
|
|
CSV.foreach(file, col_sep: ';', converters: :float) do |row|
|
|
rows << Transaction.new(row[0],row[1],row[2],row[3])
|
|
end
|
|
rows
|
|
end
|
|
|
|
def read_db(db)
|
|
transactions = Array.new
|
|
db.query("SELECT * FROM Transactions").each do |trans|
|
|
if date_of_prev('monday').to_time < trans["time"]
|
|
transactions << Transaction.new(trans["time"], trans["buyer"],
|
|
trans["product"], trans["amount"])
|
|
end
|
|
end
|
|
return transactions
|
|
end
|
|
|
|
# Partition transactions into the persons who they belong to
|
|
def partition_transactions(transactions)
|
|
persons = Hash.new
|
|
transactions.each do |trans|
|
|
# Initialize data structures if not already done
|
|
if !(persons[trans.person])
|
|
persons[trans.person] = {}
|
|
end
|
|
if !(persons[trans.person][trans.product])
|
|
persons[trans.person][trans.product] = 0
|
|
end
|
|
|
|
persons[trans.person][trans.product] += trans.amount
|
|
end
|
|
|
|
return persons
|
|
end
|
|
|
|
def generate_receipt(persons,draft)
|
|
counter = 0
|
|
|
|
# Iterate through persons who have purchased something in the given timeframe
|
|
persons.each do |person, products|
|
|
counter += 1
|
|
|
|
yaml = YAML_BASE.clone
|
|
yaml["to"] = Names[person]
|
|
yaml["invoice-nr"] = Time.now.strftime('%Y%W') + "-" + counter.to_s
|
|
|
|
# Iterate through products purchased and write data to yaml hash
|
|
products.each do |id, amount|
|
|
yaml["service"] = Array.new unless yaml["service"]
|
|
hash = {description: Products[id].description,
|
|
pieceprice: Products[id].price.to_i,
|
|
price: Products[id].price.to_i*amount, amount: amount}
|
|
|
|
# Convert Symbols to Strings to ensure compatibility with pandoc
|
|
yaml["service"] << Hash[hash.map{ |k, v| [k.to_s, v] }]
|
|
end
|
|
|
|
# Write draft watermark on invoice if marked as draft
|
|
if draft
|
|
yaml["draft"] = "true"
|
|
yaml["drafttext"] = "Udkast"
|
|
end
|
|
|
|
# Convert Symbols to Strings to ensure compatibility with pandoc
|
|
output = Hash[yaml.map{ |k, v| [k.to_s, v] }].to_yaml
|
|
output += "---\n"
|
|
|
|
# Write output to file
|
|
File.open("./pandoc/details.yml","w") do |file|
|
|
file << output
|
|
end
|
|
|
|
# Generate PDF using pandoc
|
|
`cd pandoc/; make -B`
|
|
`cp pandoc/output.pdf #{OUTPUT_PATH}/#{yaml["invoice-nr"]}.pdf`
|
|
|
|
puts "#{person}: #{yaml["invoice-nr"]}.pdf" unless Options[:silent]
|
|
end
|
|
end
|
|
|
|
def date_of_next(day)
|
|
date = Date.parse(day)
|
|
delta = date > Date.today ? 0 : 7
|
|
date + delta
|
|
end
|
|
|
|
def date_of_prev(day)
|
|
date = Date.parse(day)
|
|
delta = date < Date.today ? 0 : 7
|
|
date - delta
|
|
end
|
|
|
|
db = Mysql2::Client.new(:host => DB_HOST, :username => DB_USER, :database => DB_DB)
|
|
|
|
# Initialize product and member hashses
|
|
Products = Hash.new
|
|
Names = Hash.new
|
|
Prices = Hash.new
|
|
db.query("SELECT * FROM Products").each do |p|
|
|
product = Product.new(p['id'], p['description'], p['price'])
|
|
Products[p['id']] = product
|
|
|
|
if Options[:verbose]
|
|
puts "Found product: #{product}"
|
|
end
|
|
end
|
|
|
|
db.query("SELECT * FROM Persons").each do |member|
|
|
Names[member['id']] = member['name']
|
|
end
|
|
|
|
partition = partition_transactions(read_db(db))
|
|
generate_receipt(partition,Options[:draft])
|