From d957e0dc6dcfb15304a6499e1b9ddee4e18be531 Mon Sep 17 00:00:00 2001
From: Administrator <admin@example.com>
Date: 星期四, 02 一月 2025 18:10:23 +0800
Subject: [PATCH] 拆单初步规划/禁止交易板块管理

---
 third_data/custom_block_in_money_manager.py |   26 ++++++++++++++++++++++++++
 1 files changed, 26 insertions(+), 0 deletions(-)

diff --git a/third_data/custom_block_in_money_manager.py b/third_data/custom_block_in_money_manager.py
index d1e2ae0..0267992 100644
--- a/third_data/custom_block_in_money_manager.py
+++ b/third_data/custom_block_in_money_manager.py
@@ -18,6 +18,11 @@
     def __init__(self):
         # 鎬荤殑鍑�娴佸叆
         self.__code_money_dict = {}
+        # 鎬讳拱鍗曚俊鎭細{"code":[閲戦, 鏁伴噺]}
+        self.__code_buy_money_dict = {}
+        # 鎬诲崠鍗曚俊鎭細{"code":[閲戦, 鏁伴噺]}
+        self.__code_sell_money_dict = {}
+
         # 鍑�娴佸叆澶у崟閲戦
         self.__code_big_buy_mmoney_list_dict = {}
         self.__latest_price = {}
@@ -37,6 +42,13 @@
         code = item[0]
         if code not in self.__code_money_dict:
             self.__code_money_dict[code] = 0
+
+        if code not in self.__code_buy_money_dict:
+            self.__code_buy_money_dict[code] = [0, 0]
+
+        if code not in self.__code_sell_money_dict:
+            self.__code_sell_money_dict[code] = [0, 0]
+
         if not tool.is_ge_code(code) and item[2][2] < 299e4:
             return
         if tool.is_ge_code(code) and item[2][2] < 299e4 and item[2][1] < 290000:
@@ -44,6 +56,9 @@
         if item[1] == 0:
             # item[2]鐨勬暟鎹粨鏋勶細  (涔板崟鍙�, 閲�, 閲戦, 鏃堕棿, 鏈�鏂版垚浜や环鏍�)
             self.__code_money_dict[code] += item[2][2]
+            self.__code_buy_money_dict[code][0] += item[2][2]
+            self.__code_buy_money_dict[code][1] += 1
+
             if code not in self.__code_big_buy_mmoney_list_dict:
                 self.__code_big_buy_mmoney_list_dict[code] = []
             # 澶т拱鍗曚俊鎭細(閲戦,鏈�鍚庝环鏍�)
@@ -51,6 +66,8 @@
                 self.__code_big_buy_mmoney_list_dict[code].append((item[2][2], item[2][4], item[2][0]))
         else:
             self.__code_money_dict[code] -= item[2][2]
+            self.__code_sell_money_dict[code][0] += item[2][2]
+            self.__code_sell_money_dict[code][1] += 1
         self.__latest_price[code] = item[2][4]
 
     def get_code_money_dict(self):
@@ -61,6 +78,15 @@
             return self.__code_money_dict.get(code)
         return 0
 
+    def get_money_info(self, code):
+        """
+        鑾峰彇浠g爜娴佸叆淇℃伅
+        @param code: 浠g爜淇℃伅
+        @return: 鍑�娴佸叆閲戦,[澶у崟涔伴噾棰�, 澶у崟涔版暟閲廬,[澶у崟鍗栭噾棰�,澶у崟鍗栨暟閲廬
+        """
+        return self.__code_money_dict.get(code), self.__code_buy_money_dict.get(code), self.__code_sell_money_dict.get(
+            code)
+
     def set_money(self, code, money):
         self.__code_money_dict[code] = money
 

--
Gitblit v1.8.0