From 86e0061f9cf211b98252a9e6b71d6c9801e4a16b Mon Sep 17 00:00:00 2001
From: Administrator <admin@example.com>
Date: 星期日, 18 十二月 2022 18:08:03 +0800
Subject: [PATCH] 撤单策略再次修改

---
 gpcode_manager.py |   21 ++++++++++++++++++++-
 1 files changed, 20 insertions(+), 1 deletions(-)

diff --git a/gpcode_manager.py b/gpcode_manager.py
index 557efd1..93aaefa 100644
--- a/gpcode_manager.py
+++ b/gpcode_manager.py
@@ -14,7 +14,7 @@
 __redisManager = redis_manager.RedisManager(0)
 
 
-def set_gp_list(code_datas):
+def __parse_codes_data(code_datas):
     codes = []
     name_codes = {}
     for _data in code_datas:
@@ -26,7 +26,11 @@
                 codes.append(code)
                 # 淇濆瓨浠g爜瀵瑰簲鐨勫悕绉�
                 name_codes[name] = code
+    return codes, name_codes
 
+
+def set_gp_list(code_datas):
+    codes, name_codes = __parse_codes_data(code_datas)
     redis_instance = __redisManager.getRedis()
     # 鍒犻櫎涔嬪墠鐨�
     redis_instance.delete("gp_list")
@@ -36,6 +40,21 @@
     redis_instance.set("gp_list_names", json.dumps(name_codes))
 
 
+# 鏂板浠g爜
+def add_gp_list(code_datas):
+    redis_instance = __redisManager.getRedis()
+    codes, name_codes = __parse_codes_data(code_datas)
+    for d in codes:
+        redis_instance.sadd("gp_list", d)
+    old_name_codes = get_name_codes()
+    if old_name_codes is None:
+        old_name_codes = name_codes
+    else:
+        for key in name_codes:
+            old_name_codes[key] = name_codes[key]
+    redis_instance.set("gp_list_names", json.dumps(old_name_codes))
+
+
 # 鑾峰彇鍚嶇О瀵瑰簲鐨勪唬鐮�
 def get_name_code(name):
     redis_instance = __redisManager.getRedis()

--
Gitblit v1.8.0