diff --git a/piker/brokers/kraken.py b/piker/brokers/kraken.py index 3f7b1eb0..a873575a 100644 --- a/piker/brokers/kraken.py +++ b/piker/brokers/kraken.py @@ -438,7 +438,7 @@ def normalize_symbol( async def handle_order_requests( - client: #kraken, + client: 'test',#kraken, ems_order_stream: tractor.MsgStream, ) -> None: @@ -447,61 +447,61 @@ async def handle_order_requests( async for request_msg in ems_order_stream: log.info(f'Received order request {request_msg}') - action = request_msg['action'] + # action = request_msg['action'] - if action in {'buy', 'sell'}: + # if action in {'buy', 'sell'}: - account = request_msg['account'] - if account != 'kraken.spot': - log.error( - 'This is a kraken account, \ - only a `kraken.spot` selection is valid' - ) - await ems_order_stream.send(BrokerError( - oid=request_msg['oid'] - symbol=request_msg['symbol'] - reason=f'Kraken only, No account found: `{account}` ?', - ).dict()) - continue + # account = request_msg['account'] + # if account != 'kraken.spot': + # log.error( + # 'This is a kraken account, \ + # only a `kraken.spot` selection is valid' + # ) + # await ems_order_stream.send(BrokerError( + # oid=request_msg['oid'], + # symbol=request_msg['symbol'], + # reason=f'Kraken only, No account found: `{account}` ?', + # ).dict()) + # continue - # validate - order = BrokerdOrder(**request_msg) + # # validate + # order = BrokerdOrder(**request_msg) - # call our client api to submit the order - ## TODO: look into the submit_limit method, do it write my own? - reqid = await client.submit_limit( + # # call our client api to submit the order + # ## TODO: look into the submit_limit method, do it write my own? + # reqid = await client.submit_limit( - oid=order.oid, - symbol=order.symbol, - price=order.price, - action=order.action, - size=order.size, - ## XXX: how do I handle new orders - reqid=order.reqid, - ) + # oid=order.oid, + # symbol=order.symbol, + # price=order.price, + # action=order.action, + # size=order.size, + # ## XXX: how do I handle new orders + # reqid=order.reqid, + # ) - # deliver ack that order has been submitted to broker routing - await ems_order_stream.send( - BrokerdOrderAck( + # # deliver ack that order has been submitted to broker routing + # await ems_order_stream.send( + # BrokerdOrderAck( - # ems order request id - oid=order.oid, + # # ems order request id + # oid=order.oid, - # broker specific request id - reqid=reqid, + # # broker specific request id + # reqid=reqid, - ).dict() - ) + # ).dict() + # ) - elif action == 'cancel': - msg = BrokerdCancel(**request_msg) + # elif action == 'cancel': + # msg = BrokerdCancel(**request_msg) - await client.submit_cancel( - reqid=msg.reqid - ) + # await client.submit_cancel( + # reqid=msg.reqid + # ) - else: - log.error(f'Unknown order command: {request_msg}') + # else: + # log.error(f'Unknown order command: {request_msg}') @tractor.context @@ -527,17 +527,19 @@ async def trades_dialogue( all_positions.append(msg.dict()) open_orders = await client.get_user_data('OpenOrders', {}) - await tractor.breakpoint() + #await tractor.breakpoint() await ctx.started((all_positions, (acc_name,))) - await trio.sleep_forever() - - # async with ( - # ctx.open_stream() as ems_stream, - # trio.open_nursery as n, - # ): + #await trio.sleep_forever() + async with ( + ctx.open_stream() as ems_stream, + trio.open_nursery() as n, + ): + ## TODO: maybe add multiple accounts + n.start_soon(handle_order_requests, client, ems_stream) + async def stream_messages(ws):