From 21c96ed504f93f16ce6f8a3ccf164a87c9edd9c0 Mon Sep 17 00:00:00 2001
From: Administrator <admin@example.com>
Date: 星期二, 30 一月 2024 15:45:23 +0800
Subject: [PATCH] 增加调试日志

---
 third_data/code_plate_key_manager.py |   89 +++++++++++++++++++++++++++-----------------
 1 files changed, 54 insertions(+), 35 deletions(-)

diff --git a/third_data/code_plate_key_manager.py b/third_data/code_plate_key_manager.py
index c7e6f9d..3715995 100644
--- a/third_data/code_plate_key_manager.py
+++ b/third_data/code_plate_key_manager.py
@@ -8,9 +8,11 @@
 import time
 
 import constant
+from code_attribute import code_nature_analyse
 from db.redis_manager_delegate import RedisUtils
-from third_data import kpl_block_util, kpl_api
-from utils import global_util, tool
+from third_data import kpl_block_util, kpl_api, kpl_util
+from trade.trade_manager import MarketSituationManager
+from utils import global_util, tool, buy_condition_util
 from log_module import log, async_log_util
 from db import redis_manager_delegate as redis_manager
 
@@ -488,7 +490,7 @@
 
     # 杩斿洖鍐呭(鏄惁鍙拱, 鏄惁涓虹嫭鑻�, 鎻忚堪淇℃伅)
     @classmethod
