backtest-automation/run_backtest.py

49 lines
2.2 KiB
Python

from datetime import datetime, timedelta
from dotenv import load_dotenv
from os import getenv
from backtesting import available_entry_times, backtest_iron_condor
from backtesting.credit_targeting import create_strategies as credit_strategies
from backtesting.credit_targeting import iron_condor_strategy as credit_iron_condor_strategy
from backtesting.delta_targeting import create_strategies as delta_strategies
from backtesting.delta_targeting import iron_condor_strategy as delta_iron_condor_strategy
from backtesting.option_spread_strategy import OptionSpreadStrategy
from database.backtest import insert
load_dotenv()
def run_backtest(strategy_name: str, start_date: datetime, end_date: datetime,
call_spread_strategy: OptionSpreadStrategy, put_spread_strategy: OptionSpreadStrategy):
backtest_results = backtest_iron_condor(
strategy_name,
call_spread_strategy,
put_spread_strategy,
start_date,
end_date
)
if not backtest_results.empty:
backtest_results.drop('Cumulative Profit', axis = 1, inplace = True)
print(backtest_results)
insert(backtest_results)
def run_backtests(start_date: datetime, end_date: datetime):
credit_targets = [float(credit) for credit in getenv('CREDIT_TARGETS').split(',')]
delta_targets = [float(delta) for delta in getenv('DELTA_TARGETS').split(',')]
for entry_time in available_entry_times():
for credit_target in credit_targets:
strategy_name = credit_iron_condor_strategy(credit_target)
call_spread_strategy, put_spread_strategy = credit_strategies(credit_target, entry_time)
run_backtest(strategy_name, start_date, end_date, call_spread_strategy, put_spread_strategy)
for delta_target in delta_targets:
strategy_name = delta_iron_condor_strategy(delta_target)
call_spread_strategy, put_spread_strategy = delta_strategies(delta_target, entry_time)
run_backtest(strategy_name, start_date, end_date, call_spread_strategy, put_spread_strategy)
if __name__ == '__main__':
end_date = datetime.now()
start_date = end_date - timedelta(days = 1)
run_backtests(start_date, end_date)