summaryrefslogtreecommitdiff
path: root/src/banking
diff options
context:
space:
mode:
Diffstat (limited to 'src/banking')
-rw-r--r--src/banking/account.rs8
-rw-r--r--src/banking/asset.rs10
2 files changed, 13 insertions, 5 deletions
diff --git a/src/banking/account.rs b/src/banking/account.rs
index 95cb71b..b13e72f 100644
--- a/src/banking/account.rs
+++ b/src/banking/account.rs
@@ -7,12 +7,12 @@ pub struct Account {
pub transactions : Vec<Transaction>,
pub institute : String,
pub groupFile : String,
- pub category : String
-
+ pub category : String,
+ pub riskCategory : String
}
impl Account {
- pub fn new(name : String, iban : String, transactions : Vec<String>, institute : String, groupFile : String, category : String) -> Account {
+ pub fn new(name : String, iban : String, transactions : Vec<String>, institute : String, groupFile : String, category : String, riskCategory : String) -> Account {
let mut trans = Vec::new();
if institute == "SpecialParser" {
}
@@ -29,7 +29,7 @@ impl Account {
}
}
}
- Account { name : name, iban : iban, transactions : trans, institute : institute, groupFile : groupFile, category : category }
+ Account { name : name, iban : iban, transactions : trans, institute : institute, groupFile : groupFile, category : category, riskCategory : riskCategory }
}
pub fn get_balance(&self, date : chrono::NaiveDate) -> f32 {
diff --git a/src/banking/asset.rs b/src/banking/asset.rs
index 642f153..3255b29 100644
--- a/src/banking/asset.rs
+++ b/src/banking/asset.rs
@@ -59,6 +59,14 @@ impl Asset {
if let Some(i) = tmp.get(0) {
category = i.to_string();
}
+ match config.get("RiskCategory") {
+ Some(i) => tmp = i,
+ None => panic!("asset: ini file: no \"RiskCategory\" for account: {}", account_name)
+ }
+ let mut risk_category = String::from("");
+ if let Some(i) = tmp.get(0) {
+ risk_category = i.to_string();
+ }
match config.get("IBAN") {
Some(i) => tmp = i,
None => panic!("asset: no name for account {}", account_name)
@@ -75,7 +83,7 @@ impl Asset {
if let Some(i) = tmp.get(0) {
groupFile = i.to_string();
}
- accounts.push(Account::new(account_name, iban, trans_files.to_vec(), institute, groupFile, category));
+ accounts.push(Account::new(account_name, iban, trans_files.to_vec(), institute, groupFile, category, risk_category));
}