CodeMirror/Bonfire
1
0
Fork 0
mirror of synced 2024-05-19 20:12:30 +12:00
Code Issues Releases Wiki Activity

Setup notifications toggling

Browse source
This commit is contained in:
Phxntxm 2016-07-10 13:35:02 -05:00
parent 5f7ab2e127
commit 160e9614ab
1 changed files with 50 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

52
cogs/twitch.py
View file

@ -88,7 +88,7 @@ class Twitch:
config.closeConnection()
@twitch.command(name='remove',aliases=['delete'],pass_context=True,no_pm=True)
async def remove_twitch_url(self,ctx):
async def remove_twitch_url(self, ctx):
"""Removes your twitch URL"""
cursor = config.getCursor()
cursor.execute('use {}'.format(config.db_default))
@ -99,9 +99,57 @@ class Twitch:
await self.bot.say("I am no longer saving your twitch URL {}".format(ctx.message.author.mention))
config.closeConnection()
else:
await self.bot.say("I do not have your twitch URL added {}".format(ctx.message.author.mention))
await self.bot.say("I do not have your twitch URL added {}. You can save your twitch url with !twitch add".format(ctx.message.author.mention))
config.closeConnection()
@commands.group(pass_context=True, no_pm=True, invoke_without_command=True)
async def notify(self, ctx)
"""This can be used to turn notifications on or off"""
pass
@notify.command(name='on', aliases=['start,yes'], pass_context=True, no_pm=True)
async def notify_on(self, ctx):
"""Turns twitch notifications on"""
cursor = config.getCursor()
cursor.execute('use {}'.format(config.db_default))
cursor.execute('select notifications_on from twitch where user_id="{}"'.format(ctx.message.author.id))
result = cursor.fetchone()
if result is None:
await self.bot.say("I do not have your twitch URL added {}. You can save your twitch url with !twitch add".format(ctx.message.author.mention))
config.closeConnection()
return
elif result:
await self.bot.say("What do you want me to do, send two notifications? Not gonna happen {}".format(ctx.message.author.mention))
config.closeConnection()
return
else:
cursor.execute('update twitch set notifications_on=1 where user_id="{}"'.format(ctx.message.author.id))
await self.bot.say("I will notify if you go live {}, you'll get a bajillion followers I promise c:".format(ctx.message.author.mention))
config.closeConnection()
return
@notify.command(name='off', aliases=['stop,no'], pass_context=True, no_pm=True)
async def notify_off(self, ctx):
"""Turns twitch notifications off"""
cursor = config.getCursor()
cursor.execute('use {}'.format(config.db_default))
cursor.execute('select notifications_on from twitch where user_id="{}"'.format(ctx.message.author.id))
result = cursor.fetchone()
if result is None:
await self.bot.say("I do not have your twitch URL added {}. You can save your twitch url with !twitch add".format(ctx.message.author.mention))
config.closeConnection()
return
elif not result:
await self.bot.say("I am already set to not notify if you go live! Pay attention brah {}".format(ctx.message.author.mention))
config.closeConnection()
return
else:
cursor.execute('update twitch set notifications_on=0 where user_id="{}"'.format(ctx.message.author.id))
await self.bot.say("I will not notify if you go live anymore {}, are you going to stream some lewd stuff you don't want people to see?~".format(ctx.message.author.mention))
config.closeConnection()
return
def setup(bot):
bot.add_cog(Twitch(bot))
config.loop.create_task(checkChannels(bot))