Skip to content

Commit bb1f352

Browse files
committed
Fix order event pyo3 transformation functions
1 parent 8b8fc8a commit bb1f352

File tree

1 file changed

+10
-8
lines changed

1 file changed

+10
-8
lines changed

nautilus_trader/cache/transformers.py

Lines changed: 10 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -28,6 +28,7 @@
2828
from nautilus_trader.model.enums import CurrencyType
2929
from nautilus_trader.model.events import OrderAccepted
3030
from nautilus_trader.model.events import OrderCanceled
31+
from nautilus_trader.model.events import OrderCancelRejected
3132
from nautilus_trader.model.events import OrderDenied
3233
from nautilus_trader.model.events import OrderEmulated
3334
from nautilus_trader.model.events import OrderExpired
@@ -150,8 +151,7 @@ def transform_order_event_to_pyo3(order_event): # noqa: C901
150151
elif isinstance(order_event, OrderSubmitted):
151152
return nautilus_pyo3.OrderSubmitted.from_dict(OrderSubmitted.to_dict(order_event))
152153
elif isinstance(order_event, OrderAccepted):
153-
order_event_dict = OrderAccepted.to_dict(order_event)
154-
return nautilus_pyo3.OrderAccepted.from_dict(order_event_dict)
154+
return nautilus_pyo3.OrderAccepted.from_dict(OrderAccepted.to_dict(order_event))
155155
elif isinstance(order_event, OrderRejected):
156156
return nautilus_pyo3.OrderRejected.from_dict(OrderRejected.to_dict(order_event))
157157
elif isinstance(order_event, OrderCanceled):
@@ -162,16 +162,16 @@ def transform_order_event_to_pyo3(order_event): # noqa: C901
162162
return nautilus_pyo3.OrderTriggered.from_dict(OrderTriggered.to_dict(order_event))
163163
elif isinstance(order_event, OrderPendingUpdate):
164164
return nautilus_pyo3.OrderPendingUpdate.from_dict(OrderPendingUpdate.to_dict(order_event))
165-
elif isinstance(order_event, OrderModifyRejected):
166-
return nautilus_pyo3.OrderModifyRejected.from_dict(OrderModifyRejected.to_dict(order_event))
167165
elif isinstance(order_event, OrderPendingCancel):
168166
return nautilus_pyo3.OrderPendingCancel.from_dict(OrderPendingCancel.to_dict(order_event))
167+
elif isinstance(order_event, OrderCancelRejected):
168+
return nautilus_pyo3.OrderCancelRejected.from_dict(OrderCancelRejected.to_dict(order_event))
169+
elif isinstance(order_event, OrderModifyRejected):
170+
return nautilus_pyo3.OrderModifyRejected.from_dict(OrderModifyRejected.to_dict(order_event))
169171
elif isinstance(order_event, OrderUpdated):
170172
return nautilus_pyo3.OrderUpdated.from_dict(OrderUpdated.to_dict(order_event))
171173
elif isinstance(order_event, OrderFilled):
172174
return nautilus_pyo3.OrderFilled.from_dict(OrderFilled.to_dict(order_event))
173-
elif isinstance(order_event, OrderPendingCancel):
174-
return nautilus_pyo3.OrderPendingCancel.from_dict(OrderPendingCancel.to_dict(order_event))
175175
else:
176176
raise ValueError(f"Unknown order event type: {order_event}")
177177

@@ -220,12 +220,14 @@ def transform_order_event_from_pyo3(order_event_pyo3): # noqa: C901
220220
return OrderExpired.from_dict(order_event_pyo3.to_dict())
221221
elif isinstance(order_event_pyo3, nautilus_pyo3.OrderTriggered):
222222
return OrderTriggered.from_dict(order_event_pyo3.to_dict())
223+
elif isinstance(order_event_pyo3, nautilus_pyo3.OrderPendingCancel):
224+
return OrderPendingCancel.from_dict(order_event_pyo3.to_dict())
223225
elif isinstance(order_event_pyo3, nautilus_pyo3.OrderPendingUpdate):
224226
return OrderPendingUpdate.from_dict(order_event_pyo3.to_dict())
227+
elif isinstance(order_event_pyo3, nautilus_pyo3.OrderCancelRejected):
228+
return OrderCancelRejected.from_dict(order_event_pyo3.to_dict())
225229
elif isinstance(order_event_pyo3, nautilus_pyo3.OrderModifyRejected):
226230
return OrderModifyRejected.from_dict(order_event_pyo3.to_dict())
227-
elif isinstance(order_event_pyo3, nautilus_pyo3.OrderPendingCancel):
228-
return OrderPendingCancel.from_dict(order_event_pyo3.to_dict())
229231
elif isinstance(order_event_pyo3, nautilus_pyo3.OrderUpdated):
230232
return OrderUpdated.from_dict(order_event_pyo3.to_dict())
231233
elif isinstance(order_event_pyo3, nautilus_pyo3.OrderFilled):

0 commit comments

Comments
 (0)