crypto-candlesticks-rust 0.1.5

Download candlestick data fast & easy for analysis
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

//! Print a table to the console.

use crate::exchanges::bitfinex::{CandleData, FloatOrInt};
use chrono::{TimeZone, Utc};
use comfy_table::{
    presets::ASCII_NO_BORDERS, presets::UTF8_FULL, Attribute, Cell, CellAlignment, Color,
    ContentArrangement, Table,
};

/// Table format to be displayed while data is downloaded.
pub fn setup_table() -> Table {
    let mut table: Table = Table::new();
    const TABLE_COLUMNS: [&str; 8] = [
        "Open",
        "High",
        "Low",
        "Close",
        "Volume",
        "Timestamp",
        "Ticker",
        "Time",
    ];
    table
        .load_preset(UTF8_FULL)
        .set_header(TABLE_COLUMNS.iter().map(|column| -> Cell {
            Cell::new(column)
                .add_attribute(Attribute::Bold)
                .fg(Color::Green)
                .set_alignment(CellAlignment::Center)
        }))
        .set_content_arrangement(ContentArrangement::DynamicFullWidth)
        .apply_modifier(ASCII_NO_BORDERS)
        .trim_fmt();
    table
}

/// Live data from the candles.
pub fn write_to_column(
    ticker: &str,
    interval: &str,
    data_downloaded: Vec<CandleData>,
    mut table: Table,
) {
    data_downloaded
        .into_iter()
        .for_each(|candle_data: CandleData| {
            candle_data.0.into_iter().for_each(|single_candle_info| {
                let datetime: String = Utc
                    .timestamp_millis(
                        single_candle_info[0]
                            .to_string()
                            .parse()
                            .expect("Datetime could not be converted from timestamp"),
                    )
                    .to_string();
                insert_rows_to_table(single_candle_info, &mut table, ticker, interval, datetime);
            });
        });
    // clear the console
    print!("{esc}c", esc = 27 as char);
    println!("{}", table);
}

fn insert_rows_to_table(
    single_candle_info: [FloatOrInt; 6],
    table: &mut Table,
    ticker: &str,
    interval: &str,
    datetime: String,
) {
    table.add_row(vec![
        Cell::new(single_candle_info[2]).set_alignment(CellAlignment::Center),
        Cell::new(single_candle_info[1]).set_alignment(CellAlignment::Center),
        Cell::new(single_candle_info[3]).set_alignment(CellAlignment::Center),
        Cell::new(single_candle_info[4]).set_alignment(CellAlignment::Center),
        Cell::new(single_candle_info[5]).set_alignment(CellAlignment::Center),
        Cell::new(ticker).set_alignment(CellAlignment::Center),
        Cell::new(interval).set_alignment(CellAlignment::Center),
        Cell::new(datetime).set_alignment(CellAlignment::Center),
    ]);
}