From 1c76abc59af35931b70b4742038dae0cfe4890d4 Mon Sep 17 00:00:00 2001
From: Administrator <admin@example.com>
Date: 星期二, 18 七月 2023 16:28:18 +0800
Subject: [PATCH] 华鑫适配

---
 code_attribute/code_nature_analyse.py |   37 +++++++++++++++++++++++++++++++++----
 1 files changed, 33 insertions(+), 4 deletions(-)

diff --git a/code_attribute/code_nature_analyse.py b/code_attribute/code_nature_analyse.py
index 7d022fd..77d0a4d 100644
--- a/code_attribute/code_nature_analyse.py
+++ b/code_attribute/code_nature_analyse.py
@@ -54,7 +54,7 @@
 
 
 # 鑾峰彇K绾垮舰鎬�
-# 杩斿洖 (15涓氦鏄撴棩娑ㄥ箙鏄惁澶т簬24.9%,鏄惁鐮村墠楂橈紝鏄惁瓒呰穼锛屾槸鍚︽帴杩戝墠楂橈紝鏄惁N,鏄惁V)
+# 杩斿洖 (15涓氦鏄撴棩娑ㄥ箙鏄惁澶т簬24.9%,鏄惁鐮村墠楂橈紝鏄惁瓒呰穼锛屾槸鍚︽帴杩戝墠楂橈紝鏄惁N,鏄惁V,鏄惁鏈夊舰鎬�,澶╅噺澶ч槼淇℃伅,鏄惁鍏锋湁杈ㄨ瘑搴�)
 def get_k_format(limit_up_price, record_datas):
     p1_data = get_lowest_price_rate(record_datas)
     p1 = p1_data[0] >= 0.249, p1_data[1]
@@ -71,7 +71,10 @@
 
     p7 = (p1[0] or p2[0] or p3[0] or p4[0] or p5[0] or p6[0], '')
 
-    return p1, p2, p3, p4, p5, p6, p7, p8
+    # 鏄惁鍏锋湁杈ㄨ瘑搴�
+    p9 = is_special(record_datas)
+
+    return p1, p2, p3, p4, p5, p6, p7, p8, p9
 
 
 # 鏄惁鍏锋湁K绾垮舰鎬�
@@ -114,9 +117,9 @@
     for i in range(len(datas)):
         item = datas[i]
         # 鑾峰彇棣栨澘娑ㄥ仠娆℃暟
-        if __is_limit_up(item) and i>0 and not __is_limit_up(datas[i-1]):
+        if __is_limit_up(item) and i > 0 and not __is_limit_up(datas[i - 1]):
             # 棣栨澘娑ㄥ仠
-            count+=1
+            count += 1
 
     return count
 
@@ -264,6 +267,12 @@
     return abs(limit_up_price - data["close"]) < 0.001
 
 
+# 鏄惁娑ㄥ仠杩�
+def __is_limited_up(data):
+    limit_up_price = float(gpcode_manager.get_limit_up_price_by_preprice(data["pre_close"]))
+    return abs(limit_up_price - data["high"]) < 0.001
+
+
 # 棣栨澘娑ㄥ仠婧环鐜�
 def get_limit_up_premium_rate(datas):
     datas = copy.deepcopy(datas)
@@ -311,5 +320,25 @@
     return count / len(first_rate_list)
 
 
+# 鏄惁鍏锋湁杈ㄨ瘑搴�
+def is_special(datas):
+    # 30涓氦鏄撴棩鍐呮湁鈮�5澶╂浘娑ㄥ仠涓旇繛缁定鍋滄暟鎴栨浘娑ㄥ仠鈮�2澶�
+    if len(datas) > 30:
+        datas = datas[-30:]
+
+    count = 0
+    continue_count = 0
+    last_index = -1
+    for i in range(len(datas)):
+        if __is_limited_up(datas[i]):
+            if last_index >= 0 and i - last_index == 1:
+                continue_count += 1
+            count += 1
+            last_index = i
+    if count >= 5 and continue_count > 0:
+        return True, ''
+    return False, ''
+
+
 if __name__ == "__main__":
     print(CodeNatureRecordManager.get_k_format("603717"))

--
Gitblit v1.8.0