diff --git a/bfxbot/models.py b/bfxbot/models.py index 28f1fed..94f98aa 100644 --- a/bfxbot/models.py +++ b/bfxbot/models.py @@ -42,7 +42,8 @@ class PositionState(Enum): LOSS = 0, BREAK_EVEN = 1, MINIMUM_PROFIT = 2, - PROFIT = 3 + PROFIT = 3, + UNDEFINED = 4 def color(self) -> str: if self == self.LOSS or self == self.CRITICAL: @@ -77,7 +78,8 @@ class Event: class PositionWrapper: - def __init__(self, position: Position, state: PositionState = PositionState.LOSS, net_profit_loss: float = None, + def __init__(self, position: Position, state: PositionState = PositionState.UNDEFINED, + net_profit_loss: float = None, net_profit_loss_percentage: float = None): self.position: Position = position self.__net_profit_loss: float = net_profit_loss