From 857fd4c4ad11207cc10659d3d08e88a917ceb292 Mon Sep 17 00:00:00 2001
From: A-Trash-Coder <gavynlamar@gmail.com>
Date: Wed, 20 May 2020 21:45:46 -0400
Subject: [PATCH] Add a __eq__ and __ne__ to models Guild and User

---
 flask_discord/models/guild.py | 6 ++++++
 flask_discord/models/user.py  | 6 ++++++
 2 files changed, 12 insertions(+)

diff --git a/flask_discord/models/guild.py b/flask_discord/models/guild.py
index 6069baf..fe4318b 100644
--- a/flask_discord/models/guild.py
+++ b/flask_discord/models/guild.py
@@ -35,6 +35,12 @@ class Guild(DiscordModelsBase):
     def __str__(self):
         return self.name
 
+    def __eq__(self, guild):
+        return isinstance(guild, Guild) and guild.id == self.id
+
+    def __ne__(self, guild):
+        return not self.__eq__(guild)
+
     @property
     def icon_url(self):
         """A property returning direct URL to the guild's icon. Returns None if guild has no icon set."""
diff --git a/flask_discord/models/user.py b/flask_discord/models/user.py
index 0d67362..8878b38 100644
--- a/flask_discord/models/user.py
+++ b/flask_discord/models/user.py
@@ -78,6 +78,12 @@ class User(DiscordModelsBase):
     def __str__(self):
         return f"{self.name}#{self.discriminator}"
 
+    def __eq__(self, user):
+        return isinstance(user, User) and user.id == self.id
+
+    def __ne__(self, user):
+        return not self.__eq__(user)
+
     @property
     def name(self):
         """An alias to the username attribute."""