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
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151

//! Tushare Pro API helpers.

use serde::Deserialize;

use crate::client::AkShareClient;
use crate::error::{Error, Result};
use crate::types::{CandlePoint, TradeCalendarItem};

#[derive(Debug, Deserialize)]
struct TushareResponse {
    code: Option<i64>,
    msg: Option<String>,
    data: Option<TushareData>,
}

#[derive(Debug, Deserialize)]
struct TushareData {
    #[serde(default)]
    fields: Vec<String>,
    #[serde(default)]
    items: Vec<Vec<serde_json::Value>>,
}

impl AkShareClient {
    /// Generic Tushare API call.
    async fn tushare_call(
        &self,
        api_name: &str,
        params: serde_json::Value,
        fields: &str,
    ) -> Result<TushareData> {
        let token = self
            .tushare_token
            .as_ref()
            .ok_or_else(|| Error::missing_credentials("tushare_token not set"))?;

        let body = serde_json::json!({
            "api_name": api_name,
            "token": token,
            "params": params,
            "fields": fields,
        });

        let resp: TushareResponse = self
            .post("https://api.tushare.pro")
            .json(&body)
            .send()
            .await?
            .json()
            .await?;

        if resp.code != Some(0) {
            let msg = resp
                .msg
                .unwrap_or_else(|| "unknown tushare error".to_string());
            return Err(Error::upstream(format!("tushare: {msg}")));
        }

        resp.data
            .ok_or_else(|| Error::upstream("tushare: empty response"))
    }

    /// Get daily candles from Tushare.
    pub(crate) async fn tushare_daily(
        &self,
        ts_code: &str,
        start: &str,
        end: &str,
    ) -> Result<Vec<CandlePoint>> {
        let params = serde_json::json!({
            "ts_code": ts_code,
            "start_date": start,
            "end_date": end,
        });
        let data = self
            .tushare_call(
                "daily",
                params,
                "trade_date,open,high,low,close,vol,amount,pct_chg,change,turnover_rate",
            )
            .await?;

        let fields = &data.fields;
        let mut items = Vec::with_capacity(data.items.len());
        for row in &data.items {
            let get = |name: &str| -> &serde_json::Value {
                fields
                    .iter()
                    .position(|f| f == name)
                    .and_then(|i| row.get(i))
                    .unwrap_or(&serde_json::Value::Null)
            };
            items.push(CandlePoint {
                trade_date: get("trade_date").as_str().unwrap_or("").to_string(),
                open: get("open").as_f64().unwrap_or(0.0),
                close: get("close").as_f64().unwrap_or(0.0),
                high: get("high").as_f64().unwrap_or(0.0),
                low: get("low").as_f64().unwrap_or(0.0),
                volume: get("vol").as_f64().unwrap_or(0.0) as i64,
                amount: get("amount").as_f64().unwrap_or(0.0),
                amplitude_pct: 0.0,
                change_pct: get("pct_chg").as_f64().unwrap_or(0.0),
                change_amount: get("change").as_f64().unwrap_or(0.0),
                turnover_pct: get("turnover_rate").as_f64().unwrap_or(0.0),
            });
        }
        Ok(items)
    }

    /// Trade calendar from Tushare.
    pub(crate) async fn tushare_trade_calendar(
        &self,
        exchange: &str,
        start: &str,
        end: &str,
    ) -> Result<Vec<TradeCalendarItem>> {
        let params = serde_json::json!({
            "exchange": exchange,
            "start_date": start,
            "end_date": end,
        });
        let data = self
            .tushare_call(
                "trade_cal",
                params,
                "exchange,cal_date,is_open,pretrade_date",
            )
            .await?;

        let fields = &data.fields;
        let mut items = Vec::with_capacity(data.items.len());
        for row in &data.items {
            let get = |name: &str| -> &serde_json::Value {
                fields
                    .iter()
                    .position(|f| f == name)
                    .and_then(|i| row.get(i))
                    .unwrap_or(&serde_json::Value::Null)
            };
            items.push(TradeCalendarItem {
                exchange: get("exchange").as_str().unwrap_or("").to_string(),
                calendar_date: get("cal_date").as_str().unwrap_or("").to_string(),
                is_open: get("is_open").as_i64().unwrap_or(0) == 1,
                previous_trade_date: get("pretrade_date")
                    .as_str()
                    .map(std::string::ToString::to_string),
            });
        }
        Ok(items)
    }
}