-    def __is_block_can_buy(cls, code, block, current_limit_up_datas, code_limit_up_reason_dict,
+    def __is_block_can_buy(cls, code, block, current_limit_up_datas, code_limit_up_reasons_dict,
                            yesterday_current_limit_up_codes, limit_up_record_datas, current_limit_up_block_codes_dict,
                            high_level_code_blocks=None, high_level_block_codes=None):
         # 鐙嫍鍒ゆ柇
@@ -497,6 +499,8 @@
         if high_level_block_codes is None:
             high_level_block_codes = {}
         block_codes = current_limit_up_block_codes_dict.get(block)
+        if block_codes is None:
+            block_codes = set()
         if not block_codes:
             # 楂樹綅鏉挎硾鍖栨澘鍧椾腑鏃犳澘鍧�
             if not high_level_block_codes.get(block):
@@ -513,17 +517,22 @@
         # ---------------------------鍒ゆ柇寮哄娍涓荤嚎-------------------------
         is_strong_block = False
         for d in current_limit_up_datas:
-            if d[5] != block:
+            bs = kpl_util.get_current_limit_up_reasons(d)
+            if block not in bs:
                 general_blocks = high_level_code_blocks.get(d[0])
                 if not general_blocks or block not in general_blocks:
                     # 娌″湪娉涘寲鏉垮潡涓�
                     continue
-            if d[4].find("杩炴澘") > 0:
-                if d[4].replace("杩炴澘", "").isdigit():
-                    count = int(d[4].replace("杩炴澘", ""))
-                    if count >= 3:
-                        is_strong_block = True
-                        break
+            count = kpl_util.get_high_level_count(d[4])
+            if count >= 3:
+                if d[4].find("杩炴澘") > 0:
+                    is_strong_block = True
+                    break
+                elif d[0] in yesterday_current_limit_up_codes and len(block_codes) >= 2:
+                    # 鍑犲ぉ鍑犳澘锛屼笖鏈�杩�2杩炴澘
+                    # 鐪嬫槸鍚︽湁棣栨澘鍚庢帓
+                    is_strong_block = True
+                    break
 
         if not is_strong_block:
             temp_block_codes = set(copy.deepcopy(block_codes))
@@ -553,7 +562,7 @@
         # 鑾峰彇涓绘澘瀹炴椂韬綅,鍓旈櫎楂樹綅鏉�
         current_shsz_rank, front_current_shsz_rank_codes = kpl_block_util.get_code_current_rank(code, block,
                                                                                                 current_limit_up_datas,
-                                                                                                code_limit_up_reason_dict,
+                                                                                                code_limit_up_reasons_dict,
                                                                                                 yesterday_current_limit_up_codes,
                                                                                                 exclude_first_codes,
                                                                                                 len(
@@ -572,11 +581,17 @@
         if current_shsz_rank < len(current_open_limit_up_codes) + max_rank:
             return True, False, f"銆恵block}銆戝墠鎺掍唬鐮侊細{current_shsz_rank}", is_strong_block
         else:
+            # k_format = code_nature_analyse.CodeNatureRecordManager().get_k_format_cache(code)
+            # if k_format and k_format[8][0]:
+            #     # 鍏锋湁杈ㄨ瘑搴�
+            #     return True, False, f"銆恵block}銆戝叿鏈夎鲸璇嗗害", is_strong_block
             # 鐪嬭嚜鐢辨祦閫氬競鍊兼槸鍚﹀皬浜�20浜�
             if is_strong_block and current_shsz_rank < len(current_open_limit_up_codes) + max_rank + 1:
-                zyltgb = global_util.zyltgb_map.get(code)
-                if zyltgb and zyltgb < 20 * 100000000:
-                    return True, False, f"銆恵block}銆戝己鍔挎澘鍧� 鑷敱娴侀�氬競鍊�({zyltgb})灏忎簬20浜�", is_strong_block
+                zyltgb_as_yi = round(global_util.zyltgb_map.get(code) / 100000000, 2) if code in global_util.zyltgb_map else None
+                situation = MarketSituationManager().get_situation_cache()
+                zylt_threshold_as_yi = buy_condition_util.get_zyltgb_threshold(situation)
+                if zyltgb_as_yi and zylt_threshold_as_yi[2] <= zyltgb_as_yi <= zylt_threshold_as_yi[3]:
+                    return True, False, f"銆恵block}銆戝己鍔挎澘鍧� 鑷敱娴侀�氬競鍊�({zyltgb_as_yi})澶т簬{zylt_threshold_as_yi[2]}浜� 灏忎簬{zylt_threshold_as_yi[3]}浜�", is_strong_block
             return False, False, f"銆恵block}銆戝墠鎺掍唬鐮侊細{front_current_shsz_rank_codes} 瓒呰繃{len(current_open_limit_up_codes) + max_rank}涓�", is_strong_block
 
         # 杩囨椂鐨勪唬鐮�
@@ -615,10 +630,12 @@
             if limit_up_record_datas:
                 for d in limit_up_record_datas:
                     if d[2] in constant.KPL_INVALID_BLOCKS and d[3] in before_blocks_dict:
-                        code_limit_up_reason_dict[d[3]] = list(before_blocks_dict.get(d[3]))[0]
+                        code_limit_up_reasons_dict[d[3]] = {list(before_blocks_dict.get(d[3]))[0]}
                     else:
-                        code_limit_up_reason_dict[d[3]] = d[2]
-            return code_limit_up_reason_dict
+                        code_limit_up_reasons_dict[d[3]] = {d[2]}
+                        if d[6]:
+                            code_limit_up_reasons_dict[d[3]] |= set(d[6].split("銆�"))
+            return code_limit_up_reasons_dict
 
         if current_limit_up_datas is None:
             current_limit_up_datas = []
@@ -646,11 +663,11 @@
         fresults = []
         if not keys:
             return fresults, set()
-        code_limit_up_reason_dict = {}
+        code_limit_up_reasons_dict = {}
         load_code_block()
         for block in keys:
             can_buy, unique, msg, is_strong = cls.__is_block_can_buy(code, block, current_limit_up_datas,
-                                                                     code_limit_up_reason_dict,
+                                                                     code_limit_up_reasons_dict,
                                                                      yesterday_current_limit_up_codes,
                                                                      limit_up_record_datas,
                                                                      current_limit_up_block_codes_dict,
@@ -764,10 +781,11 @@
     # high_level_general_code_blocks 楂樹綅娉涘寲鏉垮潡
     @classmethod
     def update_can_buy_blocks(cls, code, current_limit_up_datas, limit_up_record_datas,
-                              yesterday_current_limit_up_records,
+                              latest_current_limit_up_records,
                               before_blocks_dict, current_limit_up_block_codes_dict):
         yesterday_current_limit_up_codes = set()
         yesterday_current_limit_up_records_dict = {}
+        yesterday_current_limit_up_records = latest_current_limit_up_records[0][1]
         if yesterday_current_limit_up_records:
             for r in yesterday_current_limit_up_records:
                 yesterday_current_limit_up_codes.add(r[0])
@@ -775,21 +793,22 @@
         high_level_general_code_blocks = {}
         # 鏄惁鏄�3鏉垮強浠ヤ笂鐨勯珮浣嶆澘
         for r in current_limit_up_datas:
-            if r[4].find("杩炴澘") > 0:
-                if r[4].replace("杩炴澘", "").isdigit():
-                    count = int(r[4].replace("杩炴澘", ""))
-                    if count >= 3:
-                        # 鏄珮浣嶆澘
-                        # 褰撴棩绮鹃��
-                        blocks = set(r[6].split("銆�"))
-                        if r[0] in yesterday_current_limit_up_records_dict:
-                            # 鏄ㄦ棩娑ㄥ仠鍘熷洜
-                            blocks.add(yesterday_current_limit_up_records_dict.get(r[0])[5])
-                        f_blocks = []
-                        for b in blocks:
-                            if b:
-                                f_blocks.append(b)
-                        high_level_general_code_blocks[r[0]] = f_blocks
+            count = kpl_util.get_high_level_count(r[4])
+            if count >= 3 and r[0] in yesterday_current_limit_up_codes:
+                latest_datas = latest_current_limit_up_records[:count - 1]
+                # 鏄珮浣嶆澘
+                # 褰撴棩绮鹃��
+                blocks = set(r[6].split("銆�"))
+                for d in latest_datas:
+                    for dd in d[1]:
+                        if dd[0] == r[0]:
+                            blocks.add(dd[5])
+                            break
+                f_blocks = []
+                for b in blocks:
+                    if b:
+                        f_blocks.append(b)
+                high_level_general_code_blocks[r[0]] = f_blocks
 
         can_buy_blocks, unique, msg, can_buy_strong_blocks, keys = cls.__compute_can_buy_blocks(code,
                                                                                                 current_limit_up_datas,

--
Gitblit v1.8.0