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

REF: Made sure everything conforms to Flake8's standards

Browse Source 
FIX: Fixed the Permission issues with logger
FEAT: Changed how the XP system works. Now does a scaling curve with xp being 1-5
This commit is contained in:
Dan
2024-06-22 08:55:26 -04:00
parent 4d9214e378
commit 39603b1e06
12 changed files with 327 additions and 195 deletions
Download Patch File Download Diff File Expand all files Collapse all files

80
modules/social/twitch_module.py
View File

@@ -1,10 +1,12 @@
import asyncio
import logging
import discord
from discord import app_commands
from twitchAPI.twitch import Twitch
from twitchAPI.helper import first
import logging
from twitchAPI.twitch import Twitch
import config
import asyncio
from modules.data.db import get_connection
logger = logging.getLogger(__name__)
@@ -13,8 +15,7 @@ logger = logging.getLogger(__name__)
class TwitchModule:
def __init__(self, bot):
self.bot = bot
self.twitch = Twitch(config.TWITCH_CLIENT_ID,
config.TWITCH_CLIENT_SECRET)
self.twitch = Twitch(config.TWITCH_CLIENT_ID, config.TWITCH_CLIENT_SECRET)
self.bot.loop.create_task(self.authenticate_twitch())
self.bot.loop.create_task(self.check_live_streams())
self.add_commands()
@@ -26,15 +27,17 @@ class TwitchModule:
while True:
conn = get_connection()
c = conn.cursor()
c.execute('''
c.execute(
"""
CREATE TABLE IF NOT EXISTS live_status (
twitch_name TEXT PRIMARY KEY,
is_live BOOLEAN
)
''')
"""
)
conn.commit()
c.execute('SELECT twitch_name, discord_channel_id FROM followed_channels')
c.execute("SELECT twitch_name, discord_channel_id FROM followed_channels")
followed_channels = c.fetchall()
for twitch_name, discord_channel_id in followed_channels:
@@ -47,7 +50,10 @@ class TwitchModule:
streams = await first(self.twitch.get_streams(user_id=[user_id]))
is_live = streams is not None
c.execute('SELECT is_live FROM live_status WHERE twitch_name = ?', (twitch_name,))
c.execute(
"SELECT is_live FROM live_status WHERE twitch_name = ?",
(twitch_name,),
)
row = c.fetchone()
was_live = row[0] if row else False
@@ -61,18 +67,26 @@ class TwitchModule:
f"**Game:** {streams.game_name}\n"
f"**Viewers:** {streams.viewer_count}"
),
color=discord.Color.green()
color=discord.Color.green(),
)
embed.set_thumbnail(
url=streams.thumbnail_url.replace('{width}', '320').replace('{height}', '180')
url=streams.thumbnail_url.replace(
"{width}", "320"
).replace("{height}", "180")
)
await channel.send(embed=embed)
c.execute('INSERT OR REPLACE INTO live_status (twitch_name, is_live) VALUES (?, ?)', (twitch_name, is_live))
c.execute(
"INSERT OR REPLACE INTO live_status (twitch_name, is_live) VALUES (?, ?)",
(twitch_name, is_live),
)
conn.commit()
except Exception as e:
logger.error(f"Error checking live status for {twitch_name}: {e}", exc_info=True)
logger.error(
f"Error checking live status for {twitch_name}: {e}",
exc_info=True,
)
conn.close()
await asyncio.sleep(300) # Check every 5 minutes
@@ -80,15 +94,19 @@ class TwitchModule:
def add_commands(self):
@app_commands.command(
name="follow_twitch",
description="Follow a Twitch channel to get live alerts"
description="Follow a Twitch channel to get live alerts",
)
async def follow_twitch(interaction: discord.Interaction, twitch_name: str, channel: discord.TextChannel = None):
async def follow_twitch(
interaction: discord.Interaction,
twitch_name: str,
channel: discord.TextChannel = None,
):
channel = channel or interaction.channel
conn = get_connection()
c = conn.cursor()
c.execute(
'INSERT OR REPLACE INTO followed_channels (twitch_name, discord_channel_id) VALUES (?, ?)',
(twitch_name, channel.id)
"INSERT OR REPLACE INTO followed_channels (twitch_name, discord_channel_id) VALUES (?, ?)",
(twitch_name, channel.id),
)
conn.commit()
conn.close()
@@ -96,40 +114,34 @@ class TwitchModule:
embed=discord.Embed(
title="Followed Twitch Channel",
description=f"Now following {twitch_name}. Alerts will be sent to {channel.mention}.",
color=discord.Color.green()
color=discord.Color.green(),
)
)
logger.info(f"Now following {twitch_name} for alerts in {channel.name}")
@app_commands.command(
name="unfollow_twitch",
description="Unfollow a Twitch channel"
name="unfollow_twitch", description="Unfollow a Twitch channel"
)
async def unfollow_twitch(interaction: discord.Interaction, twitch_name: str):
conn = get_connection()
c = conn.cursor()
c.execute(
'DELETE FROM followed_channels WHERE twitch_name = ?',
(twitch_name,)
)
c.execute(
'DELETE FROM live_status WHERE twitch_name = ?',
(twitch_name,)
"DELETE FROM followed_channels WHERE twitch_name = ?", (twitch_name,)
)
c.execute("DELETE FROM live_status WHERE twitch_name = ?", (twitch_name,))
conn.commit()
conn.close()
await interaction.response.send_message(
embed=discord.Embed(
title="Unfollowed Twitch Channel",
description=f"No longer following {twitch_name}.",
color=discord.Color.red()
color=discord.Color.red(),
)
)
logger.info(f"No longer following {twitch_name}")
@app_commands.command(
name="twitch_live",
description="Check if a Twitch streamer is live"
name="twitch_live", description="Check if a Twitch streamer is live"
)
async def twitch_live(interaction: discord.Interaction, streamer: str):
await interaction.response.defer()
@@ -141,7 +153,7 @@ class TwitchModule:
embed=discord.Embed(
title="Twitch Live Check",
description=f"Streamer {streamer} not found.",
color=discord.Color.red()
color=discord.Color.red(),
)
)
logger.info(f"Streamer {streamer} not found.")
@@ -158,10 +170,12 @@ class TwitchModule:
f"**Game:** {streams.game_name}\n"
f"**Viewers:** {streams.viewer_count}"
),
color=discord.Color.green()
color=discord.Color.green(),
)
embed.set_thumbnail(
url=streams.thumbnail_url.replace('{width}', '320').replace('{height}', '180')
url=streams.thumbnail_url.replace("{width}", "320").replace(
"{height}", "180"
)
)
await interaction.followup.send(embed=embed)
logger.info(f"Streamer {streamer} is live.")
@@ -170,7 +184,7 @@ class TwitchModule:
embed=discord.Embed(
title=f"{streamer} is not live",
description=f"{streamer} is currently offline.",
color=discord.Color.red()
color=discord.Color.red(),
)
)
logger.info(f"Streamer {streamer} is offline.")