basic bot: set unix fileformat, add KBI handling to cancel order submission
Tyler Goodlet
parent
63a6c6efde
commit
5690595064
|
|
@ -1,143 +1,146 @@
|
|||
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)
|
||||
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 = 30e3 # 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
|
||||
|
||||
try:
|
||||
# wait for ctl-c from user..
|
||||
await trio.sleep_forever()
|
||||
except KeyboardInterrupt:
|
||||
# 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)
|
||||
|
|
|
|||
Loading…
Reference in New Issue