From 65afea1ba534b51f947cbe7989d7f4d650bbc9e6 Mon Sep 17 00:00:00 2001
From: Administrator <admin@example.com>
Date: 星期一, 04 十一月 2024 18:29:00 +0800
Subject: [PATCH] 代码异常保护

---
 l2/l2_transaction_data_manager.py |  207 +++++++++++++++++++++++++++++++++++++++++++--------
 1 files changed, 175 insertions(+), 32 deletions(-)

diff --git a/l2/l2_transaction_data_manager.py b/l2/l2_transaction_data_manager.py
index 9089d03..3a84c7e 100644
--- a/l2/l2_transaction_data_manager.py
+++ b/l2/l2_transaction_data_manager.py
@@ -3,6 +3,7 @@
 """
 import json
 
+import l2_data_util
 from db import redis_manager
 from db.redis_manager_delegate import RedisUtils
 from l2 import l2_log
@@ -19,7 +20,7 @@
 
 class BigOrderDealManager:
     """
-    澶у崟鎴愪氦绠$悊
+    鎴愪氦澶у崟绠$悊
     """
     __total_buy_datas_dict = {}
     __total_sell_datas_dict = {}
@@ -48,7 +49,7 @@
         """
         娣诲姞澶т拱鍗曟暟鎹�
         @param code: 浠g爜
-        @param datas:[(涔板崟鍙�,鎬昏偂鏁�,鎬绘垚浜ら)]
+        @param datas:[(涔板崟鍙�,鎬昏偂鏁�,鎬绘垚浜ら,鎴愪氦寮�濮嬫椂闂�,鎴愪氦缁撴潫鏃堕棿)]
         @return:
         """
         if not datas:
@@ -83,6 +84,16 @@
             return 0
         return int(sum([x[2] for x in self.__total_buy_datas_dict[code]]))
 
+    def get_total_buy_count(self, code):
+        """
+        鑾峰彇澶у崟鎴愪氦鐨勭瑪鏁�
+        @param code:
+        @return:
+        """
+        if code not in self.__total_buy_datas_dict:
+            return 0
+        return len(self.__total_buy_datas_dict[code])
+
     def get_total_buy_money_list(self, code):
         """
         鑾峰彇澶у崟鍒楄〃
@@ -90,9 +101,16 @@
         @return:
         """
         if code not in self.__total_buy_datas_dict:
-            return 0
+            return []
         return [x[2] for x in self.__total_buy_datas_dict[code]]
 
+    def get_total_buy_data_list(self, code):
+        """
+        鑾峰彇鎵�鏈夌殑涔板崟鏁版嵁鍒楄〃
+        @param code:
+        @return:[(涔板崟鍙�,鎬昏偂鏁�,鎬绘垚浜ら,鎴愪氦寮�濮嬫椂闂�,鎴愪氦缁撴潫鏃堕棿)]
+        """
+        return self.__total_buy_datas_dict.get(code)
 
     def get_total_sell_money(self, code):
         """
@@ -113,6 +131,9 @@
 
     # 姝e湪鎴愪氦鐨勮鍗�
     __dealing_order_info_dict = {}
+
+    # 姝e湪鎴愪氦鐨勪富鍔ㄤ拱鐨勮鍗�
+    __dealing_active_buy_order_info_dict = {}
     # 鏈�杩戞垚浜ょ殑璁㈠崟{"code":(璁㈠崟鍙凤紝鏄惁鎴愪氦瀹屾垚)}
     __latest_deal_order_info_dict = {}
 
@@ -155,7 +176,16 @@
         return cls.__dealing_order_info_dict.get(code)
 
     @classmethod
-    def statistic_big_buy_data(cls, code, datas):
+    def get_dealing_active_order_info(cls, code):
+        """
+        鑾峰彇姝e湪涓诲姩鎴愪氦鐨勬暟鎹�
+        @param code:
+        @return:[璁㈠崟鍙凤紝鎬昏偂鏁帮紝鎴愪氦閲戦,鎴愪氦寮�濮嬫椂闂�,鎴愪氦缁撴潫鏃堕棿]
+        """
+        return cls.__dealing_active_buy_order_info_dict.get(code)
+
+    @classmethod
+    def statistic_big_buy_data(cls, code, datas, limit_up_price):
         """
         缁熻澶у崟涔�
         @param code:
@@ -164,6 +194,8 @@
         """
         big_buy_datas = []
         normal_buy_datas = []
