From a7cdb2288dc2b6204934c1acd65bacf883501916 Mon Sep 17 00:00:00 2001 From: Giulio De Pasquale Date: Tue, 15 Dec 2020 20:16:56 +0000 Subject: [PATCH] modified signature of positionwrapper --- bfxbot/models.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/bfxbot/models.py b/bfxbot/models.py index af88b23..28f1fed 100644 --- a/bfxbot/models.py +++ b/bfxbot/models.py @@ -77,11 +77,12 @@ class Event: class PositionWrapper: - def __init__(self, position: Position, net_profit_loss: float = None, net_profit_loss_percentage: float = None): + def __init__(self, position: Position, state: PositionState = PositionState.LOSS, net_profit_loss: float = None, + net_profit_loss_percentage: float = None): self.position: Position = position self.__net_profit_loss: float = net_profit_loss self.__net_profit_loss_percentage: float = net_profit_loss_percentage - self.__state: PositionState = PositionState.LOSS + self.__state: PositionState = state def net_profit_loss(self) -> float: return self.__net_profit_loss