diff --git a/bfxapi/models/position.py b/bfxapi/models/position.py index e2e517b..5a6a4a8 100644 --- a/bfxapi/models/position.py +++ b/bfxapi/models/position.py @@ -3,7 +3,7 @@ Module used to describe all of the different data types """ class PositionModel: """ - Enum used to index the different values in a raw order array + Enum used to index the different values in a raw position array """ SYMBOL = 0 STATUS = 1 diff --git a/bfxapi/models/trade.py b/bfxapi/models/trade.py index a4325f3..79d2237 100644 --- a/bfxapi/models/trade.py +++ b/bfxapi/models/trade.py @@ -4,6 +4,21 @@ Module used to describe all of the different data types import datetime +class TradeModel: + """ + Enum used to index the different values in a raw trade array + """ + ID = 0 + PAIR = 1 + MTS_CREATE = 2 + ORDER_ID = 3 + EXEC_AMOUNT = 4 + EXEC_PRICE = 5 + ORDER_TYPE = 6 + ORDER_PRICE = 7 + MAKER = 8 + FEE = 9 + FEE_CURRENCY = 10 class Trade: """ @@ -47,7 +62,19 @@ class Trade: """ # [24224048, 'tBTCUSD', 1542800024000, 1151353484, 0.09399997, 19963, None, None, # -1, -0.000188, 'BTC'] - return Trade(*raw_trade) + tid = raw_trade[TradeModel.ID] + pair = raw_trade[TradeModel.PAIR] + mtsc = raw_trade[TradeModel.MTS_CREATE] + oid = raw_trade[TradeModel.ORDER_ID] + amnt = raw_trade[TradeModel.EXEC_AMOUNT] + price = raw_trade[TradeModel.EXEC_PRICE] + otype = raw_trade[TradeModel.ORDER_TYPE] + oprice = raw_trade[TradeModel.ORDER_PRICE] + maker = raw_trade[TradeModel.MAKER] + fee = raw_trade[TradeModel.FEE] + feeccy = raw_trade[TradeModel.FEE_CURRENCY] + return Trade(tid, pair, mtsc, oid, amnt, price, otype, oprice, maker, + fee, feeccy) def __str__(self): return "Trade '{}' x {} @ {} ".format(