-
Notifications
You must be signed in to change notification settings - Fork 0
/
tradingbot.py
98 lines (89 loc) · 3.6 KB
/
tradingbot.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
from lumibot.brokers import Alpaca
from lumibot.backtesting import BacktestingBroker, YahooDataBacktesting
from lumibot.strategies.strategy import Strategy
from lumibot.traders import Trader
from datetime import datetime
from alpaca_trade_api import REST
from timedelta import Timedelta
from finbert_utils import estimate_sentiment
API_KEY = "PKQCVTV8X4SDMEO7KIMR"
API_SECRET = "PxaJnBaIfiNdBrB290hl6hanDZCL7umlQSwRapEN"
BASE_URL = "https://paper-api.alpaca.markets/v2"
ALPACA_CREDS = {
"API_KEY":API_KEY,
"API_SECRET": API_SECRET,
"PAPER": True
}
class MLTrader(Strategy):
def initialize(self, symbol:str="AAPL", cash_at_risk:float=.5):
self.symbol = symbol
self.sleeptime = "1H" #likely
self.last_trade = None
self.cash_at_risk = cash_at_risk
self.api = REST(base_url=BASE_URL, key_id=API_KEY, secret_key=API_SECRET)
def position_sizing(self):
cash = self.get_cash()
last_price = self.get_last_price(self.symbol)
quantity = round(cash * self.cash_at_risk / last_price,0)
return cash, last_price, quantity
def get_dates(self):
today = self.get_datetime()
three_hours_prior = today - Timedelta(hours=12) #likely
return today.strftime('%Y-%m-%d'), three_hours_prior.strftime('%Y-%m-%d')
def get_sentiment(self):
today, three_hours_prior = self.get_dates()
news = self.api.get_news(symbol=self.symbol,
start=three_hours_prior,
end=today,
include_content = False,
exclude_contentless = True,
)
# + " " + ev.__dict__["_raw"].get("content", "")
news = [ev.__dict__["_raw"]["headline"] for ev in news]
probability, sentiment = estimate_sentiment(news)
return probability, sentiment
def on_trading_iteration(self):
self.await_market_to_open()
cash, last_price, quantity = self.position_sizing()
probability, sentiment = self.get_sentiment()
if cash > last_price:
if sentiment == "positive":
order = self.create_order(
self.symbol,
quantity,
"buy",
type="bracket",
take_profit_price=last_price*1.20,
stop_loss_price=last_price*.95
)
self.submit_order(order)
self.last_trade = "buy"
elif sentiment == "negative" and probability > .999:
if self.last_trade == "buy":
self.sell_all()
position = self.get_position(self.symbol)
if position.quantity > 0:
order = self.create_order(
self.symbol,
quantity,
"sell",
type="bracket",
take_profit_price=last_price*.8,
stop_loss_price=last_price*1.05
)
self.submit_order(order)
self.last_trade = "sell"
self.await_market_to_close()
start_date = datetime(2024,6,6)
end_date = datetime(2024,6,7)
broker = Alpaca(ALPACA_CREDS)
strategy = MLTrader(name='mlstrat', broker=broker,
parameters={"symbol":"AAPL",
"cash_at_risk":.5})
strategy.backtest(
YahooDataBacktesting,
start_date,
end_date,
parameters={"symbol":"AAPL", "cash_at_risk":.5},
benchmark_asset='AAPL'
)