From 96dc1a4cc38b588f39387b5a85b9677100e357f1 Mon Sep 17 00:00:00 2001
From: Administrator <admin@example.com>
Date: 星期四, 23 三月 2023 23:55:40 +0800
Subject: [PATCH] 首板买入打分机制

---
 third_data/hot_block.py |   23 ++++++++++++++++-------
 1 files changed, 16 insertions(+), 7 deletions(-)

diff --git a/third_data/hot_block.py b/third_data/hot_block.py
index ce398b9..f2524f8 100644
--- a/third_data/hot_block.py
+++ b/third_data/hot_block.py
@@ -12,6 +12,15 @@
 
 
 def __parseData(driver):
+    date_element = \
+        driver.find_element(by=By.ID, value="nuxt-layout-container").find_elements(by=By.TAG_NAME, value="time")[0]
+    date_elements = date_element.find_elements(by=By.TAG_NAME, value="span")
+    month = date_elements[0].text.split(' ')[0][:-1]
+    day = date_elements[1].text
+    # 鑾峰彇褰撳墠鐨勫勾
+    year = datetime.datetime.now().strftime("%Y")
+    day_str = "{0}-{1:0>2}-{2:0>2}".format(year, int(month), int(day))
+
     items = driver.find_element(by=By.ID, value="nuxt-layout-container").find_element(by=By.CLASS_NAME,
                                                                                       value="topgainer-content-left").find_elements(
         by=By.CLASS_NAME,
@@ -44,7 +53,7 @@
         data_list.append((title, total_rate, codes_list))
         print("----------------------")
 
-    return data_list
+    return day_str, data_list
 
 
 # 鑾峰彇鐑棬鏉垮潡
@@ -66,17 +75,17 @@
         if int("113000") < int(time_str) < int("130000"):
             continue
         try:
-            result = __parseData(driver)
-            callback(result)
+            day, result = __parseData(driver)
+            callback(day, result)
         except Exception as e:
             logging.exception(e)
 
 
-def upload_data(datas):
+def upload_data(day, datas):
     client = socket.socket()  # 鐢熸垚socket锛岃繛鎺erver
     ip_port = ("192.168.3.252", 9001)  # server鍦板潃鍜岀鍙e彿锛堟渶濂芥槸10000浠ュ悗锛�
     client.connect(ip_port)
-    data = {"type": 70, "data": datas}
+    data = {"type": 70, "day": day, "data": datas}
     client.send(json.dumps(data).encode("gbk"))
     client.close()
 
@@ -85,8 +94,8 @@
 # cd D:\workspace\trade\third_data
 # C:\Users\Administrator\AppData\Roaming\Python\Python37\Scripts\pyinstaller.exe hot_block.spec
 if __name__ == "__main__":
-    def callback(result):
-        upload_data(result)
+    def callback(day, result):
+        upload_data(day, result)
         pass
 
 

--
Gitblit v1.8.0