153 lines
5.0 KiB
Python
153 lines
5.0 KiB
Python
from alpaca.common.exceptions import APIError
|
|
from alpaca.trading.client import TradingClient
|
|
from alpaca.trading.enums import OrderSide, TimeInForce
|
|
from alpaca.trading.requests import (
|
|
GetAssetsRequest,
|
|
LimitOrderRequest,
|
|
MarketOrderRequest,
|
|
)
|
|
|
|
from core.exchanges import BaseExchange
|
|
from core.lib.schemas import alpaca_s
|
|
|
|
ALPACA_SCHEMA_MAPPING = {
|
|
"get_account": alpaca_s.GetAccount,
|
|
"get_all_assets": alpaca_s.GetAllAssets,
|
|
"get_all_positions": alpaca_s.GetAllPositions,
|
|
"get_open_position": alpaca_s.GetOpenPosition,
|
|
}
|
|
|
|
|
|
class AlpacaExchange(BaseExchange):
|
|
def set_schema(self):
|
|
self.schema = ALPACA_SCHEMA_MAPPING
|
|
|
|
def connect(self):
|
|
self.client = TradingClient(
|
|
self.account.api_key,
|
|
self.account.api_secret,
|
|
paper=self.account.sandbox,
|
|
raw_data=True,
|
|
)
|
|
|
|
def get_account(self):
|
|
return self.call("get_account")
|
|
|
|
def get_supported_assets(self):
|
|
request = GetAssetsRequest(status="active", asset_class="crypto")
|
|
success, assets = self.call("get_all_assets", filter=request)
|
|
# assets = self.client.get_all_assets(filter=request)
|
|
if not success:
|
|
return (success, assets)
|
|
assets = assets["itemlist"]
|
|
asset_list = [x["symbol"] for x in assets if "symbol" in x]
|
|
print("Supported symbols", asset_list)
|
|
|
|
return (True, asset_list)
|
|
|
|
def get_balance(self):
|
|
success, account_info = self.call("get_account")
|
|
if not success:
|
|
return (success, account_info)
|
|
equity = account_info["equity"]
|
|
try:
|
|
balance = float(equity)
|
|
except ValueError:
|
|
return (False, "Invalid balance")
|
|
|
|
return (True, balance)
|
|
|
|
def get_market_value(self, symbol): # TODO: pydantic
|
|
try:
|
|
position = self.client.get_position(symbol)
|
|
except APIError as e:
|
|
self.log.error(f"Could not get market value for {symbol}: {e}")
|
|
return False
|
|
return float(position["market_value"])
|
|
|
|
def post_trade(self, trade): # TODO: pydantic
|
|
# the trade is not placed yet
|
|
if trade.direction == "buy":
|
|
direction = OrderSide.BUY
|
|
elif trade.direction == "sell":
|
|
direction = OrderSide.SELL
|
|
else:
|
|
raise Exception("Unknown direction")
|
|
|
|
cast = {
|
|
"symbol": trade.symbol,
|
|
"side": direction,
|
|
"time_in_force": TimeInForce.IOC,
|
|
}
|
|
if trade.amount is not None:
|
|
cast["qty"] = trade.amount
|
|
if trade.amount_usd is not None:
|
|
cast["notional"] = trade.amount_usd
|
|
if not trade.amount and not trade.amount_usd:
|
|
return (False, "No amount specified")
|
|
if trade.take_profit:
|
|
cast["take_profit"] = {"limit_price": trade.take_profit}
|
|
if trade.stop_loss:
|
|
stop_limit_price = trade.stop_loss - (trade.stop_loss * 0.005)
|
|
cast["stop_loss"] = {
|
|
"stop_price": trade.stop_loss,
|
|
"limit_price": stop_limit_price,
|
|
}
|
|
if trade.type == "market":
|
|
market_order_data = MarketOrderRequest(**cast)
|
|
try:
|
|
order = self.client.submit_order(order_data=market_order_data)
|
|
except APIError as e:
|
|
self.log.error(f"Error placing market order: {e}")
|
|
trade.status = "error"
|
|
trade.save()
|
|
return (False, e)
|
|
elif trade.type == "limit":
|
|
if not trade.price:
|
|
return (False, "Limit order with no price")
|
|
cast["limit_price"] = trade.price
|
|
limit_order_data = LimitOrderRequest(**cast)
|
|
try:
|
|
order = self.client.submit_order(order_data=limit_order_data)
|
|
except APIError as e:
|
|
self.log.error(f"Error placing limit order: {e}")
|
|
trade.status = "error"
|
|
trade.save()
|
|
return (False, e)
|
|
|
|
else:
|
|
raise Exception("Unknown trade type")
|
|
trade.response = order
|
|
trade.status = "posted"
|
|
trade.order_id = order["id"]
|
|
trade.client_order_id = order["client_order_id"]
|
|
trade.save()
|
|
return (True, order)
|
|
|
|
def get_trade(self, trade_id):
|
|
pass
|
|
|
|
def update_trade(self, trade):
|
|
pass
|
|
|
|
def cancel_trade(self, trade_id):
|
|
pass
|
|
|
|
def get_position_info(self, asset_id):
|
|
success, position = self.call("get_open_position", asset_id)
|
|
if not success:
|
|
return (success, position)
|
|
return (True, position)
|
|
|
|
def get_all_positions(self):
|
|
items = []
|
|
success, positions = self.call("get_all_positions")
|
|
if not success:
|
|
return (success, positions)
|
|
|
|
for item in positions["itemlist"]:
|
|
item["account_id"] = self.account.id
|
|
item["unrealized_pl"] = float(item["unrealized_pl"])
|
|
items.append(item)
|
|
return (True, items)
|