quantrs 0.1.8

A tiny Rust library for quantitative finance
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
31
32
33
34
35
36
37

//! Module listing supported data providers.

use crate::data::{CompanyOverview, GlobalQuote};

use super::traits::{FundamentalsProvider, QuoteProvider};

use std::io::Error;

pub use alpha_vantage::AlphaVantageSource;

/// Enum representing different data providers.
pub enum DataProvider {
    AlphaVantage(AlphaVantageSource),
}

mod alpha_vantage;

/// Implementation on the DataProvider enum to perform actions based on the provider type.
impl DataProvider {
    /// Data source related associated functions.
    /// This allows for reusing of the same client that is created.
    pub fn alpha_vantage(user_key: &str) -> Self {
        Self::AlphaVantage(AlphaVantageSource::new(user_key))
    }

    pub async fn get_stock_quote(&self, symbol: &str) -> Result<GlobalQuote, Error> {
        match self {
            DataProvider::AlphaVantage(source) => source.get_stock_quote(symbol).await,
        }
    }

    pub async fn get_company_overview(&self, symbol: &str) -> Result<CompanyOverview, Error> {
        match self {
            DataProvider::AlphaVantage(source) => source.get_company_overview(symbol).await,
        }
    }
}