import discord


class Role:
    async def get_role(self, interaction, role):
        fetched_role = None
        for r in interaction.guild.roles:
            if r.name == role:
                fetched_role = r
        return fetched_role

    async def add_role(self, interaction, role):
        print(f"Fetching {role}")
        fetched_role = await self.get_role(interaction, role)
        print(f"Fetched {fetched_role.name}")
        if fetched_role is not None:
            print(f"fetched_role is not None")
            member = interaction.user
            print(member)
            if fetched_role not in member.roles:
                await member.add_roles(fetched_role)
            print(member.roles)

    async def remove_role(self, interaction, role):
        fetched_role = await self.get_role(interaction, role)
        if fetched_role is not None:
            member = interaction.user
            if fetched_role in member.roles:
                await member.remove_roles(fetched_role)