yfinance-rs 0.2.0

Ergonomic Rust client for Yahoo Finance, supporting historical prices, real-time streaming, options, fundamentals, and more.
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
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101

use crate::core::conversions::*;
use crate::history::wire::QuoteBlock;
use paft::prelude::*;

use super::adjust::price_factor_for_row;

pub fn assemble_candles(
    ts: &[i64],
    q: &QuoteBlock,
    adj: &[Option<f64>],
    auto_adjust: bool,
    keepna: bool,
    cum_split_after: &[f64],
    currency: Option<&str>,
) -> (Vec<Candle>, Vec<f64>) {
    let mut out = Vec::new();
    let mut raw_close_vec = Vec::new();

    for (i, &t) in ts.iter().enumerate() {
        let getter_f64 = |v: &Vec<Option<f64>>| v.get(i).and_then(|x| *x);
        let mut open = getter_f64(&q.open);
        let mut high = getter_f64(&q.high);
        let mut low = getter_f64(&q.low);
        let mut close = getter_f64(&q.close);
        let volume0 = q.volume.get(i).and_then(|x| *x);

        let raw_close_val = close.unwrap_or(f64::NAN);

        if auto_adjust {
            let pf = price_factor_for_row(i, adj.get(i).and_then(|x| *x), close, cum_split_after);

            if let Some(v) = open.as_mut() {
                *v *= pf;
            }
            if let Some(v) = high.as_mut() {
                *v *= pf;
            }
            if let Some(v) = low.as_mut() {
                *v *= pf;
            }
            if let Some(v) = close.as_mut() {
                *v *= pf;
            }

            let volume_adj = volume0.map(|v| {
                #[allow(clippy::cast_precision_loss)]
                let v_adj = (v as f64) * cum_split_after[i];
                if v_adj.is_finite() && v_adj >= 0.0 {
                    #[allow(clippy::cast_possible_truncation, clippy::cast_sign_loss)]
                    (v_adj.round() as u64)
                } else {
                    v
                }
            });

            if let (Some(ov), Some(hv), Some(lv), Some(cv)) = (open, high, low, close) {
                out.push(Candle {
                    ts: i64_to_datetime(t),
                    open: f64_to_money_with_currency_str(ov, currency),
                    high: f64_to_money_with_currency_str(hv, currency),
                    low: f64_to_money_with_currency_str(lv, currency),
                    close: f64_to_money_with_currency_str(cv, currency),
                    volume: volume_adj,
                });
                raw_close_vec.push(raw_close_val);
            } else if keepna {
                out.push(Candle {
                    ts: i64_to_datetime(t),
                    open: f64_to_money_with_currency_str(open.unwrap_or(f64::NAN), currency),
                    high: f64_to_money_with_currency_str(high.unwrap_or(f64::NAN), currency),
                    low: f64_to_money_with_currency_str(low.unwrap_or(f64::NAN), currency),
                    close: f64_to_money_with_currency_str(close.unwrap_or(f64::NAN), currency),
                    volume: volume0,
                });
                raw_close_vec.push(raw_close_val);
            }
        } else if let (Some(ov), Some(hv), Some(lv), Some(cv)) = (open, high, low, close) {
            out.push(Candle {
                ts: i64_to_datetime(t),
                open: f64_to_money_with_currency_str(ov, currency),
                high: f64_to_money_with_currency_str(hv, currency),
                low: f64_to_money_with_currency_str(lv, currency),
                close: f64_to_money_with_currency_str(cv, currency),
                volume: volume0,
            });
            raw_close_vec.push(raw_close_val);
        } else if keepna {
            out.push(Candle {
                ts: i64_to_datetime(t),
                open: f64_to_money_with_currency_str(open.unwrap_or(f64::NAN), currency),
                high: f64_to_money_with_currency_str(high.unwrap_or(f64::NAN), currency),
                low: f64_to_money_with_currency_str(low.unwrap_or(f64::NAN), currency),
                close: f64_to_money_with_currency_str(close.unwrap_or(f64::NAN), currency),
                volume: volume0,
            });
            raw_close_vec.push(raw_close_val);
        }
    }

    (out, raw_close_vec)
}