如何在discord.py重写过程中分配角色？

import discord
from discord.ext import commands

bot = commands.Bot(command_prefix='!')

@bot.event
async def on_ready():
    print('Bot is ready.')

@bot.event
async def on_member_join(member):
    # 新成员加入时触发
    role = discord.utils.get(member.guild.roles, name='角色名称') # 获取角色对象
    await member.add_roles(role) # 分配角色给成员
    print(f'{member.name} 加入了服务器并获得了角色。')

@bot.command()
async def assign_role(ctx, role_name: str):
    role = discord.utils.get(ctx.guild.roles, name=role_name) # 获取角色对象
    if role:
        await ctx.author.add_roles(role) # 分配角色给作者
        await ctx.send(f'你获得了 {role_name} 角色。')
    else:
        await ctx.send('找不到该角色。')

@bot.command()
async def remove_role(ctx, role_name: str):
    role = discord.utils.get(ctx.guild.roles, name=role_name) # 获取角色对象
    if role:
        await ctx.author.remove_roles(role) # 移除角色
        await ctx.send(f'你失去了 {role_name} 角色。')
    else:
        await ctx.send('找不到该角色。')

bot.run('YOUR_BOT_TOKEN')