feat: Add misandrist insults and big CS2 events matches fetch from HLTV.org
This commit is contained in:
239
main.py
239
main.py
@@ -1,164 +1,129 @@
|
||||
import os, discord, json, sys, random
|
||||
import modules.setup.db as db
|
||||
from argparse import ArgumentParser
|
||||
from datetime import datetime
|
||||
from asyncio import create_task
|
||||
from dotenv import load_dotenv
|
||||
from discord import option
|
||||
from discord.ext import tasks, commands
|
||||
from modules.setup.bot import bot, reload_feature
|
||||
from modules.setup.logger import LOGGER
|
||||
from modules.cogs.productivity import TASKS as PRODUCTIVITY_TASKS, send_reminder
|
||||
|
||||
from setup.bot import bot, reload_feature
|
||||
from setup.logger import LOGGER
|
||||
PARSER = ArgumentParser(description='BotMafieux for Discord.')
|
||||
PARSER.add_argument('--guild_id',type=int,help='The guild ID.')
|
||||
PARSER.add_argument('--channel_id',type=int,help='The channel ID.')
|
||||
PARSER.add_argument('-r','--reload',action='store_true',help='Reload the bot.')
|
||||
EXEC_ARGS = PARSER.parse_args()
|
||||
|
||||
@tasks.loop(time=datetime.time(datetime.strptime('00:00','%H:%M')))
|
||||
async def birthday_anouncements_task():
|
||||
LOGGER.debug('birthday_anouncements_task started.')
|
||||
for guild in bot.guilds:
|
||||
birthday_settings = db.select('birthday_settings',f'guild_id = {guild.id}')
|
||||
if birthday_settings:
|
||||
if birthday_settings['is_enabled']:
|
||||
for user in guild.members:
|
||||
user_productivity_data = db.select('guild_user_productivity',f'user_id = {user.id} AND guild_id = {guild.id}')
|
||||
if user_productivity_data:
|
||||
if user_productivity_data['is_enabled']:
|
||||
user = guild.get_member(user.id)
|
||||
if user:
|
||||
birthday = user_productivity_data['birthday']
|
||||
today = datetime.today().strftime('%d/%m')
|
||||
if birthday == today:
|
||||
channel = guild.get_channel(user_productivity_data['channel_id'])
|
||||
if channel:
|
||||
message = user_productivity_data['birthday_message']
|
||||
if message:
|
||||
message = message.replace('{user}',user.mention)
|
||||
await channel.send(message)
|
||||
else:
|
||||
LOGGER.debug(f'Channel {user_productivity_data["channel_id"]} not found.')
|
||||
else:
|
||||
LOGGER.debug(f'No birthday today for user {user.name} ({user.id}).')
|
||||
else:
|
||||
LOGGER.debug(f'User {user.name} ({user.id}) not found.')
|
||||
else:
|
||||
LOGGER.debug(f'User {user.name} ({user.id}) has birthday announcements disabled.')
|
||||
else:
|
||||
LOGGER.debug(f'No birthday data found for user {user.name} ({user.id}).')
|
||||
else:
|
||||
LOGGER.debug(f'No productivity settings found for guild {guild.name} ({guild.id}).')
|
||||
LOGGER.debug('birthday_anouncements_task ended.')
|
||||
# @tasks.loop(time=datetime.time(datetime.strptime('00:00','%H:%M')))
|
||||
# async def birthday_anouncements_task():
|
||||
# LOGGER.debug('birthday_anouncements_task started.')
|
||||
# for guild in bot.guilds:
|
||||
# birthday_settings = db.select('birthday_settings',f'guild_id = {guild.id}')
|
||||
# if birthday_settings:
|
||||
# if birthday_settings['is_enabled']:
|
||||
# for user in guild.members:
|
||||
# user_productivity_data = db.select('guild_user_productivity',f'user_id = {user.id} AND guild_id = {guild.id}')
|
||||
# if user_productivity_data:
|
||||
# if user_productivity_data['is_enabled']:
|
||||
# user = guild.get_member(user.id)
|
||||
# if user:
|
||||
# birthday = user_productivity_data['birthday']
|
||||
# today = datetime.today().strftime('%d/%m')
|
||||
# if birthday == today:
|
||||
# channel = guild.get_channel(user_productivity_data['channel_id'])
|
||||
# if channel:
|
||||
# message = user_productivity_data['birthday_message']
|
||||
# if message:
|
||||
# message = message.replace('{user}',user.mention)
|
||||
# await channel.send(message)
|
||||
# else:
|
||||
# LOGGER.debug(f'Channel {user_productivity_data["channel_id"]} not found.')
|
||||
# else:
|
||||
# LOGGER.debug(f'No birthday today for user {user.name} ({user.id}).')
|
||||
# else:
|
||||
# LOGGER.debug(f'User {user.name} ({user.id}) not found.')
|
||||
# else:
|
||||
# LOGGER.debug(f'User {user.name} ({user.id}) has birthday announcements disabled.')
|
||||
# else:
|
||||
# LOGGER.debug(f'No birthday data found for user {user.name} ({user.id}).')
|
||||
# else:
|
||||
# LOGGER.debug(f'No productivity settings found for guild {guild.name} ({guild.id}).')
|
||||
# LOGGER.debug('birthday_anouncements_task ended.')
|
||||
|
||||
@bot.slash_command(name='help',description='Displays the help message.')
|
||||
async def help(ctx:discord.ApplicationContext):
|
||||
"""Displays this message."""
|
||||
command_name = 'help'
|
||||
LOGGER.debug(f'{ctx.author.name} used /{command_name}.')
|
||||
embed = discord.Embed(title='Help',description='List of commands and their descriptions.',color=discord.Color.from_rgb(171,0,219))
|
||||
embed.set_author(name=bot.user.name,icon_url=bot.user.avatar.url)
|
||||
for command in bot.commands:
|
||||
embed.add_field(name=command.name,value=command.description,inline=False)
|
||||
await ctx.send_response(embed=embed,ephemeral=True)
|
||||
# @bot.slash_command(name='help',description='Displays the help message.')
|
||||
# async def help(ctx:discord.ApplicationContext):
|
||||
# """Displays this message."""
|
||||
# command_name = 'help'
|
||||
# LOGGER.debug(f'{ctx.author.name} used /{command_name}.')
|
||||
# embed = discord.Embed(title='Help',description='List of commands and their descriptions.',color=discord.Color.from_rgb(171,0,219))
|
||||
# embed.set_author(name=bot.user.name,icon_url=bot.user.avatar.url)
|
||||
# for command in bot.commands:
|
||||
# embed.add_field(name=command.name,value=command.description,inline=False)
|
||||
# await ctx.send_response(embed=embed,ephemeral=True)
|
||||
|
||||
@bot.slash_command(name='reload',description='Reloads the bot.')
|
||||
@option(name='feature',description='The feature to reload.',required=True,choices=['birthday','productivity'])
|
||||
@commands.is_owner()
|
||||
async def reload(ctx:discord.ApplicationContext, feature:str):
|
||||
"""Reloads the bot."""
|
||||
command_name = 'reload'
|
||||
guild = ctx.guild
|
||||
channel = ctx.channel
|
||||
LOGGER.debug(f'{ctx.author.name} used /{command_name} {feature}.')
|
||||
try:
|
||||
reload_feature(feature)
|
||||
await ctx.send_response(f'{feature} reloaded!',ephemeral=True)
|
||||
except Exception as e:
|
||||
LOGGER.error(f'Error reloading feature {feature}: {e}')
|
||||
await ctx.send_response(f'Error reloading feature {feature}: {e}',ephemeral=True)
|
||||
# @bot.slash_command(name='reload',description='Reloads the bot.')
|
||||
# @option(name='feature',description='The feature to reload.',required=True,choices=['birthday','productivity'])
|
||||
# @commands.is_owner()
|
||||
# async def reload(ctx:discord.ApplicationContext, feature:str):
|
||||
# """Reloads the bot."""
|
||||
# command_name = 'reload'
|
||||
# guild = ctx.guild
|
||||
# channel = ctx.channel
|
||||
# LOGGER.debug(f'{ctx.author.name} used /{command_name} {feature}.')
|
||||
# try:
|
||||
# reload_feature(feature)
|
||||
# await ctx.send_response(f'{feature} reloaded!',ephemeral=True)
|
||||
# except Exception as e:
|
||||
# LOGGER.error(f'Error reloading feature {feature}: {e}')
|
||||
# await ctx.send_response(f'Error reloading feature {feature}: {e}',ephemeral=True)
|
||||
|
||||
@bot.event
|
||||
async def on_member_join(member:discord.Member):
|
||||
if not member.bot:
|
||||
user_id = member.id
|
||||
user_name = member.name
|
||||
user_mention = member.mention
|
||||
db_user = db.select('global_user',f'WHERE user_id = {user_id}')
|
||||
if not db_user:
|
||||
db.insert('global_user',user_id=user_id,user_name=user_name,user_mention=user_mention)
|
||||
# @bot.event
|
||||
# async def on_member_join(member:discord.Member):
|
||||
# if not member.bot:
|
||||
# user_id = member.id
|
||||
# user_name = member.name
|
||||
# user_mention = member.mention
|
||||
# db_user = db.select('global_user',f'WHERE user_id = {user_id}')
|
||||
# if not db_user:
|
||||
# db.insert('global_user',user_id=user_id,user_name=user_name,user_mention=user_mention)
|
||||
|
||||
@bot.event
|
||||
async def on_raw_member_remove(payload:discord.RawMemberRemoveEvent):
|
||||
user = payload.user
|
||||
if not user.bot:
|
||||
user_id = user.id
|
||||
guild_id = payload.guild_id
|
||||
db.delete('guild_user_birthday',f'user_id = {user_id} AND guild_id = {guild_id}')
|
||||
db.delete('guild_user_productivity',f'user_id = {user_id} AND guild_id = {guild_id}')
|
||||
# @bot.event
|
||||
# async def on_raw_member_remove(payload:discord.RawMemberRemoveEvent):
|
||||
# user = payload.user
|
||||
# if not user.bot:
|
||||
# user_id = user.id
|
||||
# guild_id = payload.guild_id
|
||||
# db.delete('guild_user_birthday',f'user_id = {user_id} AND guild_id = {guild_id}')
|
||||
# db.delete('guild_user_productivity',f'user_id = {user_id} AND guild_id = {guild_id}')
|
||||
|
||||
@bot.event
|
||||
async def on_guild_join(guild:discord.Guild):
|
||||
guild_id = guild.id
|
||||
guild_name = guild.name
|
||||
LOGGER.info(f'Bot joined guild {guild_name} ({guild_id}).')
|
||||
db.insert('guild',guild_id=guild_id)
|
||||
db.insert('birthday_settings',guild_id=guild_id,is_enabled=False,channel_id=random.choice(guild.text_channels).id)
|
||||
db.insert('productivity_settings',guild_id=guild_id,is_enabled=False)
|
||||
# @bot.event
|
||||
# async def on_guild_join(guild:discord.Guild):
|
||||
# guild_id = guild.id
|
||||
# guild_name = guild.name
|
||||
# LOGGER.info(f'Bot joined guild {guild_name} ({guild_id}).')
|
||||
# db.insert('guild',guild_id=guild_id)
|
||||
# db.insert('birthday_settings',guild_id=guild_id,is_enabled=False,channel_id=random.choice(guild.text_channels).id)
|
||||
# db.insert('productivity_settings',guild_id=guild_id,is_enabled=False)
|
||||
|
||||
@bot.event
|
||||
async def on_guild_remove(guild:discord.Guild):
|
||||
guild_id = guild.id
|
||||
guild_name = guild.name
|
||||
LOGGER.info(f'Bot left guild {guild_name} ({guild_id}).')
|
||||
db.delete('birthday_settings',f'guild_id = {guild_id}')
|
||||
db.delete('productivity_settings',f'guild_id = {guild_id}')
|
||||
db.delete('guild',f'guild_id = {guild_id}')
|
||||
# @bot.event
|
||||
# async def on_guild_remove(guild:discord.Guild):
|
||||
# guild_id = guild.id
|
||||
# guild_name = guild.name
|
||||
# LOGGER.info(f'Bot left guild {guild_name} ({guild_id}).')
|
||||
# db.delete('birthday_settings',f'guild_id = {guild_id}')
|
||||
# db.delete('productivity_settings',f'guild_id = {guild_id}')
|
||||
# db.delete('guild',f'guild_id = {guild_id}')
|
||||
|
||||
@bot.event
|
||||
async def on_ready():
|
||||
LOGGER.info(f'{bot.user.name} has connected to Discord!')
|
||||
LOGGER.debug(f'Guilds: {bot.guilds}')
|
||||
if EXEC_ARGS.reload and EXEC_ARGS.guild_id and EXEC_ARGS.channel_id:
|
||||
guild = bot.get_guild(EXEC_ARGS.guild_id)
|
||||
channel = guild.get_channel_or_thread(EXEC_ARGS.channel_id)
|
||||
await channel.send(f'{bot.user.name} Bot reloaded!',silent=True)
|
||||
birthday_anouncements_task.start()
|
||||
for guild in bot.guilds:
|
||||
for user in guild.members:
|
||||
user_id = user.id
|
||||
db_user = db.select('global_user',f'user_id = {user_id}')
|
||||
if not db_user:
|
||||
db.insert('global_user',user_id=user_id,user_name=user.name,user_mention=user.mention)
|
||||
|
||||
productivity_settings = db.select('productivity_settings',f'guild_id = {guild.id}')
|
||||
if productivity_settings:
|
||||
if productivity_settings['is_enabled']:
|
||||
for user in guild.members:
|
||||
user_productivity_data = db.select('guild_user_productivity',f'user_id = {user.id} AND guild_id = {guild.id}')
|
||||
if user_productivity_data:
|
||||
if user_productivity_data['is_enabled']:
|
||||
next_reminder = user_productivity_data['next_reminder']
|
||||
if next_reminder > datetime.now().timestamp():
|
||||
cooldown = next_reminder - datetime.now().timestamp()
|
||||
channel = guild.get_channel(user_productivity_data['channel_id'])
|
||||
reminder_message = user_productivity_data['reminder_message']
|
||||
PRODUCTIVITY_TASKS[str(user.id)] = create_task(send_reminder(cooldown,user,channel,reminder_message))
|
||||
else:
|
||||
user_productivity_data['next_reminder'] = datetime.now().timestamp() + user_productivity_data['cooldown']
|
||||
db.update('guild_user_productivity',f'user_id = {user.id} AND guild_id = {guild.id}',next_reminder=user_productivity_data['next_reminder'])
|
||||
channel = guild.get_channel(user_productivity_data['channel_id'])
|
||||
reminder_message = user_productivity_data['reminder_message']
|
||||
PRODUCTIVITY_TASKS[str(user.id)] = create_task(send_reminder(user_productivity_data['cooldown'],user,channel,reminder_message))
|
||||
else:
|
||||
LOGGER.debug(f'No productivity settings found for guild {guild.name} ({guild.id}).')
|
||||
if bot.user:
|
||||
LOGGER.info(f'{bot.user.name} has connected to Discord!')
|
||||
LOGGER.debug(f'Guilds: {','.join([guild.name for guild in bot.guilds])}')
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
db.check_conn()
|
||||
result = db.select('token','token_name = \'discord\'','token_value')
|
||||
if result:
|
||||
TOKEN = result['token_value']
|
||||
load_dotenv()
|
||||
TOKEN = os.getenv("TOKEN")
|
||||
if TOKEN:
|
||||
bot.run(TOKEN)
|
||||
else:
|
||||
LOGGER.error('No token found.')
|
||||
|
||||
Reference in New Issue
Block a user