Total Complexity | 4 |
Total Lines | 53 |
Duplicated Lines | 0 % |
Changes | 0 |
1 | # Copyright Pincer 2021-Present |
||
2 | # Full MIT License can be found in `LICENSE` at the project root. |
||
3 | |||
4 | """sent when a guild role was updated.""" |
||
5 | |||
6 | from __future__ import annotations |
||
7 | |||
8 | from typing import TYPE_CHECKING |
||
9 | |||
10 | from ..objects.events.guild import GuildRoleUpdateEvent |
||
11 | from ..utils import Coro |
||
12 | |||
13 | if TYPE_CHECKING: |
||
14 | from ..client import Client |
||
15 | from ..core.gateway import Gateway |
||
16 | from ..core.gateway import GatewayDispatch |
||
17 | |||
18 | |||
19 | async def guild_role_update_middleware( |
||
20 | self: Client, gateway: Gateway, payload: GatewayDispatch |
||
21 | ): |
||
22 | """|coro| |
||
23 | |||
24 | Middleware for the ``on_guild_role_update`` event. |
||
25 | |||
26 | Parameters |
||
27 | ---------- |
||
28 | payload : :class:`~pincer.core.gateway.GatewayDispatch` |
||
29 | The data received from the guild role update event. |
||
30 | gateway : :class:`~pincer.core.gateway.Gateway` |
||
31 | The gateway for the current shard. |
||
32 | |||
33 | Returns |
||
34 | ------- |
||
35 | Tuple[:class:`str`, :class:`~pincer.objects.events.guild.GuildRoleUpdateEvent`] |
||
36 | ``on_guild_role_update`` and a ``GuildRoleUpdateEvent`` |
||
37 | """ |
||
38 | |||
39 | event = GuildRoleUpdateEvent.from_dict(payload.data) |
||
40 | guild = self.guilds.get(event.guild_id) |
||
41 | |||
42 | if guild: |
||
43 | guild.roles = [ |
||
44 | role if role.id != event.role.id else event.role |
||
45 | for role in guild.roles |
||
46 | ] |
||
47 | |||
48 | return ("on_guild_role_update", event) |
||
49 | |||
50 | |||
51 | def export() -> Coro: |
||
52 | return guild_role_update_middleware |
||
53 |