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