Add account type
This commit is contained in:
moneymgr_backend
assets
migrations
2025-03-17-173101_initial_structure
src
moneymgr_web
@ -1,10 +1,31 @@
|
||||
use crate::models::users::UserID;
|
||||
use crate::schema::*;
|
||||
use diesel::prelude::*;
|
||||
use std::fmt::{Display, Formatter};
|
||||
|
||||
#[derive(Debug, Clone, Copy, serde::Serialize, serde::Deserialize, PartialEq, Eq)]
|
||||
pub struct AccountID(pub i32);
|
||||
|
||||
#[derive(Debug, Clone, Copy, serde::Serialize, serde::Deserialize, PartialEq, Eq)]
|
||||
pub enum AccountType {
|
||||
#[serde(rename = "C")]
|
||||
Cash,
|
||||
#[serde(rename = "B")]
|
||||
Bank,
|
||||
#[serde(rename = "S")]
|
||||
Saving,
|
||||
}
|
||||
|
||||
impl Display for AccountType {
|
||||
fn fmt(&self, f: &mut Formatter<'_>) -> std::fmt::Result {
|
||||
write!(
|
||||
f,
|
||||
"{}",
|
||||
serde_json::to_string(self).unwrap().replace('"', "")
|
||||
)
|
||||
}
|
||||
}
|
||||
|
||||
#[derive(Queryable, Debug, Clone, serde::Serialize)]
|
||||
pub struct Account {
|
||||
id: i32,
|
||||
@ -12,6 +33,7 @@ pub struct Account {
|
||||
user_id: i32,
|
||||
pub time_create: i64,
|
||||
pub time_update: i64,
|
||||
r#type: String,
|
||||
pub default_account: bool,
|
||||
}
|
||||
|
||||
@ -23,13 +45,22 @@ impl Account {
|
||||
pub fn user_id(&self) -> UserID {
|
||||
UserID(self.user_id)
|
||||
}
|
||||
|
||||
pub fn account_type(&self) -> AccountType {
|
||||
serde_json::from_str(format!("\"{}\"", self.r#type).as_str()).unwrap()
|
||||
}
|
||||
|
||||
pub fn set_account_type(&mut self, t: AccountType) {
|
||||
self.r#type = t.to_string();
|
||||
}
|
||||
}
|
||||
|
||||
#[derive(Insertable)]
|
||||
#[derive(Insertable, Debug)]
|
||||
#[diesel(table_name = accounts)]
|
||||
pub struct NewAccount<'a> {
|
||||
pub name: &'a str,
|
||||
pub user_id: i32,
|
||||
pub time_create: i64,
|
||||
pub time_update: i64,
|
||||
pub type_: String,
|
||||
}
|
||||
|
Reference in New Issue
Block a user