feat: new command top implmented to check RPI hardware status

Author: Sang-Buster

README.md

@@ -168,7 +168,7 @@ For convenience, you can set up an alias named `meteorix`:
    # Pure bash completion - no Python involved
    function _meteorix_complete {
       local cur=${COMP_WORDS[COMP_CWORD]}
-      local commands="upload delete check head tail info spit plot freq ifconfig monitor eda ml who -h --help"
+      local commands="upload delete check head tail info spit plot freq ifconfig top monitor eda ml who -h --help"
 
       if [ $COMP_CWORD -eq 1 ]; then
          COMPREPLY=($(compgen -W "$commands" -- "$cur"))
@@ -195,7 +195,7 @@ For convenience, you can set up an alias named `meteorix`:
 
    # Zsh completion
    function _meteorix {
-      local commands="upload delete check head tail info spit plot freq ifconfig monitor eda ml who -h --help"
+      local commands="upload delete check head tail info spit plot freq ifconfig top monitor eda ml who -h --help"
       _arguments "1: :($commands)"
    }
 

src/cli.py

@@ -14,6 +14,7 @@
     create_weather_plot,
     toggle_monitor,
     get_pi_ip,
+    get_system_stats,
 )
 
 import sys
@@ -30,7 +31,7 @@
 def get_parser():
     parser = argparse.ArgumentParser(
         description="Weather data management CLI",
-        usage="meteorix [-h] {upload, delete, check, head, tail, info, spit, plot, monitor, eda, ml, who, ifconfig} ...",
+        usage="meteorix [-h] {upload, delete, check, head, tail, info, spit, plot, monitor, ifconfig, top, eda, ml, who}",
     )
 
     subparsers = parser.add_subparsers(dest="command", required=True)
@@ -153,6 +154,16 @@ def get_parser():
                 ),
             ],
         },
+        "ifconfig": {
+            "help": "Show Raspberry Pi network information",
+            "description": "Display network interface information from the weather station Raspberry Pi.",
+            "args": [],
+        },
+        "top": {
+            "help": "Show Raspberry Pi system status",
+            "description": "Display system metrics including CPU, memory, disk usage and temperature.",
+            "args": [],
+        },
         "eda": {
             "help": "Run exploratory data analysis",
             "description": "Perform exploratory data analysis including correlation analysis and PCA, then upload results to MongoDB.",
@@ -168,11 +179,6 @@ def get_parser():
             "description": "Display detailed information about the Meteorix bot and its creators.",
             "args": [("--force", {"action": "store_true", "help": argparse.SUPPRESS})],
         },
-        "ifconfig": {
-            "help": "Show Raspberry Pi IP addresses",
-            "description": "Display network interface information from the weather station Raspberry Pi.",
-            "args": [],
-        },
     }
 
     # Create subparsers from command configurations
@@ -213,6 +219,7 @@ def main():
         "tail": lambda: show_tail(args.date if hasattr(args, "date") else None),
         "monitor": lambda: toggle_monitor(args.action),
         "ifconfig": lambda: get_pi_ip(),
+        "top": lambda: get_system_stats(),
     }
 
     # Date-based command handlers

src/cli_components/__init__.py

@@ -12,6 +12,7 @@
 from .plot import create_weather_plot
 from .monitor import toggle_monitor
 from .ifconfig import get_pi_ip
+from .top import get_system_stats
 
 __all__ = [
     "check_analysis_results",
@@ -30,4 +31,5 @@
     "create_weather_plot",
     "toggle_monitor",
     "get_pi_ip",
+    "get_system_stats",
 ]

src/cli_components/freq.py

