mirror of
https://github.com/imayushsaini/Bombsquad-Ballistica-Modded-Server.git
synced 2025-10-20 00:00:39 +00:00
65 lines
1.9 KiB
Python
65 lines
1.9 KiB
Python
"""
|
|
Simple utility to monitor current CPU,Network,RAM usage.
|
|
|
|
Author: ChatGpt
|
|
"""
|
|
|
|
|
|
from . import logger
|
|
|
|
import asyncio
|
|
import psutil
|
|
|
|
from threading import Thread
|
|
|
|
async def monitor_network():
|
|
elapsed_time = 60
|
|
|
|
while True:
|
|
# Wait for 60 seconds
|
|
await asyncio.sleep(elapsed_time)
|
|
|
|
# Get the current network counters
|
|
net_io_counters = psutil.net_io_counters()
|
|
|
|
# Calculate the upload and download speeds in bytes/sec
|
|
# seconds
|
|
upload_speed = (net_io_counters.bytes_sent - monitor_network.bytes_sent) / elapsed_time
|
|
download_speed = (net_io_counters.bytes_recv - monitor_network.bytes_recv) / elapsed_time
|
|
|
|
|
|
# Convert the speeds to human-readable format
|
|
upload_speed = f"{upload_speed/1024:.2f} KB/s"
|
|
download_speed = f"{download_speed/1024:.2f} KB/s"
|
|
|
|
logger.log(f"CPU: {psutil.cpu_percent()}% RAM: {psutil.virtual_memory().percent}% Upload: {upload_speed} Download: {download_speed}")
|
|
|
|
# Update the counters
|
|
monitor_network.bytes_sent = net_io_counters.bytes_sent
|
|
monitor_network.bytes_recv = net_io_counters.bytes_recv
|
|
|
|
async def reset_counters():
|
|
while True:
|
|
# Wait for 2+2 minutes
|
|
await asyncio.sleep(120 + 120)
|
|
|
|
# Reset the network counters for each network interface separately
|
|
psutil.net_io_counters(pernic=True)
|
|
|
|
def main():
|
|
# Initialize the counters
|
|
net_io_counters = psutil.net_io_counters()
|
|
monitor_network.bytes_sent = net_io_counters.bytes_sent
|
|
monitor_network.bytes_recv = net_io_counters.bytes_recv
|
|
|
|
# Start the network monitoring and counter resetting coroutines
|
|
loop = asyncio.new_event_loop()
|
|
|
|
# Start the event loop in a separate thread
|
|
t = Thread(target=loop.run_forever)
|
|
t.start()
|
|
|
|
# Schedule the coroutines in the event loop
|
|
asyncio.run_coroutine_threadsafe(monitor_network(), loop=loop)
|
|
asyncio.run_coroutine_threadsafe(reset_counters(), loop=loop)
|
|
|