mirror of
https://github.com/asavinov/intelligent-trading-bot.git
synced 2026-05-04 16:26:44 +00:00
42 lines
1.4 KiB
Python
42 lines
1.4 KiB
Python
from common.types import Venue
|
|
from outputs.trader_binance import (
|
|
trader_binance,
|
|
update_account_balance as update_account_balance_binance,
|
|
update_order_status as update_order_status_binance,
|
|
update_trade_status as update_trade_status_binance,
|
|
)
|
|
from outputs.trader_mt5 import (
|
|
trader_mt5,
|
|
update_account_balance as update_account_balance_mt5,
|
|
update_order_status as update_order_status_mt5,
|
|
update_trade_status as update_trade_status_mt5,
|
|
)
|
|
|
|
_TRADER_MAP: dict[Venue, dict[str, callable]] = {
|
|
Venue.BINANCE: {
|
|
"trader": trader_binance,
|
|
"update_account_balance": update_account_balance_binance,
|
|
"update_order_status": update_order_status_binance,
|
|
"update_trade_status": update_trade_status_binance,
|
|
},
|
|
Venue.MT5: {
|
|
"trader": trader_mt5,
|
|
"update_account_balance": update_account_balance_mt5,
|
|
"update_order_status": update_order_status_mt5,
|
|
"update_trade_status": update_trade_status_mt5,
|
|
},
|
|
}
|
|
|
|
def get_trader_functions(venue: Venue) -> dict[str, callable]:
|
|
"""
|
|
Return a dict of the four trader-related callables for the given venue.
|
|
|
|
Example:
|
|
funcs = get_trader_functions(Venue.BINANCE)
|
|
funcs["trader"](...)
|
|
funcs["update_order_status"](...)
|
|
"""
|
|
try:
|
|
return _TRADER_MAP[venue]
|
|
except KeyError:
|
|
raise ValueError(f"Unknown trader venue: {venue!r}")
|