Simplified logic re: like_kind dates and treatment.

This commit is contained in:
scoobybejesus 2019-10-12 13:55:41 -04:00
parent 1c7209eb95
commit 36b114223e
7 changed files with 42 additions and 65 deletions

View File

@ -163,7 +163,7 @@ pub(crate) fn elect_like_kind_treatment(cutoff_date_arg: &Option<String>) -> Res
Some(cutoff_date_arg) => { Some(cutoff_date_arg) => {
let provided_date = NaiveDate::parse_from_str(&cutoff_date_arg, "%y-%m-%d") let provided_date = NaiveDate::parse_from_str(&cutoff_date_arg, "%y-%m-%d")
.unwrap_or(NaiveDate::parse_from_str(&cutoff_date_arg, "%Y-%m-%d") .unwrap_or(NaiveDate::parse_from_str(&cutoff_date_arg, "%Y-%m-%d")
.expect("Date entered as -c command line arg has an incorrect format.")); .expect("Date entered as -l (like-kind cutoff date) command line arg has an incorrect format."));
println!("\nUse like-kind exchange treatment through {}? [Y/n/c] ('c' to 'change') ", provided_date); println!("\nUse like-kind exchange treatment through {}? [Y/n/c] ('c' to 'change') ", provided_date);

View File

@ -40,17 +40,14 @@ impl fmt::Display for InventoryCostingMethod {
} }
} }
#[derive(Clone)]
pub struct LikeKindSettings {
pub like_kind_cutoff_date: NaiveDate,
pub like_kind_basis_date_preserved: bool,
}
pub struct ImportProcessParameters { pub struct ImportProcessParameters {
pub export_path: PathBuf, pub export_path: PathBuf,
pub home_currency: String, pub home_currency: String,
pub enable_like_kind_treatment: bool, pub lk_treatment_enabled: bool,
pub lk_cutoff_date_string: String, /// NaiveDate either from "1-1-1" (default and not to be used) or the actual date chosen (or passed in via Cli option)
pub lk_cutoff_date: NaiveDate,
pub lk_basis_date_preserved: bool,
pub costing_method: InventoryCostingMethod, pub costing_method: InventoryCostingMethod,
pub input_file_date_separator: String, pub input_file_date_separator: String,
pub input_file_has_iso_date_style: bool, pub input_file_has_iso_date_style: bool,
@ -65,7 +62,6 @@ pub(crate) fn import_and_process_final(
HashMap<u16, RawAccount>, HashMap<u16, RawAccount>,
HashMap<u32, ActionRecord>, HashMap<u32, ActionRecord>,
HashMap<u32, Transaction>, HashMap<u32, Transaction>,
Option<LikeKindSettings>,
), Box<dyn Error>> { ), Box<dyn Error>> {
let mut transactions_map: HashMap<u32, Transaction> = HashMap::new(); let mut transactions_map: HashMap<u32, Transaction> = HashMap::new();
@ -121,22 +117,6 @@ pub(crate) fn import_and_process_final(
Ok(()) Ok(())
} }
let likekind_settings: Option<LikeKindSettings> = if settings.enable_like_kind_treatment {
let like_kind_cutoff_date = &settings.lk_cutoff_date_string;
Some(
LikeKindSettings {
like_kind_cutoff_date: NaiveDate::parse_from_str(&like_kind_cutoff_date, "%y-%m-%d")
.unwrap_or(NaiveDate::parse_from_str(&like_kind_cutoff_date, "%Y-%m-%d")
.expect("Found date string with improper format")),
like_kind_basis_date_preserved: true,
}
)
} else {
None
};
transactions_map = create_lots_mvmts::create_lots_and_movements( transactions_map = create_lots_mvmts::create_lots_and_movements(
transactions_map, transactions_map,
&action_records_map, &action_records_map,
@ -144,7 +124,9 @@ pub(crate) fn import_and_process_final(
&mut account_map, &mut account_map,
&settings.home_currency, &settings.home_currency,
&settings.costing_method, &settings.costing_method,
&likekind_settings, settings.lk_treatment_enabled,
settings.lk_cutoff_date,
settings.lk_basis_date_preserved,
&mut lot_map, &mut lot_map,
)?; )?;
@ -170,13 +152,11 @@ pub(crate) fn import_and_process_final(
println!(" Successfully added proceeds to movements."); println!(" Successfully added proceeds to movements.");
if let Some(lk_settings) = &likekind_settings { if settings.lk_treatment_enabled {
let cutoff_date = lk_settings.like_kind_cutoff_date; println!(" Applying like-kind treatment for cut-off date: {}.", settings.lk_cutoff_date);
println!(" Applying like-kind treatment for cut-off date: {}.", cutoff_date);
import_cost_proceeds_etc::apply_like_kind_treatment( import_cost_proceeds_etc::apply_like_kind_treatment(
cutoff_date,
&settings, &settings,
&action_records_map, &action_records_map,
&raw_account_map, &raw_account_map,
@ -185,7 +165,7 @@ pub(crate) fn import_and_process_final(
)?; )?;
println!(" Successfully applied like-kind treatment."); println!(" Successfully applied like-kind treatment.");
}; }
Ok((account_map, raw_account_map, action_records_map, transactions_map, likekind_settings)) Ok((account_map, raw_account_map, action_records_map, transactions_map))
} }

View File

@ -11,7 +11,7 @@ use chrono::NaiveDate;
use crate::transaction::{Transaction, ActionRecord, TxType, Polarity, TxHasMargin}; use crate::transaction::{Transaction, ActionRecord, TxType, Polarity, TxHasMargin};
use crate::account::{Account, RawAccount, Lot, Movement}; use crate::account::{Account, RawAccount, Lot, Movement};
use crate::core_functions::{InventoryCostingMethod, LikeKindSettings}; use crate::core_functions::{InventoryCostingMethod};
use crate::decimal_utils::{round_d128_1e8}; use crate::decimal_utils::{round_d128_1e8};
pub(crate) fn create_lots_and_movements( pub(crate) fn create_lots_and_movements(
@ -21,27 +21,13 @@ pub(crate) fn create_lots_and_movements(
acct_map: &HashMap<u16, Account>, acct_map: &HashMap<u16, Account>,
chosen_home_currency: &String, chosen_home_currency: &String,
chosen_costing_method: &InventoryCostingMethod, chosen_costing_method: &InventoryCostingMethod,
likekind_settings: &Option<LikeKindSettings>, enable_lk_treatment: bool,
like_kind_cutoff_date: NaiveDate,
lk_basis_date_preserved: bool,
lot_map: &HashMap<(RawAccount, u32), Lot>, lot_map: &HashMap<(RawAccount, u32), Lot>,
) -> Result<HashMap<u32,Transaction>, Box<dyn Error>> { ) -> Result<HashMap<u32,Transaction>, Box<dyn Error>> {
// Set values to be referred to repeatedly, potentially, in Incoming Exchange transactions let multiple_incoming_mvmts_per_ar = lk_basis_date_preserved;
let multiple_incoming_mvmts_per_ar = match &likekind_settings {
Some(likekind_settings) => {
likekind_settings.like_kind_basis_date_preserved
}
None => {
false
}
};
let mut like_kind_cutoff_date: NaiveDate = NaiveDate::parse_from_str("1/1/1", "%m/%d/%y")
.expect("NaiveDate string parsing failed. It shouldn't have. Try again.");
// Above sets date never to be used. Below modifies date in case it will be used.
if likekind_settings.is_some() {
let likekind_settings_clone = likekind_settings.clone().unwrap();
like_kind_cutoff_date = likekind_settings_clone.like_kind_cutoff_date;
}
// On with the creating of lots and movements. // On with the creating of lots and movements.
let length = txns_map.len(); let length = txns_map.len();

View File

@ -4,7 +4,6 @@
use std::collections::{HashMap}; use std::collections::{HashMap};
use std::error::Error; use std::error::Error;
use chrono::NaiveDate;
use decimal::d128; use decimal::d128;
use crate::transaction::{Transaction, TxType, ActionRecord, Polarity}; use crate::transaction::{Transaction, TxType, ActionRecord, Polarity};
@ -263,7 +262,6 @@ pub(crate) fn add_proceeds_to_movements(
} }
pub(crate) fn apply_like_kind_treatment( pub(crate) fn apply_like_kind_treatment(
cutoff_date: NaiveDate,
settings: &ImportProcessParameters, settings: &ImportProcessParameters,
ars: &HashMap<u32, ActionRecord>, ars: &HashMap<u32, ActionRecord>,
raw_acct_map: &HashMap<u16, RawAccount>, raw_acct_map: &HashMap<u16, RawAccount>,
@ -272,6 +270,7 @@ pub(crate) fn apply_like_kind_treatment(
) -> Result<(), Box<dyn Error>> { ) -> Result<(), Box<dyn Error>> {
let length = txns_map.len(); let length = txns_map.len();
let cutoff_date = settings.lk_cutoff_date;
for txn_num in 1..=length { for txn_num in 1..=length {
let txn_num = txn_num as u32; let txn_num = txn_num as u32;

View File

@ -123,7 +123,6 @@ fn main() -> Result<(), Box<dyn Error>> {
raw_acct_map, raw_acct_map,
action_records_map, action_records_map,
transactions_map, transactions_map,
like_kind_settings
) = core_functions::import_and_process_final(input_file_path, &settings)?; ) = core_functions::import_and_process_final(input_file_path, &settings)?;
let should_export = settings.should_export; let should_export = settings.should_export;

View File

@ -6,6 +6,8 @@ use std::path::PathBuf;
use std::error::Error; use std::error::Error;
use std::process; use std::process;
use chrono::NaiveDate;
use crate::cli_user_choices; use crate::cli_user_choices;
use crate::core_functions::{ImportProcessParameters, InventoryCostingMethod}; use crate::core_functions::{ImportProcessParameters, InventoryCostingMethod};
use crate::skip_wizard; use crate::skip_wizard;
@ -48,13 +50,24 @@ pub (crate) fn run_setup(args: super::Cli) -> Result<(PathBuf, ImportProcessPara
output_dir_path, output_dir_path,
) = wizard_or_not(args.flags.accept_args, wizard_or_not_args)?; ) = wizard_or_not(args.flags.accept_args, wizard_or_not_args)?;
let like_kind_cutoff_date;
if like_kind_election {
like_kind_cutoff_date = NaiveDate::parse_from_str(&like_kind_cutoff_date_string, "%y-%m-%d")
.unwrap_or(NaiveDate::parse_from_str(&like_kind_cutoff_date_string, "%Y-%m-%d")
.expect("Command line date (like-kind cutoff option) has an incorrect format. Program must abort."));
} else {
like_kind_cutoff_date = NaiveDate::parse_from_str(&"1-1-1", "%y-%m-%d").unwrap();
}
let settings = ImportProcessParameters { let settings = ImportProcessParameters {
home_currency: args.opts.home_currency.into_string().unwrap().to_uppercase(), home_currency: args.opts.home_currency.into_string().unwrap().to_uppercase(),
input_file_has_iso_date_style: args.flags.iso_date, input_file_has_iso_date_style: args.flags.iso_date,
input_file_date_separator: date_separator.to_string(), input_file_date_separator: date_separator.to_string(),
costing_method: costing_method_choice, costing_method: costing_method_choice,
enable_like_kind_treatment: like_kind_election, lk_treatment_enabled: like_kind_election,
lk_cutoff_date_string: like_kind_cutoff_date_string, lk_cutoff_date: like_kind_cutoff_date,
lk_basis_date_preserved: true, // TODO
should_export: should_export, should_export: should_export,
export_path: output_dir_path, export_path: output_dir_path,
}; };

View File

@ -78,12 +78,12 @@ Home currency: {}
Enable like-kind treatment: {}", Enable like-kind treatment: {}",
settings.costing_method, settings.costing_method,
settings.home_currency, settings.home_currency,
settings.enable_like_kind_treatment settings.lk_treatment_enabled
)?; )?;
if settings.enable_like_kind_treatment { if settings.lk_treatment_enabled {
writeln!(file, "Like-kind cut-off date: {}.", writeln!(file, "Like-kind cut-off date: {}.",
settings.lk_cutoff_date_string settings.lk_cutoff_date
)?; )?;
} }
@ -226,12 +226,12 @@ Home currency: {}
Enable like-kind treatment: {}", Enable like-kind treatment: {}",
settings.costing_method, settings.costing_method,
settings.home_currency, settings.home_currency,
settings.enable_like_kind_treatment settings.lk_treatment_enabled
)?; )?;
if settings.enable_like_kind_treatment { if settings.lk_treatment_enabled {
writeln!(file, "Like-kind cut-off date: {}.", writeln!(file, "Like-kind cut-off date: {}.",
settings.lk_cutoff_date_string settings.lk_cutoff_date
)?; )?;
} }
@ -308,12 +308,12 @@ Home currency: {}
Enable like-kind treatment: {}", Enable like-kind treatment: {}",
settings.costing_method, settings.costing_method,
settings.home_currency, settings.home_currency,
settings.enable_like_kind_treatment settings.lk_treatment_enabled
)?; )?;
if settings.enable_like_kind_treatment { if settings.lk_treatment_enabled {
writeln!(file, "Like-kind cut-off date: {}.", writeln!(file, "Like-kind cut-off date: {}.",
settings.lk_cutoff_date_string settings.lk_cutoff_date
)?; )?;
} }