added error handling

This commit is contained in:
phixxy 2024-01-30 03:38:16 -08:00
parent 4bdd2a7203
commit 137afd389c

View file

@ -17,30 +17,35 @@ intents.message_content = True
bot = commands.Bot(command_prefix='!', intents=intents)
async def folder_setup():
try:
folder_names = ["tmp", "extensions", "data"]
for folder_name in folder_names:
if not os.path.exists(folder_name):
os.mkdir(folder_name)
return folder_names
except Exception as e:
logger.error(f"Error setting up folders: {e}")
async def delete_all_files(path):
try:
for filename in os.listdir(path):
if os.path.isdir(path+filename):
shutil.rmtree(path+filename)
elif os.path.isfile(path+filename):
os.remove(path+filename)
except Exception as e:
logger.error(f"Error deleting files: {e}")
@tasks.loop(seconds=1) # Run the task every second
async def task_loop():
try:
current_time = time.localtime()
#Run daily tasks
if current_time.tm_hour == 0 and current_time.tm_min == 0 and current_time.tm_sec == 0:
try:
pass
#await delete_all_files("tmp/")
#logger.info("Deleted tmp/ files.")
except Exception as error:
logger.exception("Failed to delete files!")
await delete_all_files("tmp/")
logger.info("Deleted tmp/ files.")
except Exception as e:
logger.error(f"Error in task loop: {e}")
async def create_session():
return aiohttp.ClientSession()
@ -62,6 +67,7 @@ async def on_disconnect():
@bot.event
async def on_ready():
try:
bot.logger = logger
await delete_all_files("tmp/")
# Import plugins from extensions folder
@ -70,9 +76,20 @@ async def on_ready():
await bot.load_extension(f'extensions.{plugin_file[:-3]}')
logger.info('We have logged in as {0.user}'.format(bot))
task_loop.start()
except Exception as e:
logger.error(f"Error in on_ready: {e}")
raise
@bot.event
async def on_message(ctx):
try:
await bot.process_commands(ctx)
except commands.CommandNotFound:
pass
except Exception as e:
logger.error(f"Error processing commands: {e}")
try:
bot.run(discord_token, root_logger=True)
except Exception as e:
logger.critical(f"Fatal error running bot: {e}")