advtech/selena
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

REF: Updated the twich module to hopefully fix alerts

Browse Source
This commit is contained in:
Dan 2024-07-03 11:43:28 -04:00
parent b0ed598cb2
commit ed69bcf300
2 changed files with 15 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
main.py
View File

@ -69,6 +69,7 @@ class Selena(discord.Client):
from modules.social.twitch import Twitch from modules.social.twitch import Twitch
twitch = Twitch(self) twitch = Twitch(self)
twitch.setup(self.tree) twitch.setup(self.tree)
self.twitch_module = twitch
logging.info("Twitch module loaded") logging.info("Twitch module loaded")
if config['modules']['update']['enabled']: if config['modules']['update']['enabled']:

14
modules/social/twitch.py
View File

@ -103,9 +103,20 @@ class Twitch:
embed.add_field(name="Game", value=channel_info['game_name'], inline=True) embed.add_field(name="Game", value=channel_info['game_name'], inline=True)
await channel.send(embed=embed) await channel.send(embed=embed)
async def is_channel_followed(self, channel_name):
conn = sqlite3.connect(self.db_path)
cursor = conn.cursor()
cursor.execute("SELECT 1 FROM twitch_channels WHERE channel_name = ?", (channel_name,))
result = cursor.fetchone()
conn.close()
return result is not None
def setup(self, tree: app_commands.CommandTree): def setup(self, tree: app_commands.CommandTree):
@tree.command(name="add_twitch_channel", description="Add a Twitch channel to monitor") @tree.command(name="add_twitch_channel", description="Add a Twitch channel to monitor")
async def add_twitch_channel_command(interaction: discord.Interaction, channel_name: str, alert_channel: discord.TextChannel): async def add_twitch_channel_command(interaction: discord.Interaction, channel_name: str, alert_channel: discord.TextChannel):
if await self.is_channel_followed(channel_name):
await interaction.response.send_message(embed=discord.Embed(description=f"Twitch channel {channel_name} is already being monitored.", color=discord.Color.orange()))
return
conn = sqlite3.connect(self.db_path) conn = sqlite3.connect(self.db_path)
cursor = conn.cursor() cursor = conn.cursor()
cursor.execute("INSERT INTO twitch_channels (channel_name, alert_channel_id) VALUES (?, ?)", (channel_name, alert_channel.id)) cursor.execute("INSERT INTO twitch_channels (channel_name, alert_channel_id) VALUES (?, ?)", (channel_name, alert_channel.id))
@ -115,6 +126,9 @@ class Twitch:
@tree.command(name="remove_twitch_channel", description="Remove a Twitch channel from monitoring") @tree.command(name="remove_twitch_channel", description="Remove a Twitch channel from monitoring")
async def remove_twitch_channel_command(interaction: discord.Interaction, channel_name: str): async def remove_twitch_channel_command(interaction: discord.Interaction, channel_name: str):
if not await self.is_channel_followed(channel_name):
await interaction.response.send_message(embed=discord.Embed(description=f"Twitch channel {channel_name} is not being monitored.", color=discord.Color.red()))
return
conn = sqlite3.connect(self.db_path) conn = sqlite3.connect(self.db_path)
cursor = conn.cursor() cursor = conn.cursor()
cursor.execute("DELETE FROM twitch_channels WHERE channel_name = ?", (channel_name,)) cursor.execute("DELETE FROM twitch_channels WHERE channel_name = ?", (channel_name,))