2023-12-07 16:12:50 +00:00
|
|
|
import schedule
|
|
|
|
import time
|
|
|
|
|
2024-02-02 18:48:05 +00:00
|
|
|
from datetime import datetime
|
2023-12-13 14:39:58 +00:00
|
|
|
from dotenv import load_dotenv
|
|
|
|
from ibkr import Client
|
2023-12-07 16:12:50 +00:00
|
|
|
from iron_condor import enter_iron_condor
|
|
|
|
from multiprocessing import Process
|
2023-12-13 14:39:58 +00:00
|
|
|
from os import getenv
|
2024-01-29 16:39:58 +00:00
|
|
|
from pytz import timezone
|
2023-12-13 14:39:58 +00:00
|
|
|
|
|
|
|
load_dotenv()
|
2023-12-07 16:12:50 +00:00
|
|
|
|
2024-01-29 16:39:58 +00:00
|
|
|
eastern_timezone = 'America/New_York'
|
2024-01-19 21:19:13 +00:00
|
|
|
|
2023-12-07 16:12:50 +00:00
|
|
|
def enter_trade():
|
|
|
|
job_process = Process(target = enter_iron_condor)
|
|
|
|
job_process.start()
|
|
|
|
|
2023-12-13 14:39:58 +00:00
|
|
|
def connection_successful():
|
|
|
|
try:
|
|
|
|
ibkr_host = getenv('IBKR_HOST')
|
|
|
|
ibkr_port = getenv('IBKR_PORT')
|
|
|
|
Client(host = ibkr_host, port = ibkr_port)
|
|
|
|
return True
|
|
|
|
except:
|
|
|
|
return False
|
|
|
|
|
2023-12-07 16:12:50 +00:00
|
|
|
if __name__ == '__main__':
|
2023-12-13 14:39:58 +00:00
|
|
|
if connection_successful():
|
|
|
|
print('Connected to IBKR.')
|
|
|
|
|
2024-01-29 16:39:58 +00:00
|
|
|
now = datetime.now(timezone(eastern_timezone))
|
|
|
|
|
2024-02-13 16:13:51 +00:00
|
|
|
entry_times = getenv('ENTRY_TIMES').split(',')
|
2023-12-13 14:39:58 +00:00
|
|
|
for entry_time in entry_times:
|
2024-01-29 16:39:58 +00:00
|
|
|
schedule_time = datetime.strptime(entry_time, '%H:%M:%S').replace(
|
|
|
|
year = now.year,
|
|
|
|
month = now.month,
|
|
|
|
day = now.day,
|
|
|
|
tzinfo = now.tzinfo
|
|
|
|
)
|
|
|
|
|
|
|
|
# Prevent scheduling for times that have already elapsed.
|
|
|
|
if schedule_time > now:
|
|
|
|
print(f'Scheduling for {entry_time}.')
|
2024-02-13 18:50:50 +00:00
|
|
|
schedule.every().day.at(entry_time, eastern_timezone).do(enter_trade, schedule_time)
|
2023-12-07 16:12:50 +00:00
|
|
|
|
2023-12-13 14:39:58 +00:00
|
|
|
while True:
|
|
|
|
schedule.run_pending()
|
|
|
|
time.sleep(1)
|
|
|
|
else:
|
|
|
|
print('ERROR: Cannot connect to IBKR. Ensure that TWS / Gateway is running.')
|