You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
52 lines
1.6 KiB
Python
52 lines
1.6 KiB
Python
5 years ago
|
"""
|
||
|
|
Module contains all logic for retrieving CPU, Memory, and Disk stats.
|
||
|
|
"""
|
||
|
|
import psutil
|
||
|
|
import datetime as dt
|
||
|
|
|
||
|
|
from typing import Dict, Any
|
||
|
|
|
||
|
|
STATIC_STATS = {"cpu_count": psutil.cpu_count(logical=True),
|
||
|
|
"boot_time": dt.datetime.fromtimestamp(psutil.boot_time())}
|
||
|
|
|
||
|
|
|
||
|
|
def get_stats() -> Dict[str, Any]:
|
||
|
|
"""Main entrypoint for retrieving system stats."""
|
||
|
|
load_avg = [round(x / psutil.cpu_count(logical=True) * 100, 2) for x in psutil.getloadavg()]
|
||
|
|
mem = psutil.virtual_memory()
|
||
|
|
disk = psutil.disk_usage('/')
|
||
|
|
stats = {'load_avg': load_avg,
|
||
|
|
'mem': {
|
||
|
|
'total': mem.total,
|
||
|
|
'used': mem.used,
|
||
|
|
'free': mem.free,
|
||
|
|
'shared': mem.shared if hasattr(mem, 'shared') else None,
|
||
|
|
'buffers': mem.buffers if hasattr(mem, 'buffers') else None,
|
||
|
|
'cached': mem.cached if hasattr(mem, 'cached') else None,
|
||
|
|
'available': mem.available,
|
||
|
|
'percent': mem.percent,
|
||
|
|
},
|
||
|
|
'tasks': _get_tasks(),
|
||
|
|
}
|
||
|
|
stats.update(STATIC_STATS)
|
||
|
|
return stats
|
||
|
|
|
||
|
|
|
||
|
|
def _get_tasks() -> Dict[str, int]:
|
||
|
|
"""Retrieves a dictionary of task information."""
|
||
|
|
tasks = {
|
||
|
|
'total': 0,
|
||
|
|
'running': 0,
|
||
|
|
'sleeping': 0,
|
||
|
|
}
|
||
|
|
|
||
|
|
for p in psutil.process_iter():
|
||
|
|
with p.oneshot():
|
||
|
|
if p.status() == psutil.STATUS_RUNNING:
|
||
|
|
tasks['running'] += 1
|
||
|
|
if p.status() == psutil.STATUS_SLEEPING:
|
||
|
|
tasks['sleeping'] += 1
|
||
|
|
tasks['total'] += 1
|
||
|
|
|
||
|
|
return tasks
|