cryo_freeze 0.3.2

cryo is the easiest way to extract blockchain data to parquet, csv, or json
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30

use crate::{collect_partition, CollectError, Query, Source};
use polars::prelude::*;

/// collect single dataframe
pub async fn collect(query: Arc<Query>, source: Arc<Source>) -> Result<DataFrame, CollectError> {
    query.is_valid()?;
    let datatype = if query.datatypes.len() != 1 {
        return Err(CollectError::CollectError(
            "collect() can only collect a single datatype".to_string(),
        ))
    } else {
        query.datatypes[0].clone()
    };
    let partition = if query.partitions.len() != 1 {
        return Err(CollectError::CollectError(
            "collect() can only collect a single datatype".to_string(),
        ))
    } else {
        query.partitions[0].clone()
    };
    let results = collect_partition(datatype, partition, query, source).await?;
    if results.len() > 1 {
        Err(CollectError::CollectError("collect() only returns single dataframes".to_string()))
    } else {
        match results.into_iter().next() {
            Some((_datatype, df)) => Ok(df),
            None => Err(CollectError::CollectError("no dataframe result returned".to_string())),
        }
    }
}