akshare 0.1.2

100% pure Rust implementation of akshare — unified access to Chinese and global financial market data APIs
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
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127

//! Yahoo Finance chart API helpers.

use serde::Deserialize;

use crate::client::AkShareClient;
use crate::error::{Error, Result};
use crate::types::CandlePoint;
use crate::util::{amplitude_pct, apply_change_metrics};

// ---------------------------------------------------------------------------
// Wire types
// ---------------------------------------------------------------------------

#[derive(Debug, Deserialize)]
struct YfChartResponse {
    chart: YfChart,
}

#[derive(Debug, Deserialize)]
struct YfChart {
    #[serde(default)]
    result: Vec<YfChartResult>,
}

#[derive(Debug, Deserialize)]
struct YfChartResult {
    timestamp: Option<Vec<i64>>,
    indicators: YfIndicators,
}

#[derive(Debug, Deserialize)]
struct YfIndicators {
    quote: Vec<YfQuote>,
}

#[derive(Debug, Deserialize)]
struct YfQuote {
    open: Vec<Option<f64>>,
    high: Vec<Option<f64>>,
    low: Vec<Option<f64>>,
    close: Vec<Option<f64>>,
    volume: Vec<Option<i64>>,
}

// ---------------------------------------------------------------------------
// Implementation
// ---------------------------------------------------------------------------

impl AkShareClient {
    /// Fetch candlestick data from Yahoo Finance chart API.
    pub(crate) async fn yahoo_candles(
        &self,
        symbol: &str,
        limit: usize,
    ) -> Result<Vec<CandlePoint>> {
        let (interval, range) = if limit <= 60 {
            ("1d", format!("{limit}d"))
        } else if limit <= 120 {
            ("1d", "6mo".to_string())
        } else {
            ("1d", "1y".to_string())
        };

        let url = format!(
            "https://query1.finance.yahoo.com/v8/finance/chart/{symbol}?interval={interval}&range={range}"
        );

        let resp: YfChartResponse = self
            .get(&url)
            .header(
                "User-Agent",
                "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36",
            )
            .send()
            .await?
            .json()
            .await?;

        let result = resp
            .chart
            .result
            .into_iter()
            .next()
            .ok_or_else(|| Error::upstream("empty yahoo chart response"))?;

        let timestamps = result.timestamp.unwrap_or_default();
        let quote = result
            .indicators
            .quote
            .into_iter()
            .next()
            .ok_or_else(|| Error::upstream("yahoo: no quote data"))?;

        let mut items = Vec::with_capacity(timestamps.len());
        for (i, ts) in timestamps.iter().enumerate() {
            let open = quote.open.get(i).and_then(|v| *v).unwrap_or(0.0);
            let high = quote.high.get(i).and_then(|v| *v).unwrap_or(0.0);
            let low = quote.low.get(i).and_then(|v| *v).unwrap_or(0.0);
            let close = quote.close.get(i).and_then(|v| *v).unwrap_or(0.0);
            let volume = quote.volume.get(i).and_then(|v| *v).unwrap_or(0);

            let date = chrono::DateTime::from_timestamp(*ts, 0)
                .map(|dt| dt.format("%Y-%m-%d").to_string())
                .unwrap_or_default();

            items.push(CandlePoint {
                trade_date: date,
                open,
                close,
                high,
                low,
                volume,
                amount: 0.0,
                amplitude_pct: amplitude_pct(high, low),
                change_pct: 0.0,
                change_amount: 0.0,
                turnover_pct: 0.0,
            });
        }
        apply_change_metrics(&mut items);

        if items.len() > limit {
            items = items[items.len() - limit..].to_vec();
        }
        Ok(items)
    }
}