added error handling
This commit is contained in:
parent
4bdd2a7203
commit
137afd389c
1 changed files with 51 additions and 34 deletions
|
|
@ -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}")
|
||||
|
||||
bot.run(discord_token, root_logger=True)
|
||||
try:
|
||||
bot.run(discord_token, root_logger=True)
|
||||
except Exception as e:
|
||||
logger.critical(f"Fatal error running bot: {e}")
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue