Swaps budget updates for budget drips

1 files changed, 7 insertions, 0 deletions

diff --git a/rust/schist_models/src/category.rs b/rust/schist_models/src/category.rs
index 55c7b97..c85ff51 100644
--- a/rust/schist_models/src/category.rs
+++ b/rust/schist_models/src/category.rs
@@ -2,9 +2,16 @@ use derive_builder::Builder;
 use diesel::prelude::*;
 use serde::{Deserialize, Serialize};
 
+use crate::{budget_period_unit::BudgetPeriodUnit, date_utc::DateUtc};
+
 #[derive(Builder, Queryable, Identifiable, Selectable, Debug, PartialEq, Serialize, Deserialize, Insertable)]
 #[diesel(table_name = crate::schema::categories)]
 pub struct Category {
     pub id: i32,
     pub name: String,
+    pub balance: i32,
+    pub balance_date: DateUtc,
+    pub budget_period: i32,
+    pub budget_period_unit: BudgetPeriodUnit,
+    pub budget_quantity: i32,
 }