235 lines
9.6 KiB
Python
235 lines
9.6 KiB
Python
import discord
|
|
from discord.ext import commands
|
|
from discord import app_commands, ui
|
|
import time
|
|
import psutil
|
|
import platform
|
|
import GPUtil
|
|
import distro # Ensure this is installed
|
|
|
|
# Import wmi for Windows motherboard info
|
|
try:
|
|
import wmi
|
|
|
|
WMI_AVAILABLE = True
|
|
except ImportError:
|
|
WMI_AVAILABLE = False
|
|
|
|
|
|
class SystemCheckCog(commands.Cog):
|
|
def __init__(self, bot):
|
|
self.bot = bot
|
|
|
|
async def _system_check_logic(self, context_or_interaction):
|
|
"""Return detailed bot and system information as a LayoutView."""
|
|
# Bot information
|
|
bot_user = self.bot.user
|
|
guild_count = len(self.bot.guilds)
|
|
|
|
# More efficient member counting - use cached members when available
|
|
# This avoids API calls that can cause timeouts
|
|
user_ids = set()
|
|
for guild in self.bot.guilds:
|
|
try:
|
|
# Use members that are already cached
|
|
for member in guild.members:
|
|
if not member.bot:
|
|
user_ids.add(member.id)
|
|
except Exception as e:
|
|
print(f"Error counting members in guild {guild.name}: {e}")
|
|
user_count = len(user_ids)
|
|
|
|
# System information
|
|
system = platform.system()
|
|
os_info = f"{system} {platform.release()}"
|
|
hostname = platform.node()
|
|
distro_info_str = "" # Renamed variable
|
|
|
|
if system == "Linux":
|
|
try:
|
|
# Use distro library for better Linux distribution detection
|
|
distro_name = distro.name(pretty=True)
|
|
distro_info_str = f"\n**Distro:** {distro_name}"
|
|
except ImportError:
|
|
distro_info_str = "\n**Distro:** (Install 'distro' package for details)"
|
|
except Exception as e:
|
|
distro_info_str = f"\n**Distro:** (Error getting info: {e})"
|
|
elif system == "Windows":
|
|
# Add Windows version details if possible
|
|
try:
|
|
win_ver = platform.version() # e.g., '10.0.19041'
|
|
win_build = platform.win32_ver()[1] # e.g., '19041'
|
|
os_info = f"Windows {win_ver} (Build {win_build})"
|
|
except Exception as e:
|
|
print(f"Could not get detailed Windows version: {e}")
|
|
# Keep the basic os_info
|
|
|
|
uptime_seconds = time.time() - psutil.boot_time()
|
|
days, remainder = divmod(uptime_seconds, 86400)
|
|
hours, remainder = divmod(remainder, 3600)
|
|
minutes, seconds = divmod(remainder, 60)
|
|
|
|
uptime_str = ""
|
|
if days > 0:
|
|
uptime_str += f"{int(days)}d "
|
|
uptime_str += f"{int(hours):02}:{int(minutes):02}:{int(seconds):02}"
|
|
uptime = uptime_str.strip()
|
|
|
|
# Hardware information - use a shorter interval for CPU usage
|
|
cpu_usage = psutil.cpu_percent(interval=0.1)
|
|
|
|
# Get CPU info with a timeout to prevent hanging
|
|
try:
|
|
# Use a simpler approach for CPU name to avoid potential slowdowns
|
|
if platform.system() == "Windows":
|
|
cpu_name_base = platform.processor()
|
|
elif platform.system() == "Linux":
|
|
try:
|
|
with open("/proc/cpuinfo", "r") as f:
|
|
for line in f:
|
|
if line.startswith("model name"):
|
|
cpu_name_base = line.split(":")[1].strip()
|
|
break
|
|
else:
|
|
cpu_name_base = "Unknown CPU"
|
|
except:
|
|
cpu_name_base = platform.processor() or "Unknown CPU"
|
|
else:
|
|
cpu_name_base = platform.processor() or "Unknown CPU"
|
|
|
|
physical_cores = psutil.cpu_count(logical=False)
|
|
total_threads = psutil.cpu_count(logical=True)
|
|
cpu_name = f"{cpu_name_base} ({physical_cores}C/{total_threads}T)"
|
|
except Exception as e:
|
|
print(f"Error getting CPU info: {e}")
|
|
cpu_name = "N/A"
|
|
|
|
# Get motherboard information
|
|
motherboard_info = self._get_motherboard_info()
|
|
|
|
memory = psutil.virtual_memory()
|
|
ram_usage = f"{memory.used // (1024 ** 2)} MB / {memory.total // (1024 ** 2)} MB ({memory.percent}%)"
|
|
|
|
# GPU Information (using GPUtil for cross-platform consistency if available)
|
|
gpu_info_lines = []
|
|
try:
|
|
gpus = GPUtil.getGPUs()
|
|
if gpus:
|
|
for gpu in gpus:
|
|
gpu_info_lines.append(
|
|
f"{gpu.name} ({gpu.load*100:.1f}% Load, {gpu.memoryUsed:.0f}/{gpu.memoryTotal:.0f} MB VRAM)"
|
|
)
|
|
gpu_info = "\n".join(gpu_info_lines)
|
|
else:
|
|
gpu_info = "No dedicated GPU detected by GPUtil."
|
|
except ImportError:
|
|
gpu_info = "GPUtil library not installed. Cannot get detailed GPU info."
|
|
except Exception as e:
|
|
print(f"Error getting GPU info via GPUtil: {e}")
|
|
gpu_info = f"Error retrieving GPU info: {e}"
|
|
|
|
# Determine user and avatar URL based on context type
|
|
if isinstance(context_or_interaction, commands.Context):
|
|
user = context_or_interaction.author
|
|
avatar_url = user.display_avatar.url
|
|
elif isinstance(context_or_interaction, discord.Interaction):
|
|
user = context_or_interaction.user
|
|
avatar_url = user.display_avatar.url
|
|
else:
|
|
# Fallback or handle error if needed
|
|
user = self.bot.user # Or some default
|
|
avatar_url = self.bot.user.display_avatar.url if self.bot.user else None
|
|
|
|
view = ui.LayoutView(timeout=None)
|
|
container = ui.Container(accent_colour=discord.Color.blue())
|
|
view.add_item(container)
|
|
|
|
if bot_user:
|
|
header = ui.Section(
|
|
accessory=ui.Thumbnail(media=bot_user.display_avatar.url)
|
|
)
|
|
header.add_item(ui.TextDisplay("**📊 System Status**"))
|
|
container.add_item(header)
|
|
else:
|
|
container.add_item(ui.TextDisplay("**📊 System Status**"))
|
|
|
|
container.add_item(ui.Separator(spacing=discord.SeparatorSpacing.small))
|
|
|
|
if bot_user:
|
|
container.add_item(ui.TextDisplay(f"**Bot Name:** {bot_user.name}"))
|
|
container.add_item(ui.TextDisplay(f"**Bot ID:** {bot_user.id}"))
|
|
else:
|
|
container.add_item(ui.TextDisplay("Bot user information not available."))
|
|
|
|
container.add_item(ui.TextDisplay(f"**Servers:** {guild_count}"))
|
|
container.add_item(ui.TextDisplay(f"**Unique Users:** {user_count}"))
|
|
|
|
container.add_item(ui.Separator(spacing=discord.SeparatorSpacing.small))
|
|
container.add_item(ui.TextDisplay(f"**OS:** {os_info}{distro_info_str}"))
|
|
container.add_item(ui.TextDisplay(f"**Hostname:** {hostname}"))
|
|
container.add_item(ui.TextDisplay(f"**Uptime:** {uptime}"))
|
|
|
|
container.add_item(ui.Separator(spacing=discord.SeparatorSpacing.small))
|
|
container.add_item(ui.TextDisplay(f"**Device Model:** {motherboard_info}"))
|
|
container.add_item(ui.TextDisplay(f"**CPU:** {cpu_name}"))
|
|
container.add_item(ui.TextDisplay(f"**CPU Usage:** {cpu_usage}%"))
|
|
container.add_item(ui.TextDisplay(f"**RAM Usage:** {ram_usage}"))
|
|
container.add_item(ui.TextDisplay(f"**GPU Info:** {gpu_info}"))
|
|
|
|
container.add_item(ui.Separator(spacing=discord.SeparatorSpacing.small))
|
|
footer = discord.utils.format_dt(discord.utils.utcnow(), style="f")
|
|
if user:
|
|
footer += f" | Requested by: {user.display_name}"
|
|
container.add_item(ui.TextDisplay(footer))
|
|
|
|
return view
|
|
|
|
@commands.hybrid_command(
|
|
name="systemcheck", description="Check the bot and system status"
|
|
)
|
|
async def system_check(self, ctx: commands.Context):
|
|
"""Hybrid command for checking the bot and system status."""
|
|
if ctx.interaction:
|
|
await ctx.interaction.response.defer(thinking=True)
|
|
try:
|
|
view = await self._system_check_logic(ctx)
|
|
await ctx.reply(view=view)
|
|
except Exception as e:
|
|
print(f"Error in system_check command: {e}")
|
|
await ctx.reply(
|
|
f"An error occurred while checking system status: {e}",
|
|
ephemeral=bool(ctx.interaction),
|
|
)
|
|
|
|
def _get_motherboard_info(self):
|
|
"""Get motherboard information based on the operating system."""
|
|
system = platform.system()
|
|
try:
|
|
if system == "Windows":
|
|
if WMI_AVAILABLE:
|
|
w = wmi.WMI()
|
|
for board in w.Win32_BaseBoard():
|
|
return f"{board.Manufacturer} {board.Product}"
|
|
return "WMI module not available"
|
|
elif system == "Linux":
|
|
# Read motherboard product name from sysfs
|
|
try:
|
|
with open("/sys/devices/virtual/dmi/id/product_name", "r") as f:
|
|
product_name = f.read().strip()
|
|
return product_name if product_name else "Unknown motherboard"
|
|
except FileNotFoundError:
|
|
return "/sys/devices/virtual/dmi/id/product_name not found"
|
|
except Exception as e:
|
|
return f"Error reading motherboard info: {e}"
|
|
except Exception as e:
|
|
return f"Error: {str(e)}"
|
|
else:
|
|
return f"Unsupported OS: {system}"
|
|
except Exception as e:
|
|
print(f"Error getting motherboard info: {e}")
|
|
return "Error retrieving motherboard info"
|
|
|
|
|
|
async def setup(bot):
|
|
await bot.add_cog(SystemCheckCog(bot))
|