matthew/enhanced-discord.py
1
0
Fork 0
Code Issues 4 Pull Requests 21 Projects Releases Wiki Activity

fix machine crashes?

Browse Source
This commit is contained in:
iDutchy 2021-01-05 19:15:24 -06:00
parent be0933c928
commit a0d491c71f
1 changed files with 10 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
discord/ext/commands/bot.py
View File

@ -830,13 +830,7 @@ class BotBase(GroupMixin):
if not isinstance(ret, str):
try:
if self.case_insensitive_prefix:
temp = []
for pre in ret:
temp += list(map(''.join, itertools.product(*((c.upper(), c.lower()) for c in pre))))
ret = temp
else:
ret = list(ret)
ret = list(ret)
except TypeError:
# It's possible that a generator raised this exception. Don't
# replace it with our own error if that's the case.
@ -850,7 +844,15 @@ class BotBase(GroupMixin):
raise ValueError("Iterable command_prefix must contain at least one prefix")
if self.case_insensitive_prefix:
return list(map(''.join, itertools.product(*((c.upper(), c.lower()) for c in ret))))
if isinstance(ret, list):
temp = []
for pre in ret:
if pre in (self.user.mention + ' ', '<@!%s> ' % self.user.id):
continue
temp += list(map(''.join, itertools.product(*((c.upper(), c.lower()) for c in pre))))
ret = temp
else:
ret = list(map(''.join, itertools.product(*((c.upper(), c.lower()) for c in ret))))
return ret
async def get_context(self, message, *, cls=Context):