@@ -0,0 +1,109 @@
+from rich import print as rprint
+from .ifconfig import SSHClient, find_pi_address
+
+
+def get_system_stats():
+    """Get system statistics from Raspberry Pi"""
+    try:
+        # Find the Pi's current address
+        pi_address = find_pi_address()
+        if not pi_address:
+            raise Exception("Could not locate the Raspberry Pi on the network")
+
+        # Get SSH client instance
+        ssh = SSHClient.get_instance().get_client(pi_address)
+
+        # Commands to get various system metrics
+        commands = {
+            "loadavg": "cat /proc/loadavg",  # Load averages
+            "memory": "free -h",  # Memory usage
+            "cpu_temp": "vcgencmd measure_temp",  # CPU temperature
+            "cpu_usage": "top -bn1 | grep '%Cpu'",  # CPU usage
+            "disk": "df -h /",  # Disk usage for root partition
+            "uptime": "uptime -p",  # System uptime
+            "processes": "ps aux --sort=-%cpu | grep -v 'ps aux' | head -6",  # Top 5 CPU-consuming processes, excluding ps command
+        }
+
+        # Collect system information
+        system_info = {}
+        for key, cmd in commands.items():
+            stdin, stdout, stderr = ssh.exec_command(cmd)
+            system_info[key] = stdout.read().decode().strip()
+            error = stderr.read().decode()
+            if error and "command not found" not in error:
+                rprint(f"[yellow]Warning when running {cmd}: {error}[/yellow]")
+
+        # Parse and display results
+        rprint(f"\n[bold blue]{'='*60}[/bold blue]")
+        rprint("[bold green]Raspberry Pi System Status[/bold green]")
+        rprint(f"[bold blue]{'='*60}[/bold blue]\n")
+
+        # System Uptime
+        rprint("[yellow]System Uptime:[/yellow]")
+        rprint(f"  • {system_info['uptime']}\n")
+
+        # Load Average
+        load_1, load_5, load_15, *_ = system_info["loadavg"].split()
+        rprint("[yellow]Load Average:[/yellow]")
+        rprint(f"  • 1 min: {load_1}")
+        rprint(f"  • 5 min: {load_5}")
+        rprint(f"  • 15 min: {load_15}\n")
+
+        # CPU Usage and Temperature
+        rprint("[yellow]CPU Status:[/yellow]")
+        cpu_usage = system_info["cpu_usage"].replace("  ", " ").split(",")
+        user_cpu = cpu_usage[0].split()[1]
+        system_cpu = cpu_usage[2].split()[0]
+        idle_cpu = cpu_usage[3].split()[0]
+        rprint(f"  • User: {user_cpu}%")
+        rprint(f"  • System: {system_cpu}%")
+        rprint(f"  • Idle: {idle_cpu}%")
+
+        # Temperature
+        temp = system_info["cpu_temp"].replace("temp=", "").replace("'C", "°C")
+        rprint(f"  • Temperature: {temp}\n")
+
+        # Memory Usage
+        rprint("[yellow]Memory Usage:[/yellow]")
+        memory_lines = system_info["memory"].split("\n")
+        for line in memory_lines:
+            if line.startswith("Mem:"):
+                total, used, free, shared, buff_cache, available = line.split()[1:]
+                rprint(f"  • Total: {total}")
+                rprint(f"  • Used: {used}")
+                rprint(f"  • Free: {free}")
+                rprint(f"  • Buff/Cache: {buff_cache}")
+                rprint(f"  • Available: {available}\n")
+
+        # Disk Usage
+        rprint("[yellow]Disk Usage:[/yellow]")
+        disk_lines = system_info["disk"].split("\n")
+        for line in disk_lines[1:]:  # Skip header
+            fs, size, used, avail, use_percent, mount = line.split()
+            rprint(f"  • Size: {size}")
+            rprint(f"  • Used: {used} ({use_percent})")
+            rprint(f"  • Available: {avail}\n")
+
+        # Top Processes
+        rprint("[yellow]Top Processes (CPU):[/yellow]")
+        process_lines = system_info["processes"].split("\n")
+        for line in process_lines[1:]:  # Skip header
+            parts = line.split()
+            if len(parts) >= 11:
+                user = parts[0]
+                pid = parts[1]
+                cpu = parts[2]
+                mem = parts[3]
+                command = " ".join(parts[10:])
+                rprint(
+                    f"  • {user:<8} PID: {pid:<6} {command[:30]:<30} CPU: {cpu:>5}% MEM: {mem:>5}%"
+                )
+
+        rprint(f"\n[bold blue]{'='*60}[/bold blue]")
+        return True
+
+    except Exception as e:
+        rprint(f"[red]Error: {str(e)}[/red]")
+        # If there's an error, try to close the connection to allow for a fresh start next time
+        SSHClient.get_instance().close()
+        return False

