144 lines
3.6 KiB
Python
144 lines
3.6 KiB
Python
|
import tractor
|
||
|
|
import trio
|
||
|
|
from uuid import uuid4
|
||
|
|
|
||
|
|
from piker.clearing import (
|
||
|
|
open_ems,
|
||
|
|
OrderClient,
|
||
|
|
)
|
||
|
|
|
||
|
|
# TODO: we should probably expose these top level in this subsys?
|
||
|
|
from piker.clearing._messages import (
|
||
|
|
Order,
|
||
|
|
Status,
|
||
|
|
BrokerdPosition,
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
async def wait_for_order_status(
|
||
|
|
trades_stream: tractor.MsgStream,
|
||
|
|
oid: str,
|
||
|
|
expect_status: str,
|
||
|
|
|
||
|
|
) -> tuple[
|
||
|
|
list[Status],
|
||
|
|
list[BrokerdPosition],
|
||
|
|
]:
|
||
|
|
'''
|
||
|
|
Wait for a specific order status for a given dialog, return msg flow
|
||
|
|
up to that msg and any position update msgs in a tuple.
|
||
|
|
|
||
|
|
'''
|
||
|
|
# Wait for position message before moving on to verify flow(s)
|
||
|
|
# for the multi-order position entry/exit.
|
||
|
|
status_msgs: list[Status] = []
|
||
|
|
pp_msgs: list[BrokerdPosition] = []
|
||
|
|
|
||
|
|
async for msg in trades_stream:
|
||
|
|
match msg:
|
||
|
|
case {'name': 'position'}:
|
||
|
|
ppmsg = BrokerdPosition(**msg)
|
||
|
|
pp_msgs.append(ppmsg)
|
||
|
|
|
||
|
|
case {
|
||
|
|
'name': 'status',
|
||
|
|
}:
|
||
|
|
msg = Status(**msg)
|
||
|
|
status_msgs.append(msg)
|
||
|
|
|
||
|
|
# if we get the status we expect then return all
|
||
|
|
# collected msgs from the brokerd dialog up to the
|
||
|
|
# exected msg B)
|
||
|
|
if (
|
||
|
|
msg.resp == expect_status
|
||
|
|
and msg.oid == oid
|
||
|
|
):
|
||
|
|
return status_msgs, pp_msgs
|
||
|
|
|
||
|
|
|
||
|
|
async def bot_main():
|
||
|
|
'''
|
||
|
|
Boot the piker runtime, open an ems connection, submit
|
||
|
|
and process orders statuses in real-time.
|
||
|
|
|
||
|
|
'''
|
||
|
|
from piker.service import maybe_open_pikerd
|
||
|
|
|
||
|
|
port: int = 6666 # non-default pikerd addr
|
||
|
|
ll: str = 'info'
|
||
|
|
reg_addr: tuple[str, int] = ('127.0.0.1', port)
|
||
|
|
|
||
|
|
# open an order ctl client
|
||
|
|
client: OrderClient
|
||
|
|
trades_stream: tractor.MsgStream
|
||
|
|
accounts: list[str]
|
||
|
|
|
||
|
|
fqme: str = 'btcusdt.usdtm.perp.binance'
|
||
|
|
|
||
|
|
async with (
|
||
|
|
|
||
|
|
# init and sync actor-service runtime
|
||
|
|
maybe_open_pikerd(
|
||
|
|
registry_addr=reg_addr,
|
||
|
|
loglevel=ll,
|
||
|
|
debug_mode=False,
|
||
|
|
|
||
|
|
),
|
||
|
|
tractor.wait_for_actor(
|
||
|
|
'pikerd',
|
||
|
|
arbiter_sockaddr=reg_addr,
|
||
|
|
),
|
||
|
|
|
||
|
|
open_ems(
|
||
|
|
fqme,
|
||
|
|
mode='paper', # {'live', 'paper'}
|
||
|
|
loglevel=ll,
|
||
|
|
|
||
|
|
) as (
|
||
|
|
client, # OrderClient
|
||
|
|
trades_stream, # tractor.MsgStream startup_pps,
|
||
|
|
_, # positions
|
||
|
|
accounts,
|
||
|
|
_, # dialogs
|
||
|
|
)
|
||
|
|
):
|
||
|
|
print(f'Loaded binance accounts: {accounts}')
|
||
|
|
|
||
|
|
price: float = 10e3 # non-clearable
|
||
|
|
size: float = 0.01
|
||
|
|
oid: str = str(uuid4())
|
||
|
|
|
||
|
|
order = Order(
|
||
|
|
exec_mode='live', # {'dark', 'live', 'alert'}
|
||
|
|
action='buy', # TODO: remove this from our schema?
|
||
|
|
oid=oid,
|
||
|
|
account='paper', # use binance.usdtm for binance futes
|
||
|
|
size=size,
|
||
|
|
symbol=fqme,
|
||
|
|
price=price,
|
||
|
|
brokers=['binance'],
|
||
|
|
)
|
||
|
|
await client.send(order)
|
||
|
|
|
||
|
|
msgs, pps = await wait_for_order_status(
|
||
|
|
trades_stream,
|
||
|
|
oid,
|
||
|
|
'open'
|
||
|
|
)
|
||
|
|
|
||
|
|
assert not pps
|
||
|
|
assert msgs[-1].oid == oid
|
||
|
|
|
||
|
|
# cancel the open order
|
||
|
|
await client.cancel(oid)
|
||
|
|
msgs, pps = await wait_for_order_status(
|
||
|
|
trades_stream,
|
||
|
|
oid,
|
||
|
|
'canceled'
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
trio.run(bot_main)
|