stock-calc 1.0.1

股票收益计算器 - 命令行工具
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

use serde::{Deserialize, Serialize};
use chrono::{DateTime, Utc};

#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct StockInfo {
    pub name: String,
    pub code: String,
    pub current_price: f64,
    pub yesterday_close: f64,
    pub open_price: f64,
    pub volume: u64,
    pub turnover: f64,
    pub high_price: f64,
    pub low_price: f64,
    pub change_amount: f64,
    pub change_percent: f64,
}

#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct StockData {
    pub code: String,
    pub quantity: f64,
    pub avg_price: f64,
    pub target_profit: f64,
    pub max_loss: f64,
    pub last_updated: DateTime<Utc>,
}

#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct StockAnalysis {
    pub code: String,
    pub current_price: f64,
    pub quantity: f64,
    pub avg_price: f64,
    pub target_profit: f64,
    pub max_loss: f64,
    pub target_price: f64,
    pub stop_loss_price: f64,
    pub current_profit: f64,
    pub profit_ratio: f64,
    pub distance_to_target: f64,
    pub distance_to_stop_loss: f64,
    pub risk_reward_ratio: f64,
    pub investment_amount: f64,
    pub investment_scale: InvestmentScale,
    pub status: StockStatus,
    pub last_updated: DateTime<Utc>,
}

#[derive(Debug, Clone, Serialize, Deserialize)]
pub enum InvestmentScale {
    Small,      // < 1万
    Medium,     // 1-10万
    Large,      // > 10万
}

#[derive(Debug, Clone, Serialize, Deserialize)]
pub enum StockStatus {
    Profitable,     // 盈利
    Loss,           // 亏损
    AtTarget,       // 达到目标
    AtStopLoss,     // 达到止损
    NearTarget,     // 接近目标
    NearStopLoss,   // 接近止损
}

#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct StockDatabase {
    pub stocks: std::collections::HashMap<String, StockData>,
}

impl StockDatabase {
    pub fn new() -> Self {
        Self {
            stocks: std::collections::HashMap::new(),
        }
    }

    pub fn add_stock(&mut self, stock: StockData) {
        self.stocks.insert(stock.code.clone(), stock);
    }

    pub fn remove_stock(&mut self, code: &str) -> Option<StockData> {
        self.stocks.remove(code)
    }

    pub fn get_stock(&self, code: &str) -> Option<&StockData> {
        self.stocks.get(code)
    }

    pub fn list_stocks(&self) -> Vec<&StockData> {
        self.stocks.values().collect()
    }
}

impl Default for StockDatabase {
    fn default() -> Self {
        Self::new()
    }
}