devops_tool.py

import subprocess
from pathlib import Path
import typer, psutil
from rich import print

app = typer.Typer(help='DevOps utility CLI')

@app.command()
def health():
    print({
        'cpu_percent': psutil.cpu_percent(interval=1),
        'memory_percent': psutil.virtual_memory().percent,
        'disk_percent': psutil.disk_usage('/').percent,
        'boot_time': psutil.boot_time(),
    })

@app.command()
def ports(port: int = typer.Option(..., help='TCP port to inspect')):
    out = subprocess.check_output(['sh', '-lc', f'ss -lntp | grep :{port} || true'], text=True)
    print(out or f'No listening process found on port {port}')

@app.command()
def docker_clean(dry_run: bool = True):
    cmd = 'docker system prune -af --volumes'
    if dry_run:
        print(f'[yellow]Dry run.[/yellow] Would run: {cmd}')
    else:
        subprocess.run(['sh', '-lc', cmd], check=False)

@app.command()
def nginx_config(server_name: str, upstream: str, out: Path = Path('./nginx.conf')):
    content = f"""
server {{
    listen 80;
    server_name {server_name};

    location / {{
        proxy_pass http://{upstream};
        proxy_set_header Host $host;
        proxy_set_header X-Forwarded-For $proxy_add_x_forwarded_for;
        proxy_set_header X-Real-IP $remote_addr;
    }}
}}
"""
    out.write_text(content.strip() + '\n')
    print(f'Wrote {out}')

if __name__ == '__main__':
    app()