Add CLI arguments for continuous monitoring

This commit is contained in:
m1ngsama 2024-12-19 11:30:00 +08:00
parent 524a7d684b
commit dbfb8e04b0

View file

@ -5,6 +5,7 @@ System Tracker - Monitor machine health and performance
import psutil
import time
import argparse
from datetime import datetime
from process_monitor import ProcessMonitor
@ -69,5 +70,19 @@ class SystemTracker:
if __name__ == "__main__":
parser = argparse.ArgumentParser(description='System Tracker - Monitor machine health')
parser.add_argument('-c', '--continuous', action='store_true', help='Run continuously')
parser.add_argument('-i', '--interval', type=int, default=5, help='Update interval in seconds')
args = parser.parse_args()
tracker = SystemTracker()
if args.continuous:
try:
while True:
tracker.display_stats()
time.sleep(args.interval)
except KeyboardInterrupt:
print("\n\nTracker stopped by user")
else:
tracker.display_stats()