CodeMirror/Bonfire
1
0
Fork 0
mirror of synced 2024-06-29 03:30:57 +12:00
Code Issues Releases Wiki Activity

Corrected error where list was being checked, instead of the entry in the list

Browse source
This commit is contained in:
phxntxm 2016-09-28 18:57:13 -05:00
parent 33739621f5
commit 8502c04acd
1 changed files with 10 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
cogs/twitch.py
View file

@ -105,6 +105,7 @@ class Twitch:
await self.bot.say("{} has not saved their twitch URL yet!".format(member.name))
return
result = result[0]
url = result['twitch_url']
user = re.search("(?<=twitch.tv/)(.*)", url).group(1)
twitch_url = "https://api.twitch.tv/kraken/channels/{}?client_id={}".format(user, self.key)
@ -148,16 +149,17 @@ class Twitch:
return
r_filter = {'member_id': ctx.message.author.id}
result = await config.get_content('twitch', r_filter)
entry = {'twitch_url': url,
'servers': [ctx.message.server.id],
'notifications_on': 1,
'live': 0}
update = {'twitch_url': url}
# Check to see if this user has already saved a twitch URL
# If they have, update the URL, otherwise create a new entry
# Assuming they're not live, and notifications should be on
if result is not None:
await config.update_content('twitch', {'twitch_url': url}, r_filter)
else:
await config.add_content('twitch', {'twitch_url': url, 'servers': [ctx.message.server.id],
'notifications_on': 1, 'live': 0}, r_filter)
if not await config.add_content('twitch', entry, r_filter):
await config.update_content('twitch', update, r_filter)
await self.bot.say("I have just saved your twitch url {}".format(ctx.message.author.mention))
@twitch.command(name='remove', aliases=['delete'], pass_context=True, no_pm=True)
@ -165,7 +167,8 @@ class Twitch:
async def remove_twitch_url(self, ctx):
"""Removes your twitch URL"""
# Just try to remove it, if it doesn't exist, nothing is going to happen
await config.remove_content('twitch', {'member_id': ctx.message.author.id})
r_filter = {'member_id': ctx.message.author.id}
await config.remove_content('twitch', r_filter)
await self.bot.say("I am no longer saving your twitch URL {}".format(ctx.message.author.mention))
@twitch.group(pass_context=True, no_pm=True, invoke_without_command=True)