Add close trades action

[perso/Immae/Projets/Cryptomonnaies/Cryptoportfolio/Trader.git] / portfolio.py

diff --git a/portfolio.py b/portfolio.py
index b77850bfbe1021b30e704dbd34b7c41d19ec5316..f27e84f0953200bf64606734f827ec30a2a25de6 100644 (file)
--- a/portfolio.py
+++ b/portfolio.py
@@ -290,6 +290,7 @@ class Trade:
         self.value_to = value_to
         self.orders = []
         self.market = market
+        self.closed = False
         assert self.value_from.value * self.value_to.value >= 0
         assert self.value_from.currency == self.value_to.currency
         if self.value_from != 0:
@@ -327,6 +328,13 @@ class Trade:
         else:
             return "long"
 
+    @property
+    def pending(self):
+        return not (self.is_fullfiled or self.closed)
+
+    def close(self):
+        self.closed = True
+
     @property
     def is_fullfiled(self):
         return abs(self.filled_amount(in_base_currency=True)) >= abs(self.delta)