+        # 澶у崟闃堝��
+        threshold_big_money = l2_data_util.get_big_money_val(limit_up_price, tool.is_ge_code(code))
         for data in datas:
             # q.append((data['SecurityID'], data['TradePrice'], data['TradeVolume'],
             #                   data['OrderTime'], data['MainSeq'], data['SubSeq'], data['BuyNo'],
@@ -172,25 +204,46 @@
             if code not in cls.__dealing_order_info_dict:
                 # 鏁版嵁鏍煎紡[璁㈠崟鍙凤紝鎬昏偂鏁帮紝鎴愪氦閲戦,鎴愪氦寮�濮嬫椂闂�,鎴愪氦缁撴潫鏃堕棿]
                 cls.__dealing_order_info_dict[code] = [data[6], data[2], data[2] * data[1], data[3], data[3]]
-            if cls.__dealing_order_info_dict[code][0] == data[6]:
-                # 鎴愪氦鍚屼竴涓鍗曞彿
-                cls.__dealing_order_info_dict[code][1] += data[2]
-                cls.__dealing_order_info_dict[code][2] += data[2] * data[1]
-                cls.__dealing_order_info_dict[code][4] = data[3]
             else:
-                # 淇濆瓨涓婁竴鏉℃暟鎹�
-                async_log_util.info(hx_logger_l2_transaction_desc, f"{code}#{cls.__dealing_order_info_dict[code]}")
-                # 璁剧疆鏈�杩戞垚浜ゅ畬鎴愮殑涓�鏉℃暟鎹�
-                deal_info = cls.__dealing_order_info_dict[code]
-                cls.__latest_deal_order_info_dict[code] = deal_info
-                # 鏄惁涓哄ぇ涔板崟
-                if deal_info[2] >= 2990000:
-                    big_buy_datas.append(deal_info)
-                if deal_info[2] >= 500000:
-                    normal_buy_datas.append(deal_info)
+                if cls.__dealing_order_info_dict[code][0] == data[6]:
+                    # 鎴愪氦鍚屼竴涓鍗曞彿
+                    cls.__dealing_order_info_dict[code][1] += data[2]
+                    cls.__dealing_order_info_dict[code][2] += data[2] * data[1]
+                    cls.__dealing_order_info_dict[code][4] = data[3]
+                else:
+                    # 淇濆瓨涓婁竴鏉℃暟鎹�
+                    async_log_util.info(hx_logger_l2_transaction_desc, f"{code}#{cls.__dealing_order_info_dict[code]}")
+                    # 璁剧疆鏈�杩戞垚浜ゅ畬鎴愮殑涓�鏉℃暟鎹�
+                    deal_info = cls.__dealing_order_info_dict[code]
+                    cls.__latest_deal_order_info_dict[code] = deal_info
+                    # 鏄惁涓哄ぇ涔板崟
+                    if deal_info[2] >= threshold_big_money:
+                        big_buy_datas.append(deal_info)
+                    if deal_info[2] >= 500000:
+                        normal_buy_datas.append(deal_info)
 
