phixxy/sparkytron3000
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

added error handling

Browse source
This commit is contained in:
phixxy 2024-01-30 03:38:16 -08:00
parent 4bdd2a7203
commit 137afd389c
1 changed files with 51 additions and 34 deletions
Download patch file Download diff file Expand all files Collapse all files

31
sparkytron3000.py
View file

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