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."""