Updated code formatting with ruff + supressed warnings

Author: sidsun1

cli/ai/ai_monitor.py

@@ -15,7 +15,6 @@
 from langchain.schema import SystemMessage, HumanMessage
 from rich.console import Console
 from rich.prompt import Confirm
-from rich.table import Table
 from rich.panel import Panel
 
 load_dotenv()
@@ -96,7 +95,7 @@ def run_cmd(cmd:str) -> str:
             )
 
             return result.stdout.strip()
-        except Exception as e:
+        except Exception:
             # When testing I'll see what exactly the error is then isolate it with correct error exceptons
             pass
 
@@ -206,8 +205,8 @@ def execute(self, command: str, require_confirm: bool = True) -> Tuple[bool, str
 
             return result.returncode == 0, result.stdout
 
-        except Exception as e:
-            return False, str("Error is: ", e)
+        except Exception as err:
+            return False, str("Error is: ", err)
 
 
 
@@ -287,7 +286,7 @@ def _prepare_data(self, snapshot: SystemSnapshot) -> str:
 
         else:
             structured = {
-                "load_per_core": [l/snapshot.cpu_cores for l in snapshot.load_avg],
+                "load_per_core": [line/snapshot.cpu_cores for line in snapshot.load_avg],
                 "memory_usage_percent": (snapshot.memory_db["used"] / snapshot.memory_db["total"]) * 100,
                 "disk_usage_percent": snapshot.disk_usage_percent
             }
@@ -462,8 +461,8 @@ def run_ai_monitor(
         )
         monitor.monitor_loop()
 
-    except Exception as e:
-        Console().print(f"[red]Error: {str(e)}[/red]")
+    except Exception as err:
+        Console().print(f"[red]Error: {str(err)}[/red]")
         Console().print("[yellow]Check the API key and try again[/yellow]")
 
 

cli/app.py

@@ -1,18 +1,22 @@
+import os
 import subprocess
 import typer
 from rich import print
 from typing import Annotated
-import os
 
-app = typer.Typer(help = 'Surge - A DevOps CLI Tool For System Monitoring and Production Reliability')
+app = typer.Typer(
+    help="Surge - A DevOps CLI Tool For System Monitoring and Production Reliability"
+)
 
 
 def run_cmd(cmd: str) -> str:
     """
     Helper function to abstract lengthy subprocess command implementation :D
     """
 
-    return subprocess.run(cmd, shell=True, capture_output=True, text=True).stdout.strip()
+    return subprocess.run(
+        cmd, shell=True, capture_output=True, text=True
+    ).stdout.strip()
 
 
 def get_load() -> tuple[float] | int:
@@ -21,8 +25,8 @@ def get_load() -> tuple[float] | int:
     """
 
     uptime = run_cmd("uptime | awk -F'average:' '{print $2}'")
-    averages = [float(x.replace(',', '')) for x in uptime.split()]
-    cores = run_cmd('nproc')
+    averages = [float(x.replace(",", "")) for x in uptime.split()]
+    cores = run_cmd("nproc")
 
     return averages, cores
 
@@ -31,7 +35,7 @@ def get_cpu() -> tuple[float]:
     """
     Uses top to find CPU utilization grouped by user, system, and idle percents.
     """
-    top = run_cmd('top -bn1 | grep "Cpu(s)"').split(',')
+    top = run_cmd('top -bn1 | grep "Cpu(s)"').split(",")
     user = top[0].split()[-2]
     system = top[1].split()[0]
     idle = top[3].split()[0]
@@ -43,7 +47,7 @@ def get_memory() -> tuple[float]:
     """
     Returns remaining memory by group using free.
     """
-    free = run_cmd('free -m | grep Mem').split()
+    free = run_cmd("free -m | grep Mem").split()
     total, used, free_mem = free[1], free[2], free[3]
 
     return total, used, free_mem
@@ -53,7 +57,7 @@ def get_disk() -> tuple[float]:
     """
     Returns disk usage with df.
     """
-    df = run_cmd('df -h / | tail -1').split()
+    df = run_cmd("df -h / | tail -1").split()
     size, used, available, percent = df[1], df[2], df[3], df[4]
 
     return size, used, available, percent
@@ -66,13 +70,21 @@ def get_io() -> tuple[float]:
 
 @app.command()
 def monitor(
-    load: Annotated[bool, typer.Option('-l', '--load', help = 'Show system load averages')] = True,
-    cpu: Annotated[bool, typer.Option('-c', '--cpu', help = 'Show CPU usage')] = True,
-    ram: Annotated[bool, typer.Option('-r', '--ram', help = 'Show RAM usage')] = True,
-    disk: Annotated[bool, typer.Option('-d', '--disk', help = 'Show Disk usage')] = True,
-    io: Annotated[bool, typer.Option('-o', '--io', help = 'Show Disk I/O statistics')] = True,
-    interval: Annotated[int, typer.Option('-i', '--interval', help = 'Polling interval in seconds')] = 5,
-    verbose: Annotated[bool, typer.Option('-v', '--verbose', help = 'Show detailed system metrics')] = False
+    load: Annotated[
+        bool, typer.Option("-l", "--load", help="Show system load averages")
+    ] = True,
+    cpu: Annotated[bool, typer.Option("-c", "--cpu", help="Show CPU usage")] = True,
+    ram: Annotated[bool, typer.Option("-r", "--ram", help="Show RAM usage")] = True,
+    disk: Annotated[bool, typer.Option("-d", "--disk", help="Show Disk usage")] = True,
+    io: Annotated[
+        bool, typer.Option("-o", "--io", help="Show Disk I/O statistics")
+    ] = True,
+    interval: Annotated[
+        int, typer.Option("-i", "--interval", help="Polling interval in seconds")
+    ] = 5,
+    verbose: Annotated[
+        bool, typer.Option("-v", "--verbose", help="Show detailed system metrics")
+    ] = False,
 ):
     """
     Summary of all system metrics, including utilization of CPU, Memory, Network, and I/O.
@@ -85,82 +97,95 @@ def monitor(
     #   if verbose:
     #       ...
 
-
     if load:
         averages, cores = get_load()
 
-        print('\n[bold]System Load Averages[/bold]')
-        print('---------------------')
+        print("\n[bold]System Load Averages[/bold]")
+        print("---------------------")
 
         if not verbose:
-            print(f'Load avg (1m): {averages[0]}')
-            print(f'Load avg (5m): {averages[1]}')
-            print(f'Load avg (15m): {averages[2]}')
+            print(f"Load avg (1m): {averages[0]}")
+            print(f"Load avg (5m): {averages[1]}")
+            print(f"Load avg (15m): {averages[2]}")
         else:
-            print(f'Load avg (1m): {averages[0]:.2f} ({averages[0] / cores:.3f} per CPU)')
-            print(f'Load avg (5m): {averages[1]:.2f} ({averages[1] / cores:.3f} per CPU)')
-            print(f'Load avg (15m): {averages[2]:.2f} ({averages[2] / cores:.3f} per CPU)')
-    
+            print(
+                f"Load avg (1m): {averages[0]:.2f} ({averages[0] / cores:.3f} per CPU)"
+            )
+            print(
+                f"Load avg (5m): {averages[1]:.2f} ({averages[1] / cores:.3f} per CPU)"
+            )
+            print(
+                f"Load avg (15m): {averages[2]:.2f} ({averages[2] / cores:.3f} per CPU)"
+            )
+
     if cpu:
         user, system, idle = get_cpu()
-        print('\n[bold]CPU Utilization[/bold]')
-        print('---------------------')
-        print(f'User: {user}% | System: {system}% | Idle: {idle}%')
-    
+        print("\n[bold]CPU Utilization[/bold]")
+        print("---------------------")
+        print(f"User: {user}% | System: {system}% | Idle: {idle}%")
+
     if ram:
         total, used, free = get_memory()
-        print('\n[bold]Memory Usage (MB)[/bold]')
-        print('---------------------')
-        print(f'Total: {total} | Used: {used} | Free: {free}')
-    
+        print("\n[bold]Memory Usage (MB)[/bold]")
+        print("---------------------")
+        print(f"Total: {total} | Used: {used} | Free: {free}")
+
     if disk:
         size, used, available, percent = get_disk()
-        print('\n[bold]Disk Usage[/bold]')
-        print('---------------------')
-        print(f'Size: {size} | Used: {used} | Available: {available} | Usage: {percent}')
+        print("\n[bold]Disk Usage[/bold]")
+        print("---------------------")
+        print(
+            f"Size: {size} | Used: {used} | Available: {available} | Usage: {percent}"
+        )
 
 
 @app.command()
 def network(
-    url: Annotated[str, typer.Argument(help = 'URL to test network/API metrics')],
-    requests: Annotated[int, typer.Option('-n', '--count', help = 'Number of requests to send')] = 5
+    url: Annotated[str, typer.Argument(help="URL to test network/API metrics")],
+    requests: Annotated[
+        int, typer.Option("-n", "--count", help="Number of requests to send")
+    ] = 5,
 ):
     """
     Run basic network/API tests with a number of requests.
     """
-    print(f'Testing network connection to {url} with {requests} requests.')
+    print(f"Testing network connection to {url} with {requests} requests.")
     # TODO: Add http requests or use curl through subprocess
 
 
 @app.command()
 def ai(
-    format: Annotated[str, typer.Option("--format", '-f',help='Data format: raw/structured/hybrid')] = "hybrid",
-    verbosity: Annotated[str, typer.Option('--verbosity', '-v', help="Output detail: concise/normal/hybrid")] = 'normal',
-    auto_fix: Annotated[bool, typer.Option("--auto-fix", help="Auto-execute safe fixes")] = False,
+    format: Annotated[
+        str, typer.Option("--format", "-f", help="Data format: raw/structured/hybrid")
+    ] = "hybrid",
+    verbosity: Annotated[
+        str,
+        typer.Option("--verbosity", "-v", help="Output detail: concise/normal/hybrid"),
+    ] = "normal",
+    auto_fix: Annotated[
+        bool, typer.Option("--auto-fix", help="Auto-execute safe fixes")
+    ] = False,
 ):
     """
     Using Gemini 2.5 Flash for now for simple AI suggestions and reading through machine metrics
     giving suggest fixes upon user confirmations
     """
 
-    if not os.getenv('GEMINI_API_KEY'):
-        print('[red]Error: GEMINI_API_KEY is empty[/red]')
+    if not os.getenv("GEMINI_API_KEY"):
+        print("[red]Error: GEMINI_API_KEY is empty[/red]")
         print('Set it with: export GEMINI_API_KEY="your api key"')
         return
 
     try:
         from ai.ai_monitor import run_ai_monitor
-        
-        run_ai_monitor(
-            data_format=format,
-            verbosity=verbosity,
-            auto_fix=auto_fix
-        )
+
+        run_ai_monitor(data_format=format, verbosity=verbosity, auto_fix=auto_fix)
     except ImportError:
-        print('[red]AI packages not installed[/red]')
-        print('Run: pip install langchain langchain-google-genai rich')
+        print("[red]AI packages not installed[/red]")
+        print("Run: pip install langchain langchain-google-genai rich")
     except Exception as e:
-        print(f'[red]Error: {str(e)}[/red]')
+        print(f"[red]Error: {str(e)}[/red]")
+
 
 if __name__ == "__main__":
-    app()
+    app()

cli/requirements.txt

@@ -41,6 +41,7 @@ requests
 requests-toolbelt
 rich
 rsa
+ruff
 shellingham
 sniffio
 SQLAlchemy