debug mode var and ingame_channel shorthand getter
All checks were successful
Build and Push Docker Image / build (push) Successful in 23s

This commit is contained in:
Corban-Lee Jones 2024-12-11 15:37:19 +00:00
parent 3e622fcecb
commit 77cb214276
2 changed files with 32 additions and 28 deletions

25
bot.py
View File

@ -11,7 +11,7 @@ import logging.config
from os import getenv
from pathlib import Path
from discord import Intents
from discord import Intents, TextChannel
from discord.ext import commands
from dotenv import load_dotenv
from tortoise import Tortoise
@ -28,12 +28,9 @@ class DiscordBot(commands.Bot):
Represents a Discord bot.
Contains controls to interact with the bot via the Discord API.
"""
in_game_channel_id: int
steam_api_key: str
rcon_details: dict
def __init__(self):
def __init__(self, debug_mode: bool):
super().__init__(command_prefix="-", intents=Intents.all())
self.debug_mode = debug_mode
self.in_game_channel_id = int(getenv("SPIFFO__DISCORD_CHANNEL_ID"))
self.steam_api_key = getenv("SPIFFO__STEAM_API_KEY")
self.rcon_details = {
@ -75,6 +72,12 @@ class DiscordBot(commands.Bot):
if path.suffix == ".py":
await self.load_extension(f"cogs.{path.stem}")
async def get_ingame_channel(self) -> TextChannel:
"""
"""
channel = self.bot.get_channel(self.bot.in_game_channel_id)
return channel or await self.bot.fetch_channel(self.bot.in_game_channel_id)
def get_bot_token() -> str:
"""
@ -87,7 +90,7 @@ def get_bot_token() -> str:
return bot_token
def setup_logging_config():
def setup_logging_config(debug_mode: bool):
"""
Loads the logging configuration and creates an asynchronous queue handler.
"""
@ -101,6 +104,9 @@ def setup_logging_config():
# Ensure the logging directory exists
(BASE_DIR / "logs").mkdir(exist_ok=True)
# update the log level dependent on debug mode
log_config["loggers"]["root"]["level"] = "DEBUG" if debug_mode else "INFO"
# Load the config
logging.config.dictConfig(log_config) # This 'logging.config' path is jank.
@ -114,10 +120,11 @@ async def main():
"""
The entrypoint function, initialises the application.
"""
setup_logging_config()
debug_mode = getenv("SPIFFO__DEBUG").lower() == "true"
setup_logging_config(debug_mode)
bot_token = get_bot_token()
async with DiscordBot() as bot:
async with DiscordBot(debug_mode) as bot:
await bot.load_cogs()
await bot.start(bot_token, reconnect=True)

View File

@ -16,11 +16,7 @@ from utils.models import Player
ZOMBOID_FOLDER_PATH = Path(getenv("SPIFFO__ZOMBOID_FOLDER_PATH"))
LOGS_FOLDER_PATH = ZOMBOID_FOLDER_PATH / "Logs"
USER_LOG_FILE_PATH = None
for path in LOGS_FOLDER_PATH.iterdir():
if path.name.endswith("_user.txt"):
USER_LOG_FILE_PATH = path
break
USER_LOG_FILE_PATH = next(LOGS_FOLDER_PATH.glob("*_user.txt"), None)
log = logging.getLogger(__name__)
@ -38,6 +34,8 @@ class PlayersCog(commands.Cog):
@tasks.loop(seconds=3)
async def listen_for_changes(self):
"""
"""
log.debug("listening for changes")
for line in await self.file_handler.read():
await self.process_log_line(line)
@ -52,7 +50,7 @@ class PlayersCog(commands.Cog):
elif "disconnected player" in line:
await self.process_disconnected_player(line)
async def process_player_death(self, line: str):
async def process_player_death(self, line: str, alert: bool = False):
log.debug("processing player death")
re_pattern = r"\[(?P<timestamp>[\d\- :\.]+)\] user (?P<username>.+?) died at \((?P<x>\d+),(?P<y>\d+),(?P<z>\d+)\) \((?P<cause>.+?)\)"
re_match = re.search(re_pattern, line)
@ -78,19 +76,19 @@ class PlayersCog(commands.Cog):
)
await player.save()
channel = self.bot.get_channel(self.bot.in_game_channel_id)
channel = channel or await self.bot.fetch_channel(self.bot.in_game_channel_id)
log.debug("successfully registered player death to %s", re_match.group("username"))
if not alert:
return
channel = await self.bot.get_ingame_channel()
embed = await player.get_embed()
embed.title = "Player Has Died"
embed.colour = Colour.dark_orange()
await channel.send(embed=embed)
log.debug("successfully registered player death to %s", re_match.group("username"))
async def process_connected_player(self, line: str):
async def process_connected_player(self, line: str, alert: bool = False):
"""
"""
log.debug("processing connected player")
@ -115,16 +113,17 @@ class PlayersCog(commands.Cog):
await player.save()
return
channel = self.bot.get_channel(self.bot.in_game_channel_id)
channel = channel or await self.bot.fetch_channel(self.bot.in_game_channel_id)
if not alert:
return
channel = await self.bot.get_ingame_channel()
embed = await player.get_embed()
embed.title = "Player Has Connected"
embed.colour = Colour.brand_green()
await channel.send(embed=embed)
async def process_disconnected_player(self, line: str):
async def process_disconnected_player(self, line: str, alert: bool = False):
"""
"""
log.debug("processing disconnected player")
@ -143,12 +142,10 @@ class PlayersCog(commands.Cog):
await player.update_steam_summary(re_match.group("steam_id"), self.bot.steam_api_key)
await player.save()
if player.is_dead:
if player.is_dead or not alert:
return
channel = self.bot.get_channel(self.bot.in_game_channel_id)
channel = channel or await self.bot.fetch_channel(self.bot.in_game_channel_id)
channel = await self.bot.get_ingame_channel()
embed = await player.get_embed()
embed.title = "Player Has Disconnected"
embed.colour = Colour.brand_red()