30 lines
1,005 B
Python
30 lines
1,005 B
Python
|
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)
|