src/cli_components/top.py

@@ -128,6 +128,8 @@ async def help_command(ctx, command_name=None):
 • `ml` - Run machine learning analysis
 • `plot <start_date> [end_date]` - Generate weather plots for specified dates
 • `monitor` - Monitor data collection status
+• `ifconfig` - Show Raspberry Pi network information
+• `top` - Show Raspberry Pi system status
 • `who` - Show information about the bot
 • `help` - Show this help message
 • `help <command>` - Show detailed help for a specific command
@@ -146,7 +148,9 @@ async def help_command(ctx, command_name=None):
 `@meteorix who` - Show bot information
 `@meteorix monitor enable` - Enable data collection monitoring
 `@meteorix monitor disable` - Disable data collection monitoring
-`@meteorix monitor status` - Check current monitoring status"""
+`@meteorix monitor status` - Check current monitoring status
+`@meteorix ifconfig` - Show Pi network status
+`@meteorix top` - Show Pi system status"""
             await ctx.send(help_text)
             return
 
@@ -165,6 +169,8 @@ async def help_command(ctx, command_name=None):
 • `ml` - Run machine learning analysis
 • `plot <start_date> [end_date]` - Generate weather plots for specified dates
 • `monitor` - Monitor data collection status
+• `ifconfig` - Show Raspberry Pi network information
+• `top` - Show Raspberry Pi system status
 • `who` - Show information about the bot
 • `help` - Show this help message
 • `help <command>` - Show detailed help for a specific command
@@ -203,6 +209,8 @@ async def help_command(ctx, command_name=None):
 • `ml` - Run machine learning analysis
 • `plot <start_date> [end_date]` - Generate weather plots for specified dates
 • `monitor` - Monitor data collection status
+• `ifconfig` - Show Raspberry Pi network information
+• `top` - Show Raspberry Pi system status
 • `who` - Show information about the bot
 • `help` - Show this help message
 • `help <command>` - Show detailed help for a specific command
@@ -221,7 +229,9 @@ async def help_command(ctx, command_name=None):
 `@meteorix who` - Show bot information
 `@meteorix monitor enable` - Enable data collection monitoring
 `@meteorix monitor disable` - Disable data collection monitoring
-`@meteorix monitor status` - Check current monitoring status"""
+`@meteorix monitor status` - Check current monitoring status
+`@meteorix ifconfig` - Show Pi network status
+`@meteorix top` - Show Pi system status"""
         await ctx.send(help_text)
 
 
@@ -359,6 +369,12 @@ async def ifconfig(ctx):
     await run_cli_command(ctx, ["ifconfig"])
 
 
+@bot.command(name="top")
+@check_channel()
+async def top(ctx):
+    await run_cli_command(ctx, ["top"])
+
+
 # Slash commands
 @bot.tree.command(name="info", description="Show available date range")
 @app_commands.describe(month="Optional: Month to show statistics for (YYYY_MM)")
@@ -506,12 +522,18 @@ async def monitor_slash(interaction: discord.Interaction, action: str):
     await interaction.response.send_message(f"```\n{output}\n```")
 
 
-@bot.tree.command(name="ifconfig", description="Show Raspberry Pi IP addresses")
+@bot.tree.command(name="ifconfig", description="Show Raspberry Pi network information")
 @app_commands.check(check_channel_slash)
 async def ifconfig_slash(interaction: discord.Interaction):
     await run_cli_command_slash(interaction, ["ifconfig"])
 
 
