task-072

Co-authored-by: Joe Carstairs <jcarstairs@scottlogic.com>
Reviewed-on: https://git.joeac.net/joeac/schist/pulls/3
Co-authored-by: Joe Carstairs <me@joeac.net>
Co-committed-by: Joe Carstairs <me@joeac.net>

1 files changed, 10 insertions, 1 deletions

diff --git a/transformers/actualbudget_transformer/actualbudget_models/src/actualbudget_transaction.rs b/transformers/actualbudget_transformer/actualbudget_models/src/actualbudget_transaction.rs
index c8ac522..d6270da 100644
--- a/transformers/actualbudget_transformer/actualbudget_models/src/actualbudget_transaction.rs
+++ b/transformers/actualbudget_transformer/actualbudget_models/src/actualbudget_transaction.rs
@@ -3,7 +3,9 @@ use diesel::prelude::{Identifiable, Queryable, QueryableByName, Selectable};
 
 use crate::actualbudget_date::ActualbudgetDate;
 
-#[derive(Builder, Debug, PartialEq, Eq, Identifiable, QueryableByName, Queryable, Selectable, Clone)]
+#[derive(
+    Builder, Debug, PartialEq, Eq, Identifiable, QueryableByName, Queryable, Selectable, Clone,
+)]
 #[diesel(table_name = crate::actualbudget_schema::v_transactions)]
 pub struct ActualbudgetTransaction {
     pub id: String,
@@ -17,4 +19,11 @@ pub struct ActualbudgetTransaction {
     pub notes: Option<String>,
     pub date: ActualbudgetDate,
     pub transfer_id: Option<String>,
+    pub starting_balance_flag: i32,
+}
+
+impl ActualbudgetTransaction {
+    pub fn is_starting_balance(&self) -> bool {
+        self.starting_balance_flag != 0
+    }
 }