import typing import discord from botcore.loggers import get_logger log = get_logger(__name__) async def get_or_fetch_member(guild: discord.Guild, member_id: int) -> typing.Optional[discord.Member]: """ Attempt to get a member from cache; on failure fetch from the API. Return `None` to indicate the member could not be found. """ if member := guild.get_member(member_id): log.trace(f"{member} retrieved from cache.") else: try: member = await guild.fetch_member(member_id) except discord.errors.NotFound: log.trace(f"Failed to fetch {member_id} from API.") return None log.trace(f"{member} fetched from API.") return member async def handle_role_change( member: discord.Member, coro: typing.Callable[..., typing.Coroutine], role: discord.Role ) -> None: """ Change `member`'s cooldown role via awaiting `coro` and handle errors. `coro` is intended to be `discord.Member.add_roles` or `discord.Member.remove_roles`. """ try: await coro(role) except discord.NotFound: log.error(f"Failed to change role for {member} ({member.id}): member not found") except discord.Forbidden: log.error( f"Forbidden to change role for {member} ({member.id}); " f"possibly due to role hierarchy" ) except discord.HTTPException as e: log.error(f"Failed to change role for {member} ({member.id}): {e.status} {e.code}")