4c39e9c0ce
Sends instant notifications to Discord when the bot restarts, helping track stability issues and auto-recovery events. Changes: - Add notify_restart.py script to send webhook notifications - Integrate notification into bot startup (on_ready event) The notification includes: - Timestamp of restart (CST) - Reason (healthcheck failure detection) - Reference to diagnostics logs Configuration: Set RESTART_WEBHOOK_URL environment variable in docker-compose.yml to enable notifications. This provides immediate visibility when Docker auto-restarts the bot due to crashes or healthcheck failures. Co-Authored-By: Claude Sonnet 4.5 <noreply@anthropic.com>
126 lines
3.8 KiB
Python
126 lines
3.8 KiB
Python
import discord
|
|
import datetime
|
|
import logging
|
|
from logging.handlers import RotatingFileHandler
|
|
import asyncio
|
|
import os
|
|
|
|
from discord.ext import commands
|
|
|
|
from in_game.gameplay_models import create_db_and_tables
|
|
from health_server import run_health_server
|
|
from notify_restart import send_restart_notification
|
|
|
|
raw_log_level = os.getenv('LOG_LEVEL')
|
|
if raw_log_level == 'DEBUG':
|
|
log_level = logging.DEBUG
|
|
elif raw_log_level == 'INFO':
|
|
log_level = logging.INFO
|
|
elif raw_log_level == 'WARN':
|
|
log_level = logging.WARNING
|
|
else:
|
|
log_level = logging.ERROR
|
|
|
|
# date = f'{datetime.datetime.now().year}-{datetime.datetime.now().month}-{datetime.datetime.now().day}'
|
|
# logger.basicConfig(
|
|
# filename=f'logs/{date}.log',
|
|
# format='%(asctime)s - %(levelname)s - %(message)s',
|
|
# level=log_level
|
|
# )
|
|
# logger.getLogger('discord.http').setLevel(logger.INFO)
|
|
logger = logging.getLogger('discord_app')
|
|
logger.setLevel(log_level)
|
|
|
|
handler = RotatingFileHandler(
|
|
filename='logs/discord.log',
|
|
# encoding='utf-8',
|
|
maxBytes=32 * 1024 * 1024, # 32 MiB
|
|
backupCount=5, # Rotate through 5 files
|
|
)
|
|
|
|
formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
|
|
handler.setFormatter(formatter)
|
|
|
|
# dt_fmt = '%Y-%m-%d %H:%M:%S'
|
|
# formatter = logger.Formatter('[{asctime}] [{levelname:<8}] {name}: {message}', dt_fmt, style='{')
|
|
# handler.setFormatter(formatter)
|
|
logger.addHandler(handler)
|
|
|
|
COGS = [
|
|
'cogs.owner',
|
|
'cogs.admins',
|
|
'cogs.economy',
|
|
'cogs.players',
|
|
'cogs.gameplay',
|
|
]
|
|
|
|
intents = discord.Intents.default()
|
|
intents.members = True
|
|
intents.message_content = True
|
|
bot = commands.Bot(command_prefix='.',
|
|
intents=intents,
|
|
# help_command=None,
|
|
description='The Paper Dynasty Bot\nIf you have questions, feel free to contact Cal.',
|
|
case_insensitive=True,
|
|
owner_id=258104532423147520)
|
|
|
|
|
|
@bot.event
|
|
async def on_ready():
|
|
logger.info('Logged in as:')
|
|
logger.info(bot.user.name)
|
|
logger.info(bot.user.id)
|
|
|
|
# Send restart notification if configured
|
|
send_restart_notification()
|
|
|
|
|
|
@bot.tree.error
|
|
async def on_app_command_error(interaction: discord.Interaction, error: discord.app_commands.AppCommandError):
|
|
"""Global error handler for all app commands (slash commands)."""
|
|
logger.error(f'App command error in {interaction.command}: {error}', exc_info=error)
|
|
|
|
# Try to respond to the user
|
|
try:
|
|
if not interaction.response.is_done():
|
|
await interaction.response.send_message(
|
|
f'❌ An error occurred: {str(error)}',
|
|
ephemeral=True
|
|
)
|
|
else:
|
|
await interaction.followup.send(
|
|
f'❌ An error occurred: {str(error)}',
|
|
ephemeral=True
|
|
)
|
|
except Exception as e:
|
|
logger.error(f'Failed to send error message to user: {e}')
|
|
|
|
|
|
async def main():
|
|
create_db_and_tables()
|
|
for c in COGS:
|
|
try:
|
|
await bot.load_extension(c)
|
|
logger.info(f'Loaded cog: {c}')
|
|
except Exception as e:
|
|
logger.error(f'Failed to load cog: {c}')
|
|
logger.error(f'{e}')
|
|
|
|
# Start health server and bot concurrently
|
|
async with bot:
|
|
# Create health server task
|
|
health_task = asyncio.create_task(run_health_server(bot))
|
|
|
|
try:
|
|
# Start bot (this blocks until bot stops)
|
|
await bot.start(os.environ.get('BOT_TOKEN', 'NONE'))
|
|
finally:
|
|
# Cleanup: cancel health server when bot stops
|
|
health_task.cancel()
|
|
try:
|
|
await health_task
|
|
except asyncio.CancelledError:
|
|
pass
|
|
|
|
asyncio.run(main())
|