define candle eq

StoicPerlman · Jun 16, 2018 · 7689bee · 7689bee
1 parent f0623f7
commit 7689bee
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/binance_db/candle.py b/binance_db/candle.py
@@ -35,12 +35,17 @@ def __init__(self, pair, kline):
         self.trades = kline[rest.TRADES]
         self.tbbav = float(kline[rest.TBBAV])
         self.tbqav = float(kline[rest.TBQAV])
-
+
+    def __eq__(self, other):
+        if other == None:
+            return False
+        return self.pair == other.pair and self.close_time == other.close_time
+
     def __repr__(self):
         date = self.open_time.strftime('%Y-%m-%d %H:%M:%S')
         return "<Candle(pair={}, open_time={}, open={}, close={})>".format(
                 self.pair, date, self.open_price, self.close_price)
-    
+
     @staticmethod
     def to_date(timestamp):
         return datetime.utcfromtimestamp(timestamp / 1000)