123 lines
3.4 KiB
Python
123 lines
3.4 KiB
Python
#!/usr/bin/env python3
|
|
|
|
import sys
|
|
import json
|
|
from hysteria2_api import Hysteria2Client
|
|
import time
|
|
from init_paths import *
|
|
from paths import *
|
|
|
|
|
|
def get_secret() -> str:
|
|
if not CONFIG_FILE.exists():
|
|
print("Error: config.json file not found!", file=sys.stderr)
|
|
sys.exit(1)
|
|
|
|
with CONFIG_FILE.open() as f:
|
|
data = json.load(f)
|
|
|
|
secret = data.get("trafficStats", {}).get("secret")
|
|
if not secret:
|
|
print("Error: secret not found in config.json!", file=sys.stderr)
|
|
sys.exit(1)
|
|
|
|
return secret
|
|
|
|
|
|
def convert_bytes(bytes_val: int) -> str:
|
|
units = [("TB", 1 << 40), ("GB", 1 << 30), ("MB", 1 << 20), ("KB", 1 << 10)]
|
|
for unit, factor in units:
|
|
if bytes_val >= factor:
|
|
return f"{bytes_val / factor:.2f} {unit}"
|
|
return f"{bytes_val} B"
|
|
|
|
|
|
def get_cpu_usage(interval: float = 0.1) -> float:
|
|
def read_cpu_times():
|
|
with open("/proc/stat") as f:
|
|
line = f.readline()
|
|
fields = list(map(int, line.strip().split()[1:]))
|
|
idle, total = fields[3], sum(fields)
|
|
return idle, total
|
|
|
|
idle1, total1 = read_cpu_times()
|
|
time.sleep(interval)
|
|
idle2, total2 = read_cpu_times()
|
|
|
|
idle_delta = idle2 - idle1
|
|
total_delta = total2 - total1
|
|
cpu_usage = 100.0 * (1 - idle_delta / total_delta) if total_delta else 0.0
|
|
return round(cpu_usage, 1)
|
|
|
|
|
|
|
|
def get_memory_usage() -> tuple[int, int]:
|
|
with open("/proc/meminfo") as f:
|
|
lines = f.readlines()
|
|
|
|
mem_total = int(next(line for line in lines if "MemTotal" in line).split()[1]) // 1024
|
|
mem_available = int(next(line for line in lines if "MemAvailable" in line).split()[1]) // 1024
|
|
mem_used = mem_total - mem_available
|
|
|
|
return mem_total, mem_used
|
|
|
|
|
|
|
|
def get_online_user_count(secret: str) -> int:
|
|
try:
|
|
client = Hysteria2Client(
|
|
base_url=API_BASE_URL,
|
|
secret=secret
|
|
)
|
|
online_users = client.get_online_clients()
|
|
return sum(1 for user in online_users.values() if user.is_online)
|
|
except Exception as e:
|
|
print(f"Error getting online users: {e}", file=sys.stderr)
|
|
return 0
|
|
|
|
|
|
def get_total_traffic() -> tuple[int, int]:
|
|
if not USERS_FILE.exists():
|
|
return 0, 0
|
|
|
|
try:
|
|
with USERS_FILE.open() as f:
|
|
users = json.load(f)
|
|
|
|
total_upload = 0
|
|
total_download = 0
|
|
|
|
for user_data in users.values():
|
|
total_upload += int(user_data.get("upload_bytes", 0) or 0)
|
|
total_download += int(user_data.get("download_bytes", 0) or 0)
|
|
|
|
return total_upload, total_download
|
|
except Exception as e:
|
|
print(f"Error parsing traffic data: {e}", file=sys.stderr)
|
|
return 0, 0
|
|
|
|
|
|
|
|
def main():
|
|
secret = get_secret()
|
|
|
|
cpu_usage = get_cpu_usage()
|
|
mem_total, mem_used = get_memory_usage()
|
|
online_users = get_online_user_count(secret)
|
|
|
|
print(f"📈 CPU Usage: {cpu_usage}")
|
|
print(f"📋 Total RAM: {mem_total}MB")
|
|
print(f"💻 Used RAM: {mem_used}MB")
|
|
print(f"👥 Online Users: {online_users}")
|
|
print()
|
|
|
|
total_upload, total_download = get_total_traffic()
|
|
|
|
print(f"🔼 Uploaded Traffic: {convert_bytes(total_upload)}")
|
|
print(f"🔽 Downloaded Traffic: {convert_bytes(total_download)}")
|
|
print(f"📊 Total Traffic: {convert_bytes(total_upload + total_download)}")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|