Added the ability to change which channel is notified whne you go live
This commit is contained in:
parent
c3d66e60b4
commit
6538ab0651
1 changed files with 20 additions and 7 deletions
|
@ -30,7 +30,7 @@ class Picarto:
|
||||||
async def check_channels(self):
|
async def check_channels(self):
|
||||||
await self.bot.wait_until_ready()
|
await self.bot.wait_until_ready()
|
||||||
while not self.bot.is_closed:
|
while not self.bot.is_closed:
|
||||||
picarto = config.getContent('picarto')
|
picarto = config.getContent('picarto') or {}
|
||||||
for m_id, r in picarto.items():
|
for m_id, r in picarto.items():
|
||||||
url = r['picarto_url']
|
url = r['picarto_url']
|
||||||
live = r['live']
|
live = r['live']
|
||||||
|
@ -108,7 +108,7 @@ class Picarto:
|
||||||
picarto_urls[ctx.message.author.id]['picarto_url'] = url
|
picarto_urls[ctx.message.author.id]['picarto_url'] = url
|
||||||
else:
|
else:
|
||||||
picarto_urls[ctx.message.author.id] = {'picarto_url': url, 'server_id': ctx.message.server.id,
|
picarto_urls[ctx.message.author.id] = {'picarto_url': url, 'server_id': ctx.message.server.id,
|
||||||
'notifications_on': 1, 'live': 0}
|
'channel_id': ctx.message.channel.id, 'notifications_on': 1, 'live': 0}
|
||||||
config.saveContent('picarto', picarto_urls)
|
config.saveContent('picarto', picarto_urls)
|
||||||
await self.bot.say("I have just saved your Picarto url {}".format(ctx.message.author.mention))
|
await self.bot.say("I have just saved your Picarto url {}".format(ctx.message.author.mention))
|
||||||
|
|
||||||
|
@ -116,7 +116,7 @@ class Picarto:
|
||||||
@checks.customPermsOrRole(send_messages=True)
|
@checks.customPermsOrRole(send_messages=True)
|
||||||
async def remove_picarto_url(self, ctx):
|
async def remove_picarto_url(self, ctx):
|
||||||
"""Removes your picarto URL"""
|
"""Removes your picarto URL"""
|
||||||
picarto = config.getContent('picarto')
|
picarto = config.getContent('picarto') or {}
|
||||||
if picarto.get(ctx.message.author.id) is not None:
|
if picarto.get(ctx.message.author.id) is not None:
|
||||||
del picarto[ctx.message.author.id]
|
del picarto[ctx.message.author.id]
|
||||||
config.saveContent('picarto', picarto)
|
config.saveContent('picarto', picarto)
|
||||||
|
@ -128,15 +128,28 @@ class Picarto:
|
||||||
|
|
||||||
@picarto.group(pass_context=True, no_pm=True, invoke_without_command=True)
|
@picarto.group(pass_context=True, no_pm=True, invoke_without_command=True)
|
||||||
@checks.customPermsOrRole(send_messages=True)
|
@checks.customPermsOrRole(send_messages=True)
|
||||||
async def notify(self, ctx):
|
async def notify(self, ctx, channel: discord.channel=None):
|
||||||
"""This can be used to turn picarto notifications on or off"""
|
"""This can be used to turn picarto notifications on or off"""
|
||||||
pass
|
channel = channel or ctx.message.channel
|
||||||
|
server = ctx.message.server
|
||||||
|
member = ctx.message.author
|
||||||
|
|
||||||
|
picarto = config.getContent('picarto') or {}
|
||||||
|
result = picarto.get(member.id)
|
||||||
|
if result is None:
|
||||||
|
await self.bot.say(
|
||||||
|
"I do not have your picarto URL added {}. You can save your picarto url with !picarto add".format(
|
||||||
|
member.mention))
|
||||||
|
|
||||||
|
picarto[member.id]['channel_id'] = channel.id
|
||||||
|
config.saveContent('picarto', picarto)
|
||||||
|
await self.bot.say("I have just changed which channel will be notified when you go live, to {}".format(channel.name))
|
||||||
|
|
||||||
@notify.command(name='on', aliases=['start,yes'], pass_context=True, no_pm=True)
|
@notify.command(name='on', aliases=['start,yes'], pass_context=True, no_pm=True)
|
||||||
@checks.customPermsOrRole(send_messages=True)
|
@checks.customPermsOrRole(send_messages=True)
|
||||||
async def notify_on(self, ctx):
|
async def notify_on(self, ctx):
|
||||||
"""Turns picarto notifications on"""
|
"""Turns picarto notifications on"""
|
||||||
picarto = config.getContent('picarto')
|
picarto = config.getContent('picarto') or {}
|
||||||
result = picarto.get(ctx.message.author.id)
|
result = picarto.get(ctx.message.author.id)
|
||||||
if result is None:
|
if result is None:
|
||||||
await self.bot.say(
|
await self.bot.say(
|
||||||
|
@ -155,7 +168,7 @@ class Picarto:
|
||||||
@checks.customPermsOrRole(send_messages=True)
|
@checks.customPermsOrRole(send_messages=True)
|
||||||
async def notify_off(self, ctx):
|
async def notify_off(self, ctx):
|
||||||
"""Turns picarto notifications off"""
|
"""Turns picarto notifications off"""
|
||||||
picarto = config.getContent('picarto')
|
picarto = config.getContent('picarto') or {}
|
||||||
if picarto.get(ctx.message.author.id) is None:
|
if picarto.get(ctx.message.author.id) is None:
|
||||||
await self.bot.say(
|
await self.bot.say(
|
||||||
"I do not have your picarto URL added {}. You can save your picarto url with !picarto add".format(
|
"I do not have your picarto URL added {}. You can save your picarto url with !picarto add".format(
|
||||||
|
|
Loading…
Reference in a new issue