-                # 鍒濆鍖栨湰鏉℃暟鎹�
-                cls.__dealing_order_info_dict[code] = [data[6], data[2], data[2] * data[1], data[3], data[3]]
+                    # 鍒濆鍖栨湰鏉℃暟鎹�
+                    cls.__dealing_order_info_dict[code] = [data[6], data[2], data[2] * data[1], data[3], data[3]]
+            # 缁熻涓诲姩涔�(涔板崟鍙峰ぇ浜庡崠鍗曞彿)
+            try:
+                if data[6] > data[7]:
+                    if code not in cls.__dealing_active_buy_order_info_dict:
+                        # 鏁版嵁鏍煎紡[璁㈠崟鍙凤紝鎬昏偂鏁帮紝鎴愪氦閲戦,鎴愪氦寮�濮嬫椂闂�,鎴愪氦缁撴潫鏃堕棿]
+                        cls.__dealing_active_buy_order_info_dict[code] = [data[6], data[2], data[2] * data[1], data[3],
+                                                                          data[3]]
+                    else:
+                        if cls.__dealing_active_buy_order_info_dict[code][0] == data[6]:
+                            # 鎴愪氦鍚屼竴涓鍗曞彿
+                            cls.__dealing_active_buy_order_info_dict[code][1] += data[2]
+                            cls.__dealing_active_buy_order_info_dict[code][2] += data[2] * data[1]
+                            cls.__dealing_active_buy_order_info_dict[code][4] = data[3]
+                        else:
+                            # 鍒濆鍖栨湰鏉℃暟鎹�
+                            cls.__dealing_active_buy_order_info_dict[code] = [data[6], data[2], data[2] * data[1],
+                                                                              data[3], data[3]]
+            except:
+                pass
+
         return big_buy_datas, normal_buy_datas
 
 
@@ -255,12 +308,15 @@
 
     # 缁熻鎵�鏈夌殑鎴愪氦閲�
     __deal_volume_list_dict = {}
+    # 缁熻娑ㄥ仠涓诲姩涔扮殑鎴愪氦閲�
+    __deal_active_buy_volume_list_dict = {}
 
     @classmethod
-    def statistic_total_deal_volume(cls, code, datas):
+    def statistic_total_deal_volume(cls, code, datas, limit_up_price):
         # 鍙粺璁¤鍔ㄤ拱
         if code not in cls.__deal_volume_list_dict:
             cls.__deal_volume_list_dict[code] = []
+
         time_dict = {}
         for d in datas:
             # 鍙粺璁¤鍔ㄤ拱
@@ -284,24 +340,96 @@
         # 鍒犻櫎瓒呰繃5鏉℃暟鎹�
         if len(cls.__deal_volume_list_dict[code]) > 5:
             cls.__deal_volume_list_dict[code] = cls.__deal_volume_list_dict[code][-5:]
+
+        try:
+            # 缁熻涓诲姩涔扮殑鎴愪氦閲�
+            if code not in cls.__deal_active_buy_volume_list_dict:
+                cls.__deal_active_buy_volume_list_dict[code] = []
+            for d in datas:
+                # 鍙粺璁′富鍔ㄤ拱
+                if d[7] > d[6]:
+                    continue
+                # 鍙粺璁℃定鍋滀拱
+                if d[1] != limit_up_price:
+                    continue
+                if d[3] in time_dict:
+                    time_str = time_dict[d[3]]
+                else:
+                    time_dict[d[3]] = l2_huaxin_util.convert_time(d[3])
+                    time_str = time_dict[d[3]]
+                if cls.__deal_active_buy_volume_list_dict[code]:
+                    if cls.__deal_active_buy_volume_list_dict[code][-1][0] == time_str:
+                        # 濡傛灉鏄悓涓�绉�
+                        cls.__deal_active_buy_volume_list_dict[code][-1][1] += d[2]
+                    else:
+                        # 涓嶆槸鍚屼竴绉�
+                        cls.__deal_active_buy_volume_list_dict[code].append([time_str, d[2]])
+                else:
+                    cls.__deal_active_buy_volume_list_dict[code].append([time_str, d[2]])
+            # 鍒犻櫎瓒呰繃10鏉℃暟鎹�
+            if len(cls.__deal_active_buy_volume_list_dict[code]) > 10:
+                cls.__deal_active_buy_volume_list_dict[code] = cls.__deal_active_buy_volume_list_dict[code][-10:]
+        except:
+            pass
+
         time_dict.clear()
 
     @classmethod
-    def get_latest_2s_continue_deal_volume(cls, code):
+    def get_latest_3s_continue_deal_volumes(cls, code):
         """
-        鑾峰彇鏈�杩戣繛缁�2s鐨勬垚浜ら噺
+        鑾峰彇鏈�杩�3s鐨勬垚浜ら噺鍒嗗竷
         @param code:
-        @return: 鎴愪氦閲�,璇︾粏淇℃伅
+        @return: [(鏃堕棿,閲�)]
         """
         deal_list = cls.__deal_volume_list_dict.get(code)
         if not deal_list:
-            return 0, None
-        if len(deal_list) == 0:
-            return deal_list[0][1], deal_list
-        if tool.trade_time_sub(deal_list[-1][0], deal_list[-2][0]) > 1:
-            return deal_list[-1][1], deal_list
-        else:
-            return deal_list[-1][1] + deal_list[-2][1], deal_list
+            return []
+        fdatas = [deal_list[-1]]
+        # 浠庡�掓暟绗簩涓暟鎹绠�
+        for i in range(len(deal_list) - 2, -1, -1):
+            if tool.trade_time_sub(fdatas[0][0], deal_list[i][0]) < 3:
+                fdatas.append(deal_list[i])
+        return fdatas
+
+    @classmethod
+    def get_latest_2s_continue_deal_volumes(cls, code):
+        """
+        鑾峰彇鏈�杩�2s鐨勬垚浜ら噺鍒嗗竷
+        @param code:
+        @return: [(鏃堕棿,閲�)]
+        """
+        deal_list = cls.__deal_volume_list_dict.get(code)
+        if not deal_list:
+            return []
+        fdatas = [deal_list[-1]]
+        # 浠庡�掓暟绗簩涓暟鎹绠�
+        for i in range(len(deal_list) - 1, -1, -1):
+            if tool.trade_time_sub(fdatas[0][0], deal_list[i][0]) < 2:
+                fdatas.append(deal_list[i])
+        return fdatas
+
+    @classmethod
+    def get_latest_6s_active_buy_deal_volumes(cls, code):
+        """
+        鑾峰彇鏈�杩�6s鐨勪富鍔ㄤ拱鎴愪氦
+        @param code:
+        @return: [(鏃堕棿,閲�)]
+        """
+        deal_list = cls.__deal_active_buy_volume_list_dict.get(code)
+        if not deal_list:
+            return []
+        latest_time = deal_list[-1][0]
+        fdatas = []
+        # 浠庡�掓暟绗簩涓暟鎹绠�
+        for i in range(len(deal_list) - 1, -1, -1):
+            if tool.trade_time_sub(latest_time, deal_list[i][0]) < 6:
+                fdatas.append(deal_list[i])
+        return fdatas
+
+    @classmethod
+    def clear_latest_deal_volume(cls, code):
+        if code in cls.__deal_volume_list_dict:
+            cls.__deal_volume_list_dict.pop(code)
 
     # 杩斿洖鏈�杩�1s鐨勫ぇ鍗曞崠锛�(鎬诲崠閲戦,[(鍗栧崟鍙�,鎬绘墜鏁�,浠锋牸,锛�'寮�濮嬫椂闂�',涔板崟鍙凤級,锛�'缁撴潫鏃堕棿',涔板崟鍙凤級),...])
     @classmethod
@@ -435,6 +563,21 @@
                     total_sell_info[0] += int(latest_sell_order_info[1] * latest_sell_order_info[2])
             big_sell_orders.reverse()
             total_sell_info[1] = big_sell_orders
+        # ----------------缁熻娑ㄥ仠涓诲姩涔�-----------------
+        try:
+            limit_up_active_buy_datas = []
+            for d in datas:
+                if is_active_sell(d[7], d[6]):
+                    # 琚姩涔�
+                    continue
+                # 鏄惁鏄定鍋�
+                if d[1] == limit_up_price:
+                    # 鏈夋定鍋滀富鍔ㄤ拱
+                    limit_up_active_buy_datas.append(d)
+            L2TradeSingleDataManager.set_limit_up_active_buy(code, limit_up_active_buy_datas)
+        except:
+            pass
+
         return total_sell_info
 
     # 鑾峰彇鏈�杩戞垚浜ゆ暟鎹�

--
Gitblit v1.8.0