CodeMirror/Bonfire
1
0
Fork 0
mirror of synced 2024-09-29 08:51:53 +13:00
Code Issues Releases Wiki Activity

Corrected list comprehension to remove a poll from the list

Browse source
This commit is contained in:
phxntxm 2016-09-28 19:46:21 -05:00
parent d6dfa7e5c0
commit 0ca396cc1f
1 changed files with 7 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
cogs/strawpoll.py
View file

@ -137,8 +137,13 @@ class Strawpoll:
"""This command can be used to delete one of the existing strawpolls"""
r_filter = {'server_id': ctx.message.server.id}
content = await config.get_content('strawpolls', r_filter)
content = content[0]['polls']
polls = [{key: result} for key, result in content.items() if poll_id != result['poll_id']]
try:
content = content[0]['polls']
except TypeError:
await self.bot.say("There are no strawpolls setup on this server!")
return
polls = [poll for poll in content if poll['poll_id'] != poll_id]
update = {'polls': polls}
# Try to remove the poll based on the ID, if it doesn't exist, this will return false