NexVeridian/ark-invest-api-rust-data
1
0
Fork 0
mirror of https://github.com/NexVeridian/ark-invest-api-rust-data.git synced 2025-09-02 09:59:12 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

0.3.11

Browse source
This commit is contained in:
Elijah McMorris 2023-06-19 22:04:13 +00:00 • committed by NexVeridian
parent 1bfbbaa826
commit 0de1bcb725
Signed by: NexVeridian
SSH key fingerprint: SHA256:bsA1SKZxuEcEVHAy3gY1HUeM5ykRJl0U0kQHQn0hMg8
6 changed files with 26 additions and 17857 deletions
Download patch file Download diff file Expand all files Collapse all files

10
src/util.rs
View file

@ -6,8 +6,9 @@ use polars::prelude::{DataFrame, StrptimeOptions, UniqueKeepStrategy};
use reqwest::blocking::Client;
use serde_json::Value;
use std::error::Error;
use std::fs::File;
use std::fs::{create_dir_all, File};
use std::io::Cursor;
use std::path::{Path, PathBuf};
use std::result::Result;
use strum_macros::EnumIter;
@ -157,7 +158,12 @@ impl Ark {
}
fn write_df_parquet(path: String, df: DF) -> Result<(), Box<dyn Error>> {
ParquetWriter::new(File::create(path)?).finish(&mut df.collect()?)?;
if let Some(parent) = Path::new(&path).parent() {
if !parent.exists() {
create_dir_all(parent)?;
}
}
ParquetWriter::new(File::create(&path)?).finish(&mut df.collect()?)?;
Ok(())
}