krx-rs 0.1.0

KRX Open API를 위한 Rust 클라이언트
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

use super::{
    ApiResponse, deserialize_krx_date, deserialize_optional_f64, deserialize_optional_krx_date,
    deserialize_optional_u64,
};
use crate::error::Result;
use chrono::NaiveDate;
use polars::prelude::*;
use serde::Deserialize;

/// 사회책임투자채권 정보 레코드
#[derive(Debug, Deserialize)]
pub struct SriBondInfoRecord {
    /// 기준일자
    #[serde(rename = "BAS_DD", deserialize_with = "deserialize_krx_date")]
    pub base_date: NaiveDate,

    /// 발행기관
    #[serde(rename = "ISUR_NM")]
    pub issuer_name: String,

    /// 표준코드
    #[serde(rename = "ISU_CD")]
    pub issue_code: String,

    /// 채권종류
    #[serde(rename = "SRI_BND_TP_NM")]
    pub sri_bond_type: String,

    /// 종목명
    #[serde(rename = "ISU_NM")]
    pub issue_name: String,

    /// 상장일
    #[serde(rename = "LIST_DD", deserialize_with = "deserialize_optional_krx_date")]
    pub listing_date: Option<NaiveDate>,

    /// 발행일
    #[serde(rename = "ISU_DD", deserialize_with = "deserialize_optional_krx_date")]
    pub issue_date: Option<NaiveDate>,

    /// 상환일
    #[serde(
        rename = "REDMPT_DD",
        deserialize_with = "deserialize_optional_krx_date"
    )]
    pub redemption_date: Option<NaiveDate>,

    /// 표면이자율 (%)
    #[serde(rename = "ISU_RT", deserialize_with = "deserialize_optional_f64")]
    pub issue_rate: Option<f64>,

    /// 발행금액
    #[serde(rename = "ISU_AMT", deserialize_with = "deserialize_optional_u64")]
    pub issue_amount: Option<u64>,

    /// 상장금액
    #[serde(rename = "LIST_AMT", deserialize_with = "deserialize_optional_u64")]
    pub listing_amount: Option<u64>,

    /// 채권유형
    #[serde(rename = "BND_TP_NM")]
    pub bond_type: String,
}

/// 사회책임투자채권 정보를 DataFrame으로 변환
pub fn parse_sri_bond_info(response: ApiResponse<SriBondInfoRecord>) -> Result<DataFrame> {
    let records = response.data;

    if records.is_empty() {
        return Ok(DataFrame::empty());
    }

    let mut base_dates = Vec::with_capacity(records.len());
    let mut issuer_names = Vec::with_capacity(records.len());
    let mut issue_codes = Vec::with_capacity(records.len());
    let mut sri_bond_types = Vec::with_capacity(records.len());
    let mut issue_names = Vec::with_capacity(records.len());
    let mut listing_dates = Vec::with_capacity(records.len());
    let mut issue_dates = Vec::with_capacity(records.len());
    let mut redemption_dates = Vec::with_capacity(records.len());
    let mut issue_rates = Vec::with_capacity(records.len());
    let mut issue_amounts = Vec::with_capacity(records.len());
    let mut listing_amounts = Vec::with_capacity(records.len());
    let mut bond_types = Vec::with_capacity(records.len());

    for record in records {
        base_dates.push(record.base_date.format("%Y-%m-%d").to_string());
        issuer_names.push(record.issuer_name);
        issue_codes.push(record.issue_code);
        sri_bond_types.push(record.sri_bond_type);
        issue_names.push(record.issue_name);
        listing_dates.push(
            record
                .listing_date
                .map(|d| d.format("%Y-%m-%d").to_string()),
        );
        issue_dates.push(record.issue_date.map(|d| d.format("%Y-%m-%d").to_string()));
        redemption_dates.push(
            record
                .redemption_date
                .map(|d| d.format("%Y-%m-%d").to_string()),
        );
        issue_rates.push(record.issue_rate);
        issue_amounts.push(record.issue_amount.map(|v| v as i64));
        listing_amounts.push(record.listing_amount.map(|v| v as i64));
        bond_types.push(record.bond_type);
    }

    let df = df! {
        "기준일자" => base_dates,
        "발행기관" => issuer_names,
        "표준코드" => issue_codes,
        "채권종류" => sri_bond_types,
        "종목명" => issue_names,
        "상장일" => listing_dates,
        "발행일" => issue_dates,
        "상환일" => redemption_dates,
        "표면이자율" => issue_rates,
        "발행금액" => issue_amounts,
        "상장금액" => listing_amounts,
        "채권유형" => bond_types,
    }?;

    Ok(df)
}