From 2130f8acb9c2b02acff68460b6ad5e836d538c4f Mon Sep 17 00:00:00 2001 From: Giulio De Pasquale Date: Tue, 15 Dec 2020 20:18:31 +0000 Subject: [PATCH] added undefined state as default --- bfxbot/models.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) 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