CopyBot/Red-DiscordBot
1
0
Fork 0
mirror of https://github.com/Cog-Creators/Red-DiscordBot.git synced 2025-11-21 10:17:59 -05:00
Code Issues Packages Projects Releases Wiki Activity

[Streams] Add ability to exclude rerun streams from alerts (#2620)

Browse Source 
* [Streams] Add ability to exclude rerun streams from alerts

* [Docs] Changelog entries for contributions by EgonSpengler

* Changelog entry for #2620 [Streams] Ignore Reruns In Alerts
This commit is contained in:
Tyler Adam
2019-04-25 11:58:58 -04:00
committed by Will
parent 7e49ce9a7b
commit f28d6dff32
3 changed files with 34 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
redbot/cogs/streams/streams.py
View File

@@ -44,6 +44,7 @@ class Streams(commands.Cog):
"mention_here": False,
"live_message_mention": False,
"live_message_nomention": False,
"ignore_reruns": False,
}
role_defaults = {"mention": False}
@@ -461,6 +462,19 @@ class Streams(commands.Cog):
else:
await ctx.send(_("Notifications will no longer be deleted."))
@streamset.command(name="ignorereruns")
@commands.guild_only()
async def ignore_reruns(self, ctx: commands.Context):
"""Toggle excluding rerun streams from alerts."""
guild = ctx.guild
current_setting = await self.db.guild(guild).ignore_reruns()
if current_setting:
await self.db.guild(guild).ignore_reruns.set(False)
await ctx.send(_("Streams of type 'rerun' will be included in alerts."))
else:
await self.db.guild(guild).ignore_reruns.set(True)
await ctx.send(_("Streams of type 'rerun' will no longer send an alert."))
async def add_or_remove(self, ctx: commands.Context, stream):
if ctx.channel.id not in stream.channels:
stream.channels.append(ctx.channel.id)
@@ -524,7 +538,7 @@ class Streams(commands.Cog):
for stream in self.streams:
with contextlib.suppress(Exception):
try:
embed = await stream.is_online()
embed, is_rerun = await stream.is_online()
except OfflineStream:
if not stream._messages_cache:
continue
@@ -540,6 +554,9 @@ class Streams(commands.Cog):
continue
for channel_id in stream.channels:
channel = self.bot.get_channel(channel_id)
ignore_reruns = await self.db.guild(channel.guild).ignore_reruns()
if ignore_reruns and is_rerun:
continue
mention_str, edited_roles = await self._get_mention_str(channel.guild)
if mention_str: