added first slashcommand and refactored the preview cog as result of breaking changes in slashcommand
This commit is contained in:
112
jeeves.py
112
jeeves.py
@@ -1,66 +1,108 @@
|
||||
#!/usr/bin/env python3.8
|
||||
|
||||
import discord
|
||||
from discord.ext import commands
|
||||
from discord.ext import commands, tasks
|
||||
from jeevesbot import env
|
||||
from jeevesbot.database import init_db, get_due_reminders
|
||||
import os
|
||||
import log
|
||||
import logging.config
|
||||
from logging import getLogger
|
||||
import datetime
|
||||
import asyncio
|
||||
|
||||
|
||||
# setup root logger handlers
|
||||
logging.config.dictConfig(log.LOGGING)
|
||||
# Initialize the database
|
||||
init_db()
|
||||
|
||||
|
||||
# setup logging
|
||||
logging.config.dictConfig(log.LOGGING)
|
||||
log = getLogger(__name__)
|
||||
|
||||
|
||||
# setup discord.py bot
|
||||
intents = discord.Intents().all()
|
||||
bot = commands.Bot(command_prefix='!', intents=intents, help_command=None)
|
||||
intents.message_content = True
|
||||
e = discord.Embed()
|
||||
|
||||
|
||||
@bot.command(name='load', hidden=True)
|
||||
@commands.has_permissions(administrator=True)
|
||||
async def load(ctx, extension):
|
||||
bot.load_extension(f'cogs.{extension}')
|
||||
log.info(f'{ctx.message.author} loaded the {extension} module')
|
||||
class Jeeves(commands.Bot):
|
||||
def __init__(self):
|
||||
super().__init__(command_prefix='!', intents=intents, help_command=None)
|
||||
self.guild_ids = [env.GUILD_ID]
|
||||
|
||||
|
||||
@bot.command(name='unload', hidden=True)
|
||||
@commands.has_permissions(administrator=True)
|
||||
async def unload(ctx, extension):
|
||||
bot.unload_extension(f'cogs.{extension}')
|
||||
log.info(f'{ctx.message.author} unloaded the {extension} module')
|
||||
@commands.command(name='load', hidden=True)
|
||||
@commands.has_permissions(administrator=True)
|
||||
async def load(self, ctx, extension):
|
||||
self.load_extension(f'cogs.{extension}')
|
||||
log.info(f'{ctx.message.author} loaded the {extension} module')
|
||||
|
||||
|
||||
@bot.command(name='reload', hidden=True)
|
||||
@commands.has_permissions(administrator=True)
|
||||
async def reload(ctx, extension):
|
||||
bot.unload_extension(f'cogs.{extension}')
|
||||
bot.load_extension(f'cogs.{extension}')
|
||||
log.info(f'{ctx.message.author} reloaded the {extension} module')
|
||||
@commands.command(name='unload', hidden=True)
|
||||
@commands.has_permissions(administrator=True)
|
||||
async def unload(self, ctx, extension):
|
||||
self.unload_extension(f'cogs.{extension}')
|
||||
log.info(f'{ctx.message.author} unloaded the {extension} module')
|
||||
|
||||
|
||||
async def load_extensions():
|
||||
for filename in os.listdir('./cogs'):
|
||||
if filename.endswith('.py'):
|
||||
await bot.load_extension(f'cogs.{filename[:-3]}')
|
||||
@commands.command(name='reload', hidden=True)
|
||||
@commands.has_permissions(administrator=True)
|
||||
async def reload(self, ctx, extension):
|
||||
self.unload_extension(f'cogs.{extension}')
|
||||
self.load_extension(f'cogs.{extension}')
|
||||
log.info(f'{ctx.message.author} reloaded the {extension} module')
|
||||
|
||||
|
||||
@bot.event
|
||||
async def on_ready():
|
||||
log.info(f'Active with ID:{bot.user.id} as {bot.user.name}')
|
||||
activity = discord.Activity(name='!help', type=discord.ActivityType.listening)
|
||||
await bot.change_presence(activity=activity)
|
||||
async def load_extensions(self):
|
||||
for filename in os.listdir('./cogs'):
|
||||
if filename.endswith('.py'):
|
||||
await self.load_extension(f'cogs.{filename[:-3]}')
|
||||
|
||||
|
||||
@tasks.loop(seconds=60)
|
||||
async def check_reminders(self):
|
||||
now = datetime.datetime.now().isoformat()
|
||||
reminders = get_due_reminders(now)
|
||||
for reminder in reminders:
|
||||
user = self.get_user(reminder[1])
|
||||
if user:
|
||||
try:
|
||||
await user.send(reminder[2])
|
||||
except Exception as e:
|
||||
log.error(f'Error sending reminder to user {reminder[1]}: {e}')
|
||||
|
||||
|
||||
async def on_ready(self):
|
||||
log.info(f'Active with ID:{self.user.id} as {self.user.name}')
|
||||
activity = discord.Activity(name='!help', type=discord.ActivityType.listening)
|
||||
await self.change_presence(activity=activity)
|
||||
# Sync commands for all guilds
|
||||
for guild_id in self.guild_ids:
|
||||
guild = discord.Object(id=guild_id)
|
||||
try:
|
||||
await self.tree.sync(guild=guild)
|
||||
log.info(f'Successfully synced commands for guild {guild_id}')
|
||||
except discord.errors.Forbidden as e:
|
||||
log.error(f'Failed to sync commands for guild {guild_id}: {e}')
|
||||
# Start the reminder check loop
|
||||
if not self.check_reminders.is_running():
|
||||
self.check_reminders.start()
|
||||
|
||||
|
||||
async def on_command_error(self, ctx, error):
|
||||
if isinstance(error, commands.CommandNotFound):
|
||||
await ctx.send('Command not found.')
|
||||
log.warning(f'Command not found: {ctx.message.content}')
|
||||
else:
|
||||
await ctx.send('An error occurred.')
|
||||
log.error(f'An error occurred: {error}')
|
||||
|
||||
|
||||
async def main():
|
||||
async with bot:
|
||||
await load_extensions()
|
||||
await bot.start(env.TOKEN)
|
||||
bot = Jeeves()
|
||||
await bot.load_extensions()
|
||||
await bot.start(env.TOKEN)
|
||||
|
||||
|
||||
asyncio.run(main())
|
||||
if __name__ == "__main__":
|
||||
asyncio.run(main())
|
||||
|
||||
Reference in New Issue
Block a user