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

sparky logger should be working properly now

Browse source
This commit is contained in:
phixxy 2024-02-02 22:11:52 -08:00
parent 9e1624eb6b
commit 9cae625cc2
2 changed files with 25 additions and 13 deletions
Download patch file Download diff file Expand all files Collapse all files

9
extensions/inky_phat.py
View file

@ -17,9 +17,12 @@ class InkyScreen(commands.Cog):
self.admin_ids = [242018983241318410] self.admin_ids = [242018983241318410]
def setup(self): def setup(self):
display = inky.auto() if self.enabled:
display.set_border(inky.BLACK) display = inky.auto()
return display display.set_border(inky.BLACK)
return display
else:
return None
def write_to_display(self, text: list): def write_to_display(self, text: list):
if text is not self.old_message: if text is not self.old_message:

29
sparkytron3000.py
View file

@ -10,21 +10,20 @@ import src.logger as logger
load_dotenv() load_dotenv()
discord_token = os.getenv('discord_token') discord_token = os.getenv('discord_token')
logger = logger.logging.getLogger("bot")
intents = discord.Intents.all() intents = discord.Intents.all()
intents.message_content = True 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: try:
folder_names = ["tmp", "extensions", "data"] folder_names = ["tmp", "extensions", "data", "logs"]
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
except Exception as e: except Exception as e:
logger.error(f"Error setting up folders: {e}") logger.exception(f"Error setting up folders: {e}")
async def delete_all_files(path): async def delete_all_files(path):
try: try:
@ -34,7 +33,7 @@ async def delete_all_files(path):
elif os.path.isfile(path+filename): elif os.path.isfile(path+filename):
os.remove(path+filename) os.remove(path+filename)
except Exception as e: except Exception as e:
logger.error(f"Error deleting files: {e}") logger.exception(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():
@ -45,7 +44,7 @@ async def task_loop():
await delete_all_files("tmp/") await delete_all_files("tmp/")
logger.info("Deleted tmp/ files.") logger.info("Deleted tmp/ files.")
except Exception as e: except Exception as e:
logger.error(f"Error in task loop: {e}") logger.exception(f"Error in task loop: {e}")
async def create_session(): async def create_session():
return aiohttp.ClientSession() return aiohttp.ClientSession()
@ -65,10 +64,19 @@ async def on_resumed():
async def on_disconnect(): async def on_disconnect():
await close_session(bot.http_session) await close_session(bot.http_session)
def logger_setup():
if not os.path.isdir("logs"):
os.mkdir("logs")
with open("logs/info.log", "a") as f:
pass
logger = logger.logging.getLogger("bot")
bot.logger = logger
@bot.event @bot.event
async def on_ready(): async def on_ready():
try: try:
bot.logger = logger await folder_setup()
await delete_all_files("tmp/") await delete_all_files("tmp/")
# Import plugins from extensions folder # Import plugins from extensions folder
for plugin_file in os.listdir('extensions/'): for plugin_file in os.listdir('extensions/'):
@ -77,7 +85,7 @@ async def on_ready():
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: except Exception as e:
logger.error(f"Error in on_ready: {e}") logger.warning(f"Error in on_ready: {e}")
raise raise
@bot.event @bot.event
@ -87,9 +95,10 @@ async def on_message(ctx):
except commands.CommandNotFound: except commands.CommandNotFound:
pass pass
except Exception as e: except Exception as e:
logger.error(f"Error processing commands: {e}") logger.warning(f"Error processing commands: {e}")
try: try:
logger_setup()
bot.run(discord_token, root_logger=True) bot.run(discord_token, root_logger=True)
except Exception as e: except Exception as e:
logger.critical(f"Fatal error running bot: {e}") logger.critical(f"Fatal error running bot: {e}")