tktax-histogram 0.2.2

A Rust crate that provides robust monthly histogram utilities for financial transactions, using specialized types from the TKTAX ecosystem.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89

// ---------------- [ File: tktax-histogram/src/histogram_builder.rs ]
crate::ix!();

#[derive(Getters)]
#[getset(get="pub")]
pub struct AccountHistogramBuilder<'a> {
    account: &'a Account,
    bins:    Vec<MonetaryAmount>,
}

impl<'a> AccountHistogramBuilder<'a> {

    pub fn new(account: &'a Account, bins: Vec<MonetaryAmount>) -> Self {
        Self { account, bins }
    }

    pub fn build_histogram(&self, strategy: &HistogramDisplayStrategy) -> AccountHistogram {

        let mut histogram = AccountHistogram::new(strategy);

        let full_data = self.organize_histogram_data();

        // Collect and sort the months
        let mut months: Vec<NaiveDate> = full_data.keys().cloned().collect();

        months.sort();

        for month in &months {

            let bins = full_data.get(month).unwrap();

            let month_number = month.month();

            let mut histogram_month = HistogramMonth::new(month_number);

            for (i, bin) in bins.iter().enumerate()
                //.filter(|(i,bin)| bin.len() > 0) 
            {
                let month_txns: Vec<_> = bin.iter().sorted().rev().map(
                    |tx| HistogramMonthBinTxnBuilder::default()
                        .date(*tx.transaction_date())
                        .amount(tx.amount())
                        .description(tx.description())
                        .build()
                        .unwrap()
                ).collect();

                histogram_month.push(
                    HistogramMonthBinBuilder::default()
                        .price_range(self.bins[i] .. self.bins[i + 1])
                        .txns(month_txns)
                        .build()
                        .unwrap()
                );
            }

            histogram.push(histogram_month);
        }

        histogram
    }

    fn organize_histogram_data(&self) -> HashMap<NaiveDate, Vec<Vec<&'a Transaction>>> {

        let mut histograms: HashMap<NaiveDate, Vec<Vec<&'a Transaction>>> = HashMap::new();

        for tx in self.account.txns().iter() {

            let month_start = tx.transaction_date().with_day0(0).unwrap();

            let histogram = histograms.entry(month_start).or_insert_with(|| {
                let mut bin_vec = Vec::new();
                for _ in 0..self.bins.len() - 1 {
                    bin_vec.push(Vec::new());
                }
                bin_vec
            });

            for i in 0..self.bins.len() - 1 {
                if tx.amount() >= self.bins[i] && tx.amount() < self.bins[i + 1] {
                    histogram[i].push(tx);
                    break;
                }
            }
        }

        histograms
    }
}