import json
|
import os
|
|
import constant
|
from utils import tool
|
from utils.tool import singleton
|
|
|
@singleton
|
class KPLLimitUpDataManager:
|
__yesterday_limit_up_codes = set()
|
|
@classmethod
|
# 获取最近几天的数据，根据日期倒序返回
|
def __get_latest_from_file(cls, type, count, max_day=tool.get_now_date_str()):
|
files = os.listdir(constant.CACHE_PATH)
|
file_name_list = []
|
for f in files:
|
if f[10:] == f"_{type}.log":
|
file_name_list.append((f.split("_")[0], f))
|
|
file_name_list.sort(key=lambda x: x[0], reverse=True)
|
fresults = []
|
for file in file_name_list:
|
path = f"{constant.CACHE_PATH}/{file[1]}"
|
if not os.path.exists(path):
|
continue
|
with open(path, 'r') as f:
|
lines = f.readlines()
|
if lines:
|
if int(file[0].replace("-", "")) <= int(max_day.replace("-", "")):
|
fresults.append((file[0], json.loads(lines[0])))
|
if len(fresults) >= count:
|
break
|
return fresults
|
|
@classmethod
|
def __get_current_limit_up_data_records(cls, count, day=tool.get_now_date_str()):
|
fresults = []
|
datas = cls.__get_latest_from_file("limit_up", count + 2, max_day=day)
|
# 移除比今天还大的数据
|
fdatas = []
|
for d in datas:
|
if int(d[0].replace("-", "")) > int(day.replace("-", "")):
|
continue
|
fdatas.append(d)
|
datas = fdatas
|
for i in range(len(datas)):
|
if datas[i][0] == day:
|
continue
|
fresults.append(datas[i])
|
if len(fresults) >= count:
|
break
|
return fresults
|
|
@classmethod
|
def __get_yesterday_limit_up_codes(cls):
|
yesterday_limit_up_data_records = cls.__get_yesterday_current_limit_up_records()
|
yesterday_codes = set([x[0] for x in yesterday_limit_up_data_records])
|
return yesterday_codes
|
|
@classmethod
|
def __get_yesterday_current_limit_up_records(cls):
|
yesterday_limit_up_data_records = cls.__get_current_limit_up_data_records(1)[0][1]
|
return yesterday_limit_up_data_records
|
|
def __init__(self):
|
# 加载昨日涨停
|
self.__yesterday_limit_up_codes = self.__get_yesterday_limit_up_codes()
|
|
def get_yesterday_limit_up_codes(self):
|
return self.__yesterday_limit_up_codes
|
|
|
if __name__ == '__main__':
|
pass
|
# print(KPLLimitUpDataManager().get_yesterday_limit_up_codes())
|