+@bot.tree.command(name="top", description="Show Raspberry Pi system status")
+@app_commands.check(check_channel_slash)
+async def top_slash(interaction: discord.Interaction):
+    await run_cli_command_slash(interaction, ["top"])
+
+
 VALID_COMMANDS = [
     "info",
     "upload",
@@ -527,6 +549,7 @@ async def ifconfig_slash(interaction: discord.Interaction):
     "plot",
     "monitor",
     "ifconfig",
+    "top",
 ]
 
 
@@ -545,7 +568,8 @@ def get_command_description(cmd):
         "spit": "Get raw CSV data for specified dates",
         "plot": "Create weather data plots",
         "monitor": "Monitor data collection status",
-        "ifconfig": "Show Raspberry Pi IP addresses",
+        "ifconfig": "Show Raspberry Pi network information",
+        "top": "Show Raspberry Pi system status",
     }
     return descriptions.get(cmd, "")
 
@@ -578,6 +602,8 @@ async def help_slash(interaction: discord.Interaction, command_name: str = None)
 • `ml` - Run machine learning analysis
 • `plot <start_date> [end_date]` - Generate weather plots for specified dates
 • `monitor` - Monitor data collection status
+• `ifconfig` - Show Raspberry Pi network information
+• `top` - Show Raspberry Pi system status
 • `who` - Show information about the bot
 • `help` - Show this help message
 • `help <command>` - Show detailed help for a specific command
@@ -605,6 +631,8 @@ async def help_slash(interaction: discord.Interaction, command_name: str = None)
 • `ml` - Run machine learning analysis
 • `plot <start_date> [end_date]` - Generate weather plots for specified dates
 • `monitor` - Monitor data collection status
+• `ifconfig` - Show Raspberry Pi network information
+• `top` - Show Raspberry Pi system status
 • `who` - Show information about the bot
 • `help` - Show this help message
 • `help <command>` - Show detailed help for a specific command
@@ -620,7 +648,9 @@ async def help_slash(interaction: discord.Interaction, command_name: str = None)
 `/who` - Show bot information
 `/monitor enable` - Enable data collection monitoring
 `/monitor disable` - Disable data collection monitoring
-`/monitor status` - Check current monitoring status"""
+`/monitor status` - Check current monitoring status
+`/ifconfig` - Show Pi network status
+`/top` - Show Pi system status"""
             await interaction.followup.send(help_text)
             return
 
@@ -653,6 +683,8 @@ async def help_slash(interaction: discord.Interaction, command_name: str = None)
 • `ml` - Run machine learning analysis
 • `plot <start_date> [end_date]` - Generate weather plots for specified dates
 • `monitor` - Monitor data collection status
+• `ifconfig` - Show Raspberry Pi network information
+• `top` - Show Raspberry Pi system status
 • `who` - Show information about the bot
 • `help` - Show this help message
 • `help <command>` - Show detailed help for a specific command
@@ -669,7 +701,8 @@ async def help_slash(interaction: discord.Interaction, command_name: str = None)
 `/monitor enable` - Enable data collection monitoring
 `/monitor disable` - Disable data collection monitoring
 `/monitor status` - Check current monitoring status
-"""
+`/ifconfig` - Show Pi network status
+`/top` - Show Pi system status"""
         await interaction.followup.send(help_text)
 
 
@@ -841,6 +874,8 @@ async def on_command_error(ctx, error):
 • `ml` - Run machine learning analysis
 • `plot <start_date> [end_date]` - Generate weather plots for specified dates
 • `monitor` - Monitor data collection status
+• `ifconfig` - Show Raspberry Pi network information
+• `top` - Show Raspberry Pi system status
 • `who` - Show information about the bot
 • `help` - Show this help message
 • `help <command>` - Show detailed help for a specific command