shield-stats 0.2.3__tar.gz

shield_stats-0.2.3/PKG-INFO

@@ -0,0 +1,8 @@
+Metadata-Version: 2.4
+Name: shield-stats
+Version: 0.2.3
+Summary: Lightweight cross-distro Linux telemetry collector
+Author: Reyansh Raj Mishra
+License: MIT
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown

shield_stats-0.2.3/pyproject.toml

@@ -0,0 +1,17 @@
+[build-system]
+requires = ["setuptools>=61"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "shield-stats"
+version = "0.2.3"
+description = "Lightweight cross-distro Linux telemetry collector"
+readme = "README.md"
+requires-python = ">=3.8"
+license = {text = "MIT"}
+authors = [
+    {name = "Reyansh Raj Mishra"}
+]
+
+[project.scripts]
+shield-stats = "shield_stats.cli:main"

shield_stats-0.2.3/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

shield_stats-0.2.3/shield_stats/__init__.py

@@ -0,0 +1,3 @@
+"""shield_stats package."""
+
+__version__ = "0.1.0"

shield_stats-0.2.3/shield_stats/cli.py

@@ -0,0 +1,152 @@
+import argparse
+import sys
+from . import collector, storage
+from .storage import load_snapshot_by_date
+from datetime import date, timedelta
+import subprocess
+
+
+def _cmd_record() -> int:
+    metrics = collector.collect_metrics()
+    try:
+        storage.save_snapshot(metrics)
+    except Exception as e:
+        print(f"failed to save snapshot: {e}", file=sys.stderr)
+        return 1
+    return 0
+
+
+def _format_bytes(n, signed=False):
+    value = float(n)
+    for unit in ["B", "KB", "MB", "GB", "TB"]:
+        if abs(value) < 1024:
+            if signed:
+                return f"{value:+.2f} {unit}"
+            return f"{value:.2f} {unit}"
+        value /= 1024
+    if signed:
+        return f"{value:+.2f} PB"
+    return f"{value:.2f} PB"
+
+def _cmd_summary() -> int:
+    today = date.today()
+    data = load_snapshot_by_date(today)
+
+    if not data:
+        print("No snapshot found for today.")
+        return 1
+
+    print("System Snapshot (Today)\n")
+
+    # Disk (using df command for accurate calculation)
+    try:
+        df_output = subprocess.check_output(['df', '-h', '/'], text=True).splitlines()
+        disk_usage = df_output[1].split()
+        used = disk_usage[2]  # Used space
+        total = disk_usage[1]  # Total space
+        percent_disk = disk_usage[4]  # Percentage used
+        print(f"Disk Usage:      {used} of {total} ({percent_disk})")
+    except Exception as e:
+        print(f"Disk Usage:      Error calculating disk usage: {e}")
+
+    # Memory
+    mem_total = data["memory_total_bytes"]
+    mem_used = data["memory_used_bytes"]
+
+    if mem_total and mem_used:
+        percent_mem = (mem_used / mem_total) * 100
+        print(f"Memory Usage:    {_format_bytes(mem_used)} ({percent_mem:.1f}%)")
+    else:
+        print("Memory Usage:    unavailable")
+
+    # Packages
+    print(f"Packages:        {data['package_count']}")
+
+    # Load
+    print(f"Load (1m):       {data['load_avg_1']:.2f}")
+
+    # Uptime
+    uptime = data["uptime_seconds"]
+    if uptime:
+        hours = int(uptime // 3600)
+        minutes = int((uptime % 3600) // 60)
+        print(f"Uptime:          {hours}h {minutes}m")
+
+    # Kernel
+    print(f"Kernel:          {data['kernel_version']}")
+
+    return 0
+
+def _cmd_compare() -> int:
+    today = date.today()
+    yesterday = today - timedelta(days=1)
+
+    today_data = load_snapshot_by_date(today)
+    yesterday_data = load_snapshot_by_date(yesterday)
+
+    if not today_data:
+        print("No snapshot found for today.")
+        return 1
+
+    if not yesterday_data:
+        print("No snapshot found for yesterday.")
+        return 1
+
+    print("Changes Since Yesterday:\n")
+
+    # Disk
+    disk_delta = today_data["disk_used_bytes"] - yesterday_data["disk_used_bytes"]
+    print(f"Disk Usage:      {_format_bytes(disk_delta, signed=True)}")
+
+    # Memory
+    mem_delta = (
+        (today_data["memory_used_bytes"] or 0)
+        - (yesterday_data["memory_used_bytes"] or 0)
+    )
+    print(f"Memory Usage:    {_format_bytes(mem_delta, signed=True)}")
+
+    # Packages
+    pkg_delta = today_data["package_count"] - yesterday_data["package_count"]
+    print(f"Packages:        {pkg_delta:+}")
+
+    # Load
+    load_delta = (
+        (today_data["load_avg_1"] or 0)
+        - (yesterday_data["load_avg_1"] or 0)
+    )
+    print(f"Load (1m):       {load_delta:+.2f}")
+
+    # Kernel change
+    if today_data["kernel_version"] != yesterday_data["kernel_version"]:
+        print(
+            f"Kernel:          {yesterday_data['kernel_version']} → {today_data['kernel_version']}"
+        )
+    else:
+        print("Kernel:          unchanged")
+
+    return 0
+
+def main(argv=None) -> int:
+    parser = argparse.ArgumentParser(prog="shield-stats")
+    sub = parser.add_subparsers(dest="command")
+    sub.required = True
+
+    sub.add_parser("record", help="collect and store today's metrics")
+    sub.add_parser("summary", help="show today's system summary")
+    sub.add_parser("compare", help="compare today with yesterday")
+
+    args = parser.parse_args(argv)
+
+    if args.command == "record":
+        return _cmd_record()
+
+    if args.command == "summary":
+        return _cmd_summary()
+
+    if args.command == "compare":
+        return _cmd_compare()
+
+    return 1
+
+if __name__ == "__main__":
+    raise SystemExit(main())

shield_stats-0.2.3/shield_stats/collector.py

@@ -0,0 +1,99 @@
+"""Gather system metrics for shield-stats."""
+
+import platform
+import shutil
+import subprocess
+import time
+from pathlib import Path
+from typing import Dict, Optional
+
+
+def _read_proc_file(path: Path) -> str:
+    try:
+        with path.open("r") as f:
+            return f.read()
+    except Exception:
+        return ""
+
+
+def _parse_meminfo() -> Dict[str, int]:
+    data = _read_proc_file(Path("/proc/meminfo"))
+    result: Dict[str, int] = {}
+    for line in data.splitlines():
+        parts = line.split()
+        if len(parts) >= 2 and parts[0].endswith(":"):
+            key = parts[0].rstrip(":")
+            try:
+                value = int(parts[1]) * 1024
+            except ValueError:
+                continue
+            result[key] = value
+    return result
+
+
+def _get_uptime() -> Optional[float]:
+    data = _read_proc_file(Path("/proc/uptime")).split()
+    try:
+        return float(data[0])
+    except (IndexError, ValueError):
+        return None
+
+
+def _get_load_avg() -> Optional[float]:
+    data = _read_proc_file(Path("/proc/loadavg")).split()
+    try:
+        return float(data[0])
+    except (IndexError, ValueError):
+        return None
+
+
+def _count_packages() -> int:
+    if shutil.which("pacman"):
+        cmd = ["pacman", "-Qq"]
+
+    elif shutil.which("apt"):
+        cmd = ["dpkg-query", "-f", "${binary:Package}\n", "-W"]
+
+    elif shutil.which("dnf"):
+        cmd = ["rpm", "-qa"]
+
+    else:
+        return 0
+
+    try:
+        proc = subprocess.run(cmd, capture_output=True, text=True, check=False)
+    except Exception:
+        return 0
+
+    if proc.returncode != 0:
+        return 0
+
+    return len(proc.stdout.strip().splitlines())
+
+
+def collect_metrics() -> Dict[str, Optional[object]]:
+    now = int(time.time())
+
+    total, used, _free = shutil.disk_usage("/")
+
+    meminfo = _parse_meminfo()
+    mem_total = meminfo.get("MemTotal")
+    mem_available = meminfo.get("MemAvailable")
+    mem_used = (
+        mem_total - mem_available
+        if mem_total is not None and mem_available is not None
+        else None
+    )
+
+    return {
+        "timestamp": now,
+        "hostname": platform.node(),
+        "disk_total_bytes": total,
+        "disk_used_bytes": used,
+        "memory_total_bytes": mem_total,
+        "memory_used_bytes": mem_used,
+        "uptime_seconds": _get_uptime(),
+        "load_avg_1": _get_load_avg(),
+        "package_count": _count_packages(),
+        "kernel_version": platform.release(),
+    }

shield_stats-0.2.3/shield_stats/storage.py

@@ -0,0 +1,41 @@
+"""Handle storage of collected metrics."""
+
+import json
+from datetime import date
+from pathlib import Path
+from typing import Any, Dict
+from datetime import date
+import json
+from pathlib import Path
+
+
+def load_snapshot_by_date(d: date):
+    base = Path.home() / ".local" / "share" / "shield-stats"
+    path = base / f"{d.isoformat()}.json"
+    if not path.exists():
+        return None
+    try:
+        with path.open("r", encoding="utf-8") as f:
+            return json.load(f)
+    except Exception:
+        return None
+
+def _get_storage_path() -> Path:
+    base = Path.home() / ".local" / "share" / "shield-stats"
+    base.mkdir(parents=True, exist_ok=True)
+    today = date.today().isoformat()
+    return base / f"{today}.json"
+
+
+def save_snapshot(data: Dict[str, Any]) -> None:
+    path = _get_storage_path()
+    temp = path.with_suffix(".tmp")
+
+    try:
+        with temp.open("w", encoding="utf-8") as f:
+            json.dump(data, f, indent=2, ensure_ascii=False)
+            f.flush()
+        temp.replace(path)
+    except Exception:
+        with path.open("w", encoding="utf-8") as f:
+            json.dump(data, f, indent=2, ensure_ascii=False)

shield_stats-0.2.3/shield_stats.egg-info/PKG-INFO

@@ -0,0 +1,8 @@
+Metadata-Version: 2.4
+Name: shield-stats
+Version: 0.2.3
+Summary: Lightweight cross-distro Linux telemetry collector
+Author: Reyansh Raj Mishra
+License: MIT
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown

shield_stats-0.2.3/shield_stats.egg-info/SOURCES.txt

@@ -0,0 +1,10 @@
+pyproject.toml
+shield_stats/__init__.py
+shield_stats/cli.py
+shield_stats/collector.py
+shield_stats/storage.py
+shield_stats.egg-info/PKG-INFO
+shield_stats.egg-info/SOURCES.txt
+shield_stats.egg-info/dependency_links.txt
+shield_stats.egg-info/entry_points.txt
+shield_stats.egg-info/top_level.txt

shield_stats-0.2.3/shield_stats.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

shield_stats-0.2.3/shield_stats.egg-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+shield-stats = shield_stats.cli:main

shield_stats-0.2.3/shield_stats.egg-info/top_level.txt

@@ -0,0 +1 @@
+shield_stats