Browse Source

[guild] use a defaultdict in by_category

pull/791/head
Jake 8 years ago
committed by GitHub
parent
commit
3e00e7fe8a
  1. 11
      discord/guild.py

11
discord/guild.py

@ -27,7 +27,7 @@ DEALINGS IN THE SOFTWARE.
import copy import copy
import asyncio import asyncio
from collections import namedtuple from collections import namedtuple, defaultdict
from . import utils from . import utils
from .role import Role from .role import Role
@ -332,17 +332,12 @@ class Guild(Hashable):
List[Tuple[Optional[:class:`CategoryChannel`], List[:class:`abc.GuildChannel`]]]: List[Tuple[Optional[:class:`CategoryChannel`], List[:class:`abc.GuildChannel`]]]:
The categories and their associated channels. The categories and their associated channels.
""" """
grouped = {} grouped = defaultdict(list)
for channel in self._channels.values(): for channel in self._channels.values():
if isinstance(channel, CategoryChannel): if isinstance(channel, CategoryChannel):
continue continue
try: grouped[channel.category_id].append(channel)
channels = grouped[channel.category_id]
except KeyError:
channels = grouped[channel.category_id] = []
channels.append(channel)
def key(t): def key(t):
k, v = t k, v = t

Loading…
Cancel
Save