From 751183dcd74207a50834cacc575f0dfccb41658c Mon Sep 17 00:00:00 2001
From: Administrator <admin@example.com>
Date: 星期二, 06 六月 2023 18:13:05 +0800
Subject: [PATCH] 交易优化,看盘接口完善

---
 third_data/kpl_util.py |   33 +++++++++++++++++++++++++++++++++
 1 files changed, 33 insertions(+), 0 deletions(-)

diff --git a/third_data/kpl_util.py b/third_data/kpl_util.py
index eec6a68..b4b95eb 100644
--- a/third_data/kpl_util.py
+++ b/third_data/kpl_util.py
@@ -1,6 +1,8 @@
 import enum
 import json
 
+from db import mysql_data
+
 
 def parse_kpl_datas(results):
     start_y = -1
@@ -233,3 +235,34 @@
         # (浠g爜,鍚嶇О,寮哄害,涓诲姏鍑�棰�)
         fresult_.append((d[0], d[1], d[2], d[6]))
     return fresult_
+
+
+class KPLPlatManager:
+    def save_plat(self, _id, name):
+        if not _id:
+            return
+        mysqldb = mysql_data.Mysqldb()
+        key = f"{_id}-{name}"
+        results = mysqldb.select_one(f"select * from kpl_plate where _name='{name}'")
+        if not results:
+            mysqldb.execute(f"insert into kpl_plate(_id,_name,_key) values({_id},'{name}','{key}')")
+
+    def get_plat(self, name):
+        mysqldb = mysql_data.Mysqldb()
+        results = mysqldb.select_one(f"select * from kpl_plate where _name='{name}'")
+        if results:
+            return results[0]
+        return None
+
+    def get_same_plat_names(self, name):
+        mysqldb = mysql_data.Mysqldb()
+        plate = self.get_plat(name)
+        if not plate:
+            return {name}
+        results = mysqldb.select_all(f"select _name from kpl_plate where _id='{plate}'")
+        return set([r[0] for r in results])
+
+    def get_same_plat_names_by_id(self, id_):
+        mysqldb = mysql_data.Mysqldb()
+        results = mysqldb.select_all(f"select _name from kpl_plate where _id='{id_}'")
+        return set([r[0] for r in results])

--
Gitblit v1.8.0