服务器带宽流量

import psutil
import time

def get_bandwidth_usage():
    net_io = psutil.net_io_counters()
    bytes_sent = net_io.bytes_sent
    bytes_recv = net_io.bytes_recv
    return bytes_sent, bytes_recv

def main():
    prev_bytes_sent = 0
    prev_bytes_recv = 0
    while True:
        time.sleep(1)
        curr_bytes_sent, curr_bytes_recv = get_bandwidth_usage()
        sent_speed = (curr_bytes_sent - prev_bytes_sent) / 1024  # KB/s
        recv_speed = (curr_bytes_recv - prev_bytes_recv) / 1024  # KB/s
        print(f"Upload Speed: {sent_speed:.2f} KB/s, Download Speed: {recv_speed:.2f} KB/s")
        prev_bytes_sent = curr_bytes_sent
        prev_bytes_recv = curr_bytes_recv

if __name__ == "__main__":
    main()