From ece289b051180cdbd390c2c84682c7bdb6066c29 Mon Sep 17 00:00:00 2001 From: vveeps <54472340+vveeps@users.noreply.github.com> Date: Wed, 6 Oct 2021 23:01:22 +0300 Subject: [PATCH] add Track.__eq__ --- pomice/objects.py | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/pomice/objects.py b/pomice/objects.py index e86e203..49f1c87 100644 --- a/pomice/objects.py +++ b/pomice/objects.py @@ -30,6 +30,15 @@ class Track: self.is_seekable = info.get("isSeekable") self.position = info.get("position") + def __eq__(self, other): + if not isinstance(other, Track): + return False + + if self.ctx and other.ctx: + return other.track_id == self.track_id and other.ctx.message.id == self.ctx.message.id + + return other.track_id == self.track_id + def __str__(self): return self.title