diff --git a/aurora/models.py b/aurora/models.py index 2c74015..be0c0e2 100644 --- a/aurora/models.py +++ b/aurora/models.py @@ -55,7 +55,7 @@ class Moderation(AuroraGuildModel): return await PartialUser.from_id(self.bot, self.moderator_id) async def get_target(self) -> Union["PartialUser", "PartialChannel"]: - if self.target_type == "user": + if self.target_type == "USER": return await PartialUser.from_id(self.bot, self.target_id) else: return await PartialChannel.from_id(self.bot, self.target_id) @@ -157,8 +157,8 @@ class PartialChannel(AuroraGuildModel): return cls(bot=bot, guild_id=channel.guild.id, id=channel.id, username=channel.name) except (NotFound, InvalidData, HTTPException, Forbidden) as e: if e == Forbidden: - return cls(bot=bot, guild_id=None, id=channel_id, name="Forbidden Channel") - return cls(bot=bot, guild_id=None, id=channel_id, name="Deleted Channel") + return cls(bot=bot, guild_id=0, id=channel_id, name="Forbidden Channel") + return cls(bot=bot, guild_id=0, id=channel_id, name="Deleted Channel") class PartialRole(AuroraGuildModel): id: int