Don't fail if company is not found.

vaijira · Dec 22, 2024 · 459c4aa · 459c4aa
1 parent 4646c39
commit 459c4aa
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 4 deletions.
diff --git a/src/parsers/ib.rs b/src/parsers/ib.rs
@@ -272,10 +272,18 @@ impl IBParser {
         let company_info = self
             .companies_info
             .get(*symbol)
-            .ok_or_else(|| anyhow!("Not company info found"))?;
+            .cloned()
+            .or_else(|| {
+                log::error!("Not company info found for {}", symbol);
+                Some(CompanyInfo {
+                    name: symbol.to_string(),
+                    isin: "".to_string(),
+                })
+            })
+            .unwrap();
 
         Ok(BalanceNote::new(
-            company_info.clone(),
+            company_info,
             String::from(""),
             Decimal::from_str(&decimal::normalize_str(quantity))?
                 * Decimal::from_str(&decimal::normalize_str(mult))?,

diff --git a/src/parsers/ib_csv.rs b/src/parsers/ib_csv.rs
@@ -182,10 +182,20 @@ impl IBCSVParser {
         let company_info = self
             .companies_info
             .get(symbol)
-            .ok_or_else(|| anyhow!("Not company info found"))?;
+            .cloned()
+            .or_else({
+                || {
+                    log::error!("Not company info found for {}", symbol);
+                    Some(CompanyInfo {
+                        name: symbol.to_string(),
+                        isin: "".to_string(),
+                    })
+                }
+            })
+            .unwrap();
 
         Ok(BalanceNote::new(
-            company_info.clone(),
+            company_info,
             String::from(""),
             Decimal::from_str(&decimal::normalize_str(quantity))?
                 * Decimal::from_str(&decimal::normalize_str(mult))?,