From 9fe2c5d6fb165a0c45a7ac377b9cd1acf4116971 Mon Sep 17 00:00:00 2001
From: Palash Tyagi <23239946+Magnus167@users.noreply.github.com>
Date: Thu, 14 Nov 2024 02:02:44 +0000
Subject: [PATCH] feat: add utils module and integrate metrics into
 save_indicators_as_csv function

---
 src/lib.rs       | 1 +
 src/main.rs      | 6 +++++-
 src/utils/mod.rs | 1 +
 3 files changed, 7 insertions(+), 1 deletion(-)
 create mode 100644 src/utils/mod.rs

diff --git a/src/lib.rs b/src/lib.rs
index b3529ae..fa9cba6 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -1,3 +1,4 @@
 #![doc = include_str!("../README.md")]
 
 pub mod download;
+pub mod utils;
\ No newline at end of file
diff --git a/src/main.rs b/src/main.rs
index 871f6cd..833979f 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -26,10 +26,14 @@ fn main() {
 
     println!("Retrieving indicators for {} tickers", sel_tickers.len());
     start = std::time::Instant::now();
-
+    let all_metrics: Vec<String> = ["value", "grading", "eop_lag", "mop_lag"]
+        .iter()
+        .map(|x| x.to_string())
+        .collect();
     let res = jpamqs_download.save_indicators_as_csv(
         JPMaQSDownloadGetIndicatorArgs {
             tickers: sel_tickers.clone(),
+            metrics: all_metrics,
             ..Default::default()
         },
         "./data/",
diff --git a/src/utils/mod.rs b/src/utils/mod.rs
new file mode 100644
index 0000000..51ba6a5
--- /dev/null
+++ b/src/utils/mod.rs
@@ -0,0 +1 @@
+pub mod dftools;
\ No newline at end of file