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

Loading…
Cancel
Save