add a csv printer
This commit is contained in:
57
cmd/any2anexoj-cli/csv_writer.go
Normal file
57
cmd/any2anexoj-cli/csv_writer.go
Normal file
@@ -0,0 +1,57 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"encoding/csv"
|
||||
"fmt"
|
||||
"io"
|
||||
|
||||
"github.com/nmoniz/any2anexoj/internal"
|
||||
)
|
||||
|
||||
type CSVWriter struct {
|
||||
w *csv.Writer
|
||||
}
|
||||
|
||||
func NewCSVWriter(w io.Writer) *CSVWriter {
|
||||
return &CSVWriter{w: csv.NewWriter(w)}
|
||||
}
|
||||
|
||||
func (cw *CSVWriter) Render(aw *internal.AggregatorWriter) error {
|
||||
err := cw.w.Write([]string{
|
||||
"source_country", "code",
|
||||
"realization_year", "realization_month", "realization_day", "realization_value",
|
||||
"acquisition_year", "acquisition_month", "acquisition_day", "acquisition_value",
|
||||
"expenses", "foreign_tax_paid", "counter_country",
|
||||
})
|
||||
if err != nil {
|
||||
return fmt.Errorf("write csv header: %w", err)
|
||||
}
|
||||
|
||||
for ri := range aw.Iter() {
|
||||
err := cw.w.Write(reportItemToRow(ri))
|
||||
if err != nil {
|
||||
return fmt.Errorf("write csv row: %w", err)
|
||||
}
|
||||
}
|
||||
|
||||
cw.w.Flush()
|
||||
return cw.w.Error()
|
||||
}
|
||||
|
||||
func reportItemToRow(ri internal.ReportItem) []string {
|
||||
return []string{
|
||||
fmt.Sprintf("%d", ri.AssetCountry),
|
||||
string(ri.Nature),
|
||||
fmt.Sprintf("%d", ri.SellTimestamp.Year()),
|
||||
fmt.Sprintf("%d", int(ri.SellTimestamp.Month())),
|
||||
fmt.Sprintf("%d", ri.SellTimestamp.Day()),
|
||||
ri.SellValue.StringFixed(2),
|
||||
fmt.Sprintf("%d", ri.BuyTimestamp.Year()),
|
||||
fmt.Sprintf("%d", int(ri.BuyTimestamp.Month())),
|
||||
fmt.Sprintf("%d", ri.BuyTimestamp.Day()),
|
||||
ri.BuyValue.StringFixed(2),
|
||||
ri.Fees.StringFixed(2),
|
||||
ri.Taxes.StringFixed(2),
|
||||
fmt.Sprintf("%d", ri.BrokerCountry),
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user