#!/bin/bash

# Parse command line arguments
START_DATE=${1:-$(date -d "2 days ago" +"%Y%m%d")}
END_DATE=${2:-$(date -d "tomorrow" +"%Y%m%d")}

cd ../
source .venv/bin/activate
rm -rf user_data/models/*
rm -rf ./freqtrade/user_data/data/backtest_results/*

#docker-compose -f docker-compose_backtest.yml run --rm freqtrade >output.log 2>&1
freqtrade backtesting \
  --logfile ./user_data/logs/freqtrade.log \
  --freqaimodel XGBoostRegressor \
  --config config_examples/theforcev7.json \
  --strategy-path ./freqtrade/templates \
  --strategy TheForceV7 \
  --timerange ${START_DATE}-${END_DATE} \
  --breakdown week month \
  --export trades \
  --fee 0.0008 \
  --cache none >output.log 2>&1
sed -i 's/\x1B\[[0-9;]*m//g' output.log
python3 filter.py

rm ./result/* -fr
mv ./user_data/backtest_results/* ./result/

cd ./result
# 查找当前目录下的所有 zip 文件
zip_files=(*.zip)

# 检查是否只有一个 zip 文件
if [ ${#zip_files[@]} -eq 1 ]; then
  # 解压缩该 zip 文件到当前目录
  unzip "${zip_files[0]}"
  rm *.zip
  rm *.feather
else
  echo "当前目录下没有 zip 文件或者有多个 zip 文件，无法操作。"
fi

cd -
sed -i 's/\x1B\[[0-9;]*m//g' output.log
python3 filter.py
cp output_filted.log result/ -f
cd tools/
python tradestocsv.py
cd ../