fix: create snapshots directory if it doesn't exist (#550)

pyth-network · Jan 6, 2025 · 1a7e04a · 1a7e04a
1 parent 4962f68
commit 1a7e04a
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/staking/cli/src/instructions.rs b/staking/cli/src/instructions.rs
@@ -1098,6 +1098,10 @@ pub fn save_stake_accounts_snapshot(rpc_client: &RpcClient) {
         )
         .collect::<Vec<_>>();
 
+    if !std::path::Path::new("snapshots").exists() {
+        std::fs::create_dir_all("snapshots").unwrap();
+    }
+
     let timestamp = chrono::Utc::now().format("%Y-%m-%d_%H:%M:%S").to_string();
     let file = File::create(format!("snapshots/snapshot-{}.csv", timestamp)).unwrap();
     let mut writer = BufWriter::new(file);