REF: Removed the Twitch Module for now
This commit is contained in:
Dan
parent
bd318a7815
commit
6997d96dbd
2
.gitignore
vendored
2
.gitignore
vendored
@ -163,3 +163,5 @@ cython_debug/
|
||||
/modules/user/__init__.py
|
||||
/modules/__init__.py
|
||||
/data/selena.db
|
||||
/modules/social/__init__.py
|
||||
/clear_commands.py
|
||||
@ -6,18 +6,12 @@ load_dotenv()
|
||||
config = {
|
||||
'DISCORD_TOKEN': os.getenv('DISCORD_TOKEN'),
|
||||
'GUILD_ID': int(os.getenv('DISCORD_GUILD_ID')),
|
||||
'DISCORD_CHANNEL_ID': int(os.getenv('DISCORD_CHANNEL_ID')),
|
||||
'TWITCH_CLIENT_ID': os.getenv('TWITCH_CLIENT_ID'),
|
||||
'TWITCH_CLIENT_SECRET': os.getenv('TWITCH_CLIENT_SECRET'),
|
||||
'modules': {
|
||||
'currency': {
|
||||
'enabled': True
|
||||
},
|
||||
'xp': {
|
||||
'enabled': True
|
||||
},
|
||||
'twitch': {
|
||||
'enabled': True
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
25
main.py
25
main.py
@ -1,21 +1,31 @@
|
||||
import discord
|
||||
from config import config
|
||||
import logging
|
||||
|
||||
logging.basicConfig(level=logging.INFO)
|
||||
|
||||
TOKEN = config['DISCORD_TOKEN']
|
||||
GUILD_ID = config['GUILD_ID']
|
||||
DISCORD_CHANNEL_ID = config['DISCORD_CHANNEL_ID']
|
||||
|
||||
intents = discord.Intents.default()
|
||||
intents.messages = True
|
||||
guild = discord.Object(id=GUILD_ID)
|
||||
|
||||
|
||||
class Selena(discord.Client):
|
||||
def __init__(self):
|
||||
super().__init__(intents=intents)
|
||||
self.tree = discord.app_commands.CommandTree(self)
|
||||
self.twitch = None
|
||||
self.load_modules()
|
||||
|
||||
async def setup_hook(self):
|
||||
logging.info("Setting up modules...")
|
||||
if self.twitch:
|
||||
logging.info("Setting up Twitch module asynchronously...")
|
||||
await self.twitch.setup_hook()
|
||||
self.tree.copy_global_to(guild=discord.Object(id=GUILD_ID))
|
||||
await self.tree.sync(guild=discord.Object(id=GUILD_ID))
|
||||
|
||||
def load_modules(self):
|
||||
if config['modules']['currency']['enabled']:
|
||||
from modules.user.currency import Currency
|
||||
@ -27,21 +37,12 @@ class Selena(discord.Client):
|
||||
xp = XP(self)
|
||||
xp.setup(self.tree)
|
||||
|
||||
if config['modules']['twitch']['enabled']:
|
||||
from modules.social.twitch import Twitch
|
||||
twitch = Twitch(self)
|
||||
twitch.setup(self.tree)
|
||||
|
||||
async def setup_hook(self):
|
||||
self.tree.copy_global_to(guild=guild)
|
||||
await self.tree.sync(guild=guild)
|
||||
|
||||
|
||||
bot = Selena()
|
||||
|
||||
|
||||
@bot.event
|
||||
async def on_ready():
|
||||
print(f'{bot.user.name} has connected to Discord!')
|
||||
logging.info(f'Bot {bot.user.name} has connected to Discord!')
|
||||
|
||||
bot.run(TOKEN)
|
||||
|
||||
@ -1,127 +0,0 @@
|
||||
import discord
|
||||
from discord import app_commands
|
||||
import sqlite3
|
||||
import aiohttp
|
||||
import asyncio
|
||||
from config import config
|
||||
from datetime import datetime, timedelta
|
||||
|
||||
|
||||
class Twitch:
|
||||
def __init__(self, bot):
|
||||
self.bot = bot
|
||||
self.db_path = 'data/selena.db'
|
||||
self.client_id = config['TWITCH_CLIENT_ID']
|
||||
self.client_secret = config['TWITCH_CLIENT_SECRET']
|
||||
self._init_db()
|
||||
self.live_channels = set()
|
||||
self.token = None
|
||||
self.token_expiry = datetime.utcnow()
|
||||
|
||||
def _init_db(self):
|
||||
with sqlite3.connect(self.db_path) as conn:
|
||||
cursor = conn.cursor()
|
||||
cursor.execute('''
|
||||
CREATE TABLE IF NOT EXISTS twitch_channels (
|
||||
channel_name TEXT PRIMARY KEY
|
||||
)
|
||||
''')
|
||||
conn.commit()
|
||||
|
||||
async def fetch_twitch_token(self):
|
||||
if self.token and datetime.utcnow() < self.token_expiry:
|
||||
return self.token
|
||||
|
||||
async with aiohttp.ClientSession() as session:
|
||||
async with session.post('https://id.twitch.tv/oauth2/token', params={
|
||||
'client_id': self.client_id,
|
||||
'client_secret': self.client_secret,
|
||||
'grant_type': 'client_credentials'
|
||||
}) as response:
|
||||
data = await response.json()
|
||||
self.token = data['access_token']
|
||||
self.token_expiry = datetime.utcnow() + timedelta(seconds=data['expires_in'])
|
||||
return self.token
|
||||
|
||||
async def fetch_twitch_channel(self, channel_name):
|
||||
token = await self.fetch_titch_token()
|
||||
async with aiohttp.ClientSession() as session:
|
||||
async with session.get('https://api.twitch.tv/helix/users', headers={
|
||||
'Client-ID': self.client_id,
|
||||
'Authorization': f'Bearer {token}'
|
||||
}, params={'login': channel_name}) as response:
|
||||
data = await response.json()
|
||||
if data['data']:
|
||||
return data['data'][0]
|
||||
return None
|
||||
|
||||
async def fetch_stream_status(self, user_id):
|
||||
token = await self.fetch_twitch_token()
|
||||
async with aiohttp.ClientSession() as session:
|
||||
async with session.get('https://api.twitch.tv/helix/streams', headers={
|
||||
'Client-ID': self.client_id,
|
||||
'Authorization': f'Bearer {token}'
|
||||
}, params={'user_id': user_id}) as response:
|
||||
data = await response.json()
|
||||
if data['data']:
|
||||
return data['data'][0]
|
||||
return None
|
||||
|
||||
async def check_live_status(self):
|
||||
with sqlite3.connect(self.db_path) as conn:
|
||||
cursor = conn.cursor()
|
||||
cursor.execute('SELECT channel_name FROM twitch_channels')
|
||||
channels = cursor.fetchall()
|
||||
|
||||
for (channel_name,) in channels:
|
||||
channel = await self.fetch_twitch_channel(channel_name)
|
||||
if channel:
|
||||
stream = await self.fetch_stream_status(channel['id'])
|
||||
if stream and channel_name not in self.live_channels:
|
||||
self.live_channels.add(channel_name)
|
||||
discord_channel = self.bot.get_channel(config['DISCORD_CHANNEL_ID'])
|
||||
await discord_channel.send(f'{channel_name} is now live on Twitch!')
|
||||
elif not stream and channel_name in self.live_channels:
|
||||
self.live_channels.remove(channel_name)
|
||||
|
||||
@app_commands.command(name='add_twitch_channel', description='Follow a new Twitch channel')
|
||||
async def add_twitch_channel(self, interaction: discord.Interaction, channel_name: str):
|
||||
with sqlite3.connect(self.db_path) as conn:
|
||||
cursor = conn.cursor()
|
||||
cursor.execute('INSERT OR IGNORE INTO twitch_channels (channel_name) VALUES (?)', (channel_name,))
|
||||
conn.commit()
|
||||
await interaction.response.send_message(f'Added {channel_name} to the followed Twitch channels.')
|
||||
|
||||
@app_commands.command(name='remove_twitch_channel', description='Unfollow a Twitch channel')
|
||||
async def remove_twitch_channel(self, interaction: discord.Interaction, channel_name: str):
|
||||
with sqlite3.connect(self.db_path) as conn:
|
||||
cursor = conn.cursor()
|
||||
cursor.execute('DELETE FROM twitch_channels WHERE channel_name = ?', (channel_name,))
|
||||
conn.commit()
|
||||
await interaction.response.send_message(f'Removed {channel_name} from the followed Twitch channels.')
|
||||
|
||||
@app_commands.command(name='check_twitch_channel', description='Check if a Twitch channel is live')
|
||||
async def check_twitch_channel(self, interaction: discord.Interaction, channel_name: str):
|
||||
channel = await self.fetch_twitch_channel(channel_name)
|
||||
if channel:
|
||||
stream = await self.fetch_stream_status(channel['id'])
|
||||
if stream:
|
||||
await interaction.response.send_message(f'{channel_name} is currently live on Twitch!')
|
||||
else:
|
||||
await interaction.response.send_message(f'{channel_name} is not live on Twitch.')
|
||||
else:
|
||||
await interaction.response.send_message(f'Could not find Twitch channel: {channel_name}')
|
||||
|
||||
async def start_checking_live_status(self):
|
||||
while True:
|
||||
await self.check_live_status()
|
||||
await asyncio.sleep(60)
|
||||
|
||||
def setup(self, tree):
|
||||
tree.add_command(self.add_twitch_channel)
|
||||
tree.add_command(self.remove_twitch_channel)
|
||||
tree.add_command(self.check_twitch_channel)
|
||||
|
||||
|
||||
def setup(bot):
|
||||
bot.tree.add_cog(Twitch(bot))
|
||||
Loading…
x
Reference in New Issue
Block a user