Add a __eq__ and __ne__ to models Guild and User

This commit is contained in:
A-Trash-Coder
2020-05-20 21:45:46 -04:00
parent 4a2440ea53
commit 857fd4c4ad
2 changed files with 12 additions and 0 deletions

View File

@@ -35,6 +35,12 @@ class Guild(DiscordModelsBase):
def __str__(self): def __str__(self):
return self.name 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 @property
def icon_url(self): def icon_url(self):
"""A property returning direct URL to the guild's icon. Returns None if guild has no icon set.""" """A property returning direct URL to the guild's icon. Returns None if guild has no icon set."""

View File

@@ -78,6 +78,12 @@ class User(DiscordModelsBase):
def __str__(self): def __str__(self):
return f"{self.name}#{self.discriminator}" 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 @property
def name(self): def name(self):
"""An alias to the username attribute.""" """An alias to the username attribute."""