diff --git a/TurtleOnTime.py b/TurtleOnTime.py index fa6eda9..a62d116 100644 --- a/TurtleOnTime.py +++ b/TurtleOnTime.py @@ -11,6 +11,14 @@ import mysql_database from EmailTest import send_email, parse_return_email from dataclasses import dataclass import time + +''' +todo + +1 运行过程框架调整，支持多个turtle同时监测 +2 增加运行状态写入yaml文件，读取文件恢复状态 +''' + @dataclass class BuyState: trigger_time: float # 触发次数 @@ -620,110 +628,88 @@ class TurtleTrading_OnTime(object): Net_return=abs(self.turtle.Capital - available_cash)) self.turtle.tradeslog.append(sale_this_time) - def run_short_trading_loop(self): - while True: - # 获取当前时间 - now = datetime.now().time() + def run_short_trading_loop(self, stock_data, etf_data): - # 判断当前时间是否在交易时段内（9:30-11:30 或 13:00-15:00） - is_trading_time = ( - (now.hour == 9 and now.minute >= 30) or - (now.hour == 10 and 0 <= now.minute <= 59) or - (now.hour == 11 and now.minute <= 30) or - (now.hour == 13 and 0 <= now.minute <= 59) or - (now.hour == 14 and 0 <= now.minute <= 59) or - (now.hour == 15 and now.minute <= 0) - ) + now = datetime.now().time() + # 根据类型获取当前价格 + if self.turtle.type == "stock": + self.turtle.PriceNow = float(stock_data.loc[etf_data['代码'] == self.turtle.TradeCode, '最新价'].values[0]) + elif self.turtle.type == "etf": + # self.turtle.PriceNow = float(etf_data.loc[etf_data['基金代码'] == self.turtle.TradeCode, '当前-单位净值'].values[0]) + self.turtle.PriceNow = float(etf_data.loc[etf_data['代码'] == self.turtle.TradeCode, '最新价'].values[0]) + # # 9点30 判断是否跳空高开 + if now.hour == 9 and now.minute == 30 and self.turtle.PriceNow > self.turtle.prev_heigh: + self.turtle.is_gap_up = True + + + # 判断当前仓位状态并执行相应操作 + if self.turtle.TrigerTime == 0: + # 空仓状态 + if self.turtle.system1EnterNormal( + self.turtle.PriceNow, + self.turtle.Donchian_20_up, + self.turtle.BreakOutLog + ): + self.Buy_stock(self.turtle.PriceNow) - # if not is_trading_time: - # # 非交易时间，等待 1 分钟后继续循环 - # time.sleep(60) - # continue + # 突破 记录self.turtle.breakoutlog + today = datetime.now().strftime("%Y-%m-%d") + breakout_this_time = BreakOutLog(today, + self.turtle.Donchian_20_up, + self.turtle.Donchian_20_up - 2 * self.turtle.N, + 'valid', + None) + self.turtle.BreakOutLog.append(breakout_this_time) - # 获取股票和ETF数据 - stock_data, etf_data = self.get_stocks_data() + elif self.turtle.system1EnterSafe( + self.turtle.PriceNow, + self.turtle.Donchian_50_up + ): + self.Buy_stock(self.turtle.PriceNow) - # 根据类型获取当前价格 - if self.turtle.type == "stock": - self.turtle.PriceNow = float(stock_data.loc[etf_data['代码'] == self.turtle.TradeCode, '最新价'].values[0]) - - elif self.turtle.type == "etf": - # self.turtle.PriceNow = float(etf_data.loc[etf_data['基金代码'] == self.turtle.TradeCode, '当前-单位净值'].values[0]) - self.turtle.PriceNow = float(etf_data.loc[etf_data['代码'] == self.turtle.TradeCode, '最新价'].values[0]) - - # # 9点30 判断是否跳空高开 - if now.hour == 9 and now.minute == 30 and self.turtle.PriceNow > self.turtle.prev_heigh: - self.turtle.is_gap_up = True + elif 1 <= self.turtle.TrigerTime <= 3: + # # 突破状态 + # if self.turtle.system1EnterNormal( + # self.turtle.PriceNow, + # self.turtle.Donchian_20_up, + # self.turtle.BreakOutLog + # ): + # self.Buy_stock(self.turtle.PriceNow) + # elif self.turtle.system1EnterSafe( + # self.turtle.PriceNow, + # self.turtle.Donchian_50_up + # ): + # self.Buy_stock(self.turtle.PriceNow) + + # 加仓状态 + if self.turtle.add(self.turtle.PriceNow): + self.add_stock(self.turtle.PriceNow) - - # 判断当前仓位状态并执行相应操作 - if self.turtle.TrigerTime == 0: - # 空仓状态 - if self.turtle.system1EnterNormal( - self.turtle.PriceNow, - self.turtle.Donchian_20_up, - self.turtle.BreakOutLog - ): - self.Buy_stock(self.turtle.PriceNow) - - # 突破 记录self.turtle.breakoutlog - today = datetime.now().strftime("%Y-%m-%d") - breakout_this_time = BreakOutLog(today, - self.turtle.Donchian_20_up, - self.turtle.Donchian_20_up - 2 * self.turtle.N, - 'valid', - None) - self.turtle.BreakOutLog.append(breakout_this_time) + # 止损状态 + elif self.turtle.system_1_stop(self.turtle.PriceNow): + self.stop_sale_stock(self.turtle.PriceNow) + + # 止盈 + elif self.turtle.system_1_Out( + self.turtle.PriceNow, + self.turtle.Donchian_10_down + ): + self.out_sale_stock(self.turtle.PriceNow) - elif self.turtle.system1EnterSafe( - self.turtle.PriceNow, - self.turtle.Donchian_50_up - ): - self.Buy_stock(self.turtle.PriceNow) + elif self.turtle.TrigerTime == 4: + # 满仓 止损 止盈 + if self.turtle.system_1_stop(self.turtle.PriceNow): + self.stop_sale_stock(self.turtle.PriceNow) + elif self.turtle.system_1_Out( + self.turtle.PriceNow, + self.turtle.Donchian_10_down + ): + self.out_sale_stock(self.turtle.PriceNow) - elif 1 <= self.turtle.TrigerTime <= 3: - # # 突破状态 - # if self.turtle.system1EnterNormal( - # self.turtle.PriceNow, - # self.turtle.Donchian_20_up, - # self.turtle.BreakOutLog - # ): - # self.Buy_stock(self.turtle.PriceNow) - # elif self.turtle.system1EnterSafe( - # self.turtle.PriceNow, - # self.turtle.Donchian_50_up - # ): - # self.Buy_stock(self.turtle.PriceNow) - - # 加仓状态 - if self.turtle.add(self.turtle.PriceNow): - self.add_stock(self.turtle.PriceNow) - - # 止损状态 - elif self.turtle.system_1_stop(self.turtle.PriceNow): - self.stop_sale_stock(self.turtle.PriceNow) - - # 止盈 - elif self.turtle.system_1_Out( - self.turtle.PriceNow, - self.turtle.Donchian_10_down - ): - self.out_sale_stock(self.turtle.PriceNow) - - elif self.turtle.TrigerTime == 4: - # 满仓 止损 止盈 - if self.turtle.system_1_stop(self.turtle.PriceNow): - self.stop_sale_stock(self.turtle.PriceNow) - elif self.turtle.system_1_Out( - self.turtle.PriceNow, - self.turtle.Donchian_10_down - ): - self.out_sale_stock(self.turtle.PriceNow) - - # 等待 1 分钟后下一次循环 - time.sleep(60) + # 等待 1 分钟后下一次循环 + time.sleep(60) def Start_short_system(self): """启动short系统 @@ -745,7 +731,34 @@ class TurtleTrading_OnTime(object): self.turtle.CalPositionSize() # 每分钟获取一次数据，判断是否触发条件 9:30-11:30 13:00-15:00 - self.run_short_trading_loop() + while True: + # 获取当前时间 + now = datetime.now().time() + + # 判断当前时间是否在交易时段内（9:30-11:30 或 13:00-15:00） + is_trading_time = ( + (now.hour == 9 and now.minute >= 30) or + (now.hour == 10 and 0 <= now.minute <= 59) or + (now.hour == 11 and now.minute <= 30) or + (now.hour == 13 and 0 <= now.minute <= 59) or + (now.hour == 14 and 0 <= now.minute <= 59) or + (now.hour == 15 and now.minute <= 0) + ) + + + + if not is_trading_time: + # 非交易时间，等待 1 分钟后继续循环 + time.sleep(60) + continue + + is_stop_time = (now.hour > 15 and now.minute > 0) #收盘时间 + if is_stop_time: + break + + # 获取股票和ETF数据 + stock_data, etf_data = self.get_stocks_data() + self.run_short_trading_loop(stock_data, etf_data) # ------------------结束阶段-------------------- # 数据库更新当天数据，增加ATR、donchian数据 # 直接做个新表 @@ -754,6 +767,7 @@ class TurtleTrading_OnTime(object): time.sleep(16.5*600) if __name__ == '__main__': + user_email = "guoyize2209@163.com" t = TurtleTrading('513870', "etf", 0.0025, 100000, 200000) # t.get_ready(100)