blob: fa8481cca59711227a71497101b4de40b2dde7e0 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
|
"""Useful helper functions for interactin with :obj:`discord.Member` objects."""
import typing
from collections import abc
import discord
from pydis_core.utils import logging
log = logging.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.
Returns:
The :obj:`discord.Member` or :obj:`None` to indicate the member could not be found.
"""
if member := guild.get_member(member_id):
log.trace(f"{member} retrieved from cache.")
return member
try:
member = await guild.fetch_member(member_id)
except discord.errors.HTTPException as e:
log.trace(f"Failed to fetch {member_id} from API.")
if e.status in [400, 404]:
return None
raise
log.trace(f"{member} fetched from API.")
return member
async def handle_role_change(
member: discord.Member,
coro: typing.Callable[[discord.Role], abc.Coroutine],
role: discord.Role
) -> None:
"""
Await the given ``coro`` with ``role`` as the sole argument.
Handle errors that we expect to be raised from
:obj:`discord.Member.add_roles` and :obj:`discord.Member.remove_roles`.
Args:
member: The member that is being modified for logging purposes.
coro: This is intended to be :obj:`discord.Member.add_roles` or :obj:`discord.Member.remove_roles`.
role: The role to be passed to ``coro``.
"""
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}")
|
