From 5e66b4bb954b2cc3650d50e60be86c328fcdc158 Mon Sep 17 00:00:00 2001
From: Administrator <admin@example.com>
Date: 星期二, 04 三月 2025 18:52:57 +0800
Subject: [PATCH] 板块拉黑重新定义

---
 third_data/code_plate_key_manager.py |  111 ++++++++++++++++++++++++++++++++++++++++++++++++++++++-
 1 files changed, 108 insertions(+), 3 deletions(-)

diff --git a/third_data/code_plate_key_manager.py b/third_data/code_plate_key_manager.py
index 1311472..c63c710 100644
--- a/third_data/code_plate_key_manager.py
+++ b/third_data/code_plate_key_manager.py
@@ -5,18 +5,21 @@
 # 娑ㄥ仠浠g爜鍏抽敭璇嶆澘鍧楃鐞�
 import copy
 import datetime
+import itertools
 import json
 import time
 
 import constant
+from code_attribute import gpcode_manager
 from db.redis_manager_delegate import RedisUtils
 from third_data import kpl_block_util, kpl_api, kpl_util
 from settings.trade_setting import MarketSituationManager
+from third_data.history_k_data_util import HistoryKDatasUtils
 from third_data.kpl_data_constant import LimitUpCodesBlockRecordManager, ContainsLimitupCodesBlocksManager
 from third_data.third_blocks_manager import BlockMapManager
 from utils import global_util, tool, buy_condition_util
 from log_module import async_log_util
-from db import redis_manager_delegate as redis_manager
+from db import redis_manager_delegate as redis_manager, mysql_data_delegate as mysql_data
 
 from log_module.log import logger_kpl_block_can_buy, logger_kpl_jx_out, logger_kpl_jx_in, logger_debug
 from third_data.kpl_util import KPLPlatManager
@@ -230,6 +233,11 @@
     # 宸茬粡鍒犻櫎浜嗙殑鏉垮潡
     __deleted_kpl_forbidden_plates_cache = set()
 
+    # 鐩戞帶鐨勯珮鏍囨澘鍧椾唬鐮佸瓧鍏革細{"鏉垮潡":{"浠g爜1","浠g爜2"}}
+    __watch_block_high_codes = {}
+    # 楂樻爣浠g爜
+    __watch_high_codes = set()
+
     __instance = None
 
     def __new__(cls, *args, **kwargs):
@@ -246,6 +254,7 @@
             cls.__deleted_kpl_forbidden_plates_cache = RedisUtils.smembers(__redis, "deleted_kpl_forbidden_plates")
         finally:
             RedisUtils.realse(__redis)
+        cls.__load_latest_gb()
 
     @classmethod
     def __get_redis(cls):
@@ -281,6 +290,103 @@
         if self.__kpl_forbidden_plates_cache and plate in self.__kpl_forbidden_plates_cache:
             return True
         return False
+
+    @classmethod
+    def __load_latest_gb(cls):
+        """
+        鍔犺浇鏈�杩戠殑甯傚満楂樻爣
+        @return:
+        """
+        # 鑾峰彇鏈�杩�10涓氦鏄撴棩娑ㄥ仠鐨勬定鍋滄暟鎹�
+        dates = HistoryKDatasUtils.get_latest_trading_date_cache(10)
+        if not dates:
+            return
+        min_date = dates[-1]
+        sql = f"SELECT r.`_code`, r.`_hot_block_name`, r.`_day`, r.`_open` FROM `kpl_limit_up_record` r WHERE r.`_day`>='{min_date}'"
+        mysqldb = mysql_data.Mysqldb()
+        results = mysqldb.select_all(sql)
+        code_days_map = {}
+        # 姣忕偢鏉�
+        f_code_days_map = {}
+        for r in results:
+            if r[0] not in code_days_map:
+                code_days_map[r[0]] = set()
+            code_days_map[r[0]].add(r[2])
+            if not r[3]:
+                if r[0] not in f_code_days_map:
+                    f_code_days_map[r[0]] = set()
+                f_code_days_map[r[0]].add(r[2])
+
+        # 杩囨护娑ㄥ仠娆℃暟>=3娆$殑鏁版嵁
+        target_codes = set()
+        for code in code_days_map:
+            if f_code_days_map.get(code) and (len(f_code_days_map.get(code)) >= 4 or (
+                    tool.is_ge_code(code) and len(f_code_days_map.get(code)) >= 2)):
+                # 涓旀湁3澶╁睘浜庤繛缁定鍋�
+                day_list = list(code_days_map[code])
+                day_list.sort(reverse=True)
+                step = 3
+                has_continue = False
+                for i in range(0, len(day_list) - step + 1):
+                    item_list = day_list[i:i + step]
+                    # 鏄惁灞炰簬杩炵画娑ㄥ仠
+                    is_sub = False
+                    for j in range(0, len(dates) - step):
+                        if f"{dates[j:j + step]}" == f"{item_list}":
+                            is_sub = True
+                            break
+                    if is_sub:
+                        has_continue = True
+                        break
+                if not has_continue:
+                    continue
+                target_codes.add(code)
+        # 浠g爜瀵瑰簲鐨勬澘鍧�
+        code_blocks = {}
+        for r in results:
+            if r[0] not in target_codes:
+                continue
+            if r[0] not in code_blocks:
+                code_blocks[r[0]] = set()
+            code_blocks[r[0]].add(kpl_util.filter_block(r[1]))
+        # 鎵�鏈夋澘鍧楀搴旂殑浠g爜闆嗗悎
+        block_codes = {}
+        for code in code_blocks:
+            for b in code_blocks[code]:
+                if b in constant.KPL_INVALID_BLOCKS:
+                    continue
+                if b not in block_codes:
+                    block_codes[b] = set()
+                block_codes[b].add(code)
+        print(block_codes)
+        cls.__watch_block_high_codes = block_codes
+        cls.__watch_high_codes.clear()
+        for b in block_codes:
+            cls.__watch_high_codes |= block_codes[b]
+
+        for k in block_codes:
+            print(k, [(x, gpcode_manager.get_code_name(x)) for x in block_codes[k]])
+
+    def get_watch_high_codes(self):
+        return self.__watch_high_codes
+
+    def compute(self, code_rate_dict: dict):
+        """
+        鏍规嵁姣斾緥璁$畻闇�瑕佹媺榛戠殑浠g爜
+        @param code_rate_dict: 娑ㄥ箙鐧惧垎鏁�
+        @return:
+        """
+        if self.__watch_block_high_codes:
+            forbidden_blocks = set()
+            for b in self.__watch_block_high_codes:
+                total_rate = 0
+                for code in self.__watch_block_high_codes[b]:
+                    if code in code_rate_dict:
+                        total_rate += code_rate_dict.get(code)
+                average_rate = total_rate / len(self.__watch_block_high_codes[b])
+                if average_rate < 1:
+                    forbidden_blocks.add(b)
+            self.__kpl_forbidden_plates_cache = forbidden_blocks
 
 
 class LimitUpCodesPlateKeyManager:
@@ -1230,5 +1336,4 @@
 
 
 if __name__ == "__main__":
-    RealTimeKplMarketData.set_market_strong(120)
-    print(RealTimeKplMarketData.get_jingxuan_in_block_threshold_count())
+    KPLPlateForbiddenManager().compute()

--
Gitblit v1.8.0