diff --git a/backtest.py b/backtest.py index c205cb8..c1bdc9e 100644 --- a/backtest.py +++ b/backtest.py @@ -119,6 +119,17 @@ def load_strategy(strategy_file): return calculate_indicators, strategy_class +def apply_color_scheme(): + """ + 应用颜色方案:红涨绿跌(中国股市风格) + """ + import backtesting._plotting as plotting + from bokeh.colors.named import tomato, lime + + plotting.BULL_COLOR = tomato + plotting.BEAR_COLOR = lime + + def parse_arguments(): """ 解析命令行参数 @@ -256,6 +267,8 @@ def main(): # 解析参数 args = parse_arguments() + apply_color_scheme() + # 加载数据 print(f"加载股票数据: {args.code} ({args.start_date} ~ {args.end_date})") data = load_data_from_db(args.code, args.start_date, args.end_date)