"""
|
安全笔数管理
|
"""
|
# 下单L2的安全笔数管理
|
import json
|
|
from l2 import l2_data_source_util
|
from trade import l2_trade_factor
|
from db import redis_manager
|
import tool
|
from l2.l2_data_util import L2DataUtil
|
import l2_data_util
|
|
|
class BuyL2SafeCountManager(object):
|
__redis_manager = redis_manager.RedisManager(0)
|
|
def __init__(self):
|
self.last_buy_queue_data = {}
|
|
def __getRedis(self):
|
return self.__redis_manager.getRedis()
|
|
# 记录每一次的处理进度
|
def __save_compute_progress(self, code, last_buy_single_index, process_index, buy_num, cancel_num):
|
key = "safe_count_l2-{}-{}".format(code, last_buy_single_index)
|
self.__getRedis().setex(key, tool.get_expire(),
|
json.dumps((last_buy_single_index, process_index, buy_num, cancel_num)))
|
|
# 返回数据与更新时间
|
def __get_compute_progress(self, code, last_buy_single_index):
|
key = "safe_count_l2-{}-{}".format(code, last_buy_single_index)
|
val = self.__getRedis().get(key)
|
if val is None:
|
return None, -1, 0, 0
|
val = json.loads(val)
|
return val[0], val[1], val[2], val[3]
|
|
# 保存最近的下单信息
|
def __save_latest_place_order_info(self, code, buy_single_index, buy_exec_index, cancel_index):
|
key = "latest_place_order_info-{}".format(code)
|
self.__getRedis().setex(key, tool.get_expire(), json.dumps((buy_single_index, buy_exec_index, cancel_index)))
|
|
def __get_latest_place_order_info(self, code):
|
key = "latest_place_order_info-{}".format(code)
|
val = self.__getRedis().get(key)
|
if val is None:
|
return None, None, None
|
val = json.loads(val)
|
return val[0], val[1], val[2]
|
|
def __get_all_compute_progress(self, code):
|
key_regex = f"safe_count_l2-{code}-*"
|
keys = self.__getRedis().keys(key_regex)
|
vals = []
|
for k in keys:
|
val = self.__getRedis().get(k)
|
val = json.loads(val)
|
vals.append(val)
|
return vals
|
|
def clear_data(self, code):
|
key_regex = f"safe_count_l2-{code}-*"
|
keys = self.__getRedis().keys(key_regex)
|
for k in keys:
|
self.__getRedis().delete(k)
|
|
key = f"latest_place_order_info-{code}"
|
self.__getRedis().delete(key)
|
|
# 获取基础的安全笔数
|
def __get_base_save_count(self, code):
|
return l2_trade_factor.L2TradeFactorUtil.get_safe_buy_count(code)
|
|
# 获取最后的安全笔数
|
def get_safe_count(self, code):
|
rate = self.__get_rate(code)
|
print("--------------------------------")
|
print("安全笔数比例：", rate)
|
print("--------------------------------")
|
count = self.__get_base_save_count(code)
|
count = round(count * rate)
|
if count < 8:
|
count = 8
|
if count > 21:
|
count = 21
|
return count
|
|
# 计算留下来的比例
|
# last_buy_single_index 上一次下单信号起始位置
|
# cancel_index 上一次取消下单的位置
|
# start_index 数据开始位置
|
# end_index 数据结束位置
|
def compute_left_rate(self, code, start_index, end_index, total_datas,
|
local_today_num_operate_map):
|
last_buy_single_index, buy_exec_index, cancel_index = self.__get_latest_place_order_info(code)
|
if last_buy_single_index is None:
|
return
|
cancel_time = None
|
if cancel_index is not None:
|
cancel_time = total_datas[cancel_index]["val"]["time"]
|
# 获取处理的进度
|
last_buy_single_index_, process_index, buy_num, cancel_num = self.__get_compute_progress(code,
|
last_buy_single_index)
|
|
break_index = -1
|
for i in range(start_index, end_index):
|
data = total_datas[i]
|
val = data["val"]
|
# 如果没有取消位置就一直计算下去， 计算截至时间不能大于取消时间
|
if cancel_time and int(cancel_time.replace(":", "")) < int(val["time"].replace(":", "")):
|
break_index = i
|
break
|
if break_index >= 0:
|
end_index = break_index - 1
|
# 获取开始计算的位置
|
start_compute_index = min(start_index, last_buy_single_index)
|
if start_compute_index <= process_index:
|
start_compute_index = process_index + 1
|
|
for i in range(start_compute_index, end_index):
|
data = total_datas[i]
|
val = data["val"]
|
if process_index >= i:
|
continue
|
if L2DataUtil.is_limit_up_price_buy(val):
|
# 涨停买
|
buy_num += int(val["num"]) * data["re"]
|
elif L2DataUtil.is_limit_up_price_buy_cancel(val):
|
# 获取买入信息
|
buy_index = l2_data_source_util.L2DataSourceUtils.get_buy_index_with_cancel_data(code, data,
|
local_today_num_operate_map)
|
if buy_index is not None:
|
if last_buy_single_index <= buy_index <= end_index:
|
cancel_num += int(val["num"]) * data["re"]
|
|
process_index = end_index
|
# 保存处理进度与数量
|
self.__save_compute_progress(code, last_buy_single_index, process_index, buy_num, cancel_num)
|
|
# 获取比例
|
def __get_rate(self, code):
|
vals = self.__get_all_compute_progress(code)
|
rate = (1 - 0)
|
for val in vals:
|
temp_rate = (1 - round((val[2] - val[3]) / val[2], 4))
|
if temp_rate > 1:
|
temp_rate = 1
|
rate *= temp_rate
|
return rate
|
|
# 下单成功
|
def save_place_order_info(self, code, buy_single_index, buy_exec_index, cancel_index):
|
self.__save_latest_place_order_info(code, buy_single_index, buy_exec_index, cancel_index)
|
