tktax-check 0.2.2

A Rust library for identifying and handling treasury or check transactions in the TKTAX system.
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

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

#[derive(Getters,Debug,Clone)]
#[getset(get="pub")]
pub struct CheckOrTreasuryTransaction<'a,TxCat:TransactionCategory + 'static> {
    category: TxCat,
    tx:       &'a Transaction,
}

impl<'a,TxCat:TransactionCategory + 'static> fmt::Display for CheckOrTreasuryTransaction<'a,TxCat> {

    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        write!(f, "{} - {}", self.category, self.tx)
    }
}

impl<'a,TxCat:TransactionCategory + 'static> LineItem<TxCat> for CheckOrTreasuryTransaction<'a,TxCat> {

    fn tag() -> &'static str {
        "check-or-treasury"
    }

    fn quantity(&self) -> Decimal {
        Decimal::ONE
    }

    fn price(&self) -> MonetaryAmount {
        self.tx_amount()
    }

    fn category(&self) -> TxCat {
        self.category
    }

    fn tx_amount(&self) -> MonetaryAmount {
        self.tx.amount()
    }

    fn tx(&self) -> Option<&Transaction> {
        Some(&self.tx)
    }
}

pub trait GetTreasuryTransactionsAndChecks {

    fn treasury_transactions_and_checks<TxCat:TransactionCategory + 'static>(&self, category_map: &CategoryMap<TxCat>) 
        -> Vec<CheckOrTreasuryTransaction<TxCat>>;
}

impl GetTreasuryTransactionsAndChecks for Account {

    fn treasury_transactions_and_checks<TxCat:TransactionCategory + 'static>(&self, category_map: &CategoryMap<TxCat>) 
        -> Vec<CheckOrTreasuryTransaction<TxCat>> 
    {
        let mut line_items = vec![];

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

            let predicted  = predict_category(&tx.description(), category_map);

            if predicted.len() > 0 {

                let prediction = &predicted[0];

                let categories = TxCat::treasury_and_checks_categories();

                if categories.contains(&prediction.category()) {

                    line_items.push(CheckOrTreasuryTransaction { category: *prediction.category(), tx });
                }
            }

        }

        line_items.sort_by(|a,b| a.category.cmp(&b.category));

        line_items
    }
}