From 68464c679ae5e1ae35e7e67e3b339ba0f939cbd3 Mon Sep 17 00:00:00 2001
From: Administrator <admin@example.com>
Date: 星期三, 15 三月 2023 14:46:03 +0800
Subject: [PATCH] 选股宝板块优化
---
third_data/hot_block_data_process.py | 9 ++++++---
1 files changed, 6 insertions(+), 3 deletions(-)
diff --git a/third_data/hot_block_data_process.py b/third_data/hot_block_data_process.py
index 5f15742..b7af45a 100644
--- a/third_data/hot_block_data_process.py
+++ b/third_data/hot_block_data_process.py
@@ -20,12 +20,12 @@
for block in datas:
codes = []
for code_data in block[2]:
- code = code_data[0]
+ code = code_data[0].split(".")[0]
if code not in code_block_dict:
code_block_dict[code] = set()
code_block_dict[code].add(block[0])
codes.append(code)
- block_codes_dict[block] = codes
+ block_codes_dict[block[0]] = codes
__save_block_codes(block_codes_dict)
for key in code_block_dict:
__save_code_block(key, code_block_dict[key])
@@ -33,7 +33,7 @@
# 淇濆瓨浠ｇ爜鎵�灞炴澘鍧�
def __save_code_block(code, blocks):
- __get_redis().setex(f"code_blocks-{code}", tool.get_expire(), json.dumps(blocks))
+ __get_redis().setex(f"code_blocks-{code}", tool.get_expire(), json.dumps(list(blocks)))
# 淇濆瓨鏉垮潡涓嬬殑浠ｇ爜
@@ -62,3 +62,6 @@
if block_codes:
block_codes.get(block)
return None
+
+if __name__ == "__main__":
+ print(get_code_blocks("600468"))
\ No newline at end of file
--
Gitblit v1.8.0