dotfiles/root/home/bhasher/.config/waybar/scripts/bandwidth

67 lines
1.8 KiB
Python
Executable File

#!/usr/bin/env python
import subprocess
from time import sleep
def default_interface():
process = subprocess.run(
["ip", "route"], check=True, text=True, capture_output=True
)
for line in process.stdout.splitlines():
if line.startswith("default via"):
return line.split()[4]
raise RuntimeError("No default interface found")
def get_rx_tx_bytes(iface):
with open("/proc/net/dev") as f:
for line in f:
line = line.strip()
if not line.startswith(f"{iface}:"):
continue
rx_bytes = int(line.split()[1])
tx_bytes = int(line.split()[9])
return rx_bytes, tx_bytes
raise RuntimeError(f"Interface {iface} not found")
def format_size(size):
power_labels = {0: "B", 1: "K", 2: "M", 3: "G", 4: "T"}
#kilo = 2**10
#power = 0
#while size > kilo:
# size /= kilo
# power += 1
power = 1
return f"{size/1024:3.0f} {power_labels[power]}"
def main():
refresh_interval = 1
rx_icon = ""
tx_icon = ""
num_left = "<span font='NotoSansMono'>"
num_right = "</span>"
fmt_str = (
f"{num_left}{{rx}}{{unit_suffix}} {rx_icon}{num_right} "
f"{tx_icon}{num_left}{{tx}}{{unit_suffix}}{num_right}"
)
unit_suffix = ""
iface = default_interface()
rx_bytes, tx_bytes = get_rx_tx_bytes(iface)
while True:
prev_rx_bytes, prev_tx_bytes = rx_bytes, tx_bytes
rx_bytes, tx_bytes = get_rx_tx_bytes(iface)
drx = format_size((rx_bytes - prev_rx_bytes) / refresh_interval)
dtx = format_size((tx_bytes - prev_tx_bytes) / refresh_interval)
line = fmt_str.format(rx=drx, tx=dtx, unit_suffix=unit_suffix)
print(line, flush=True)
sleep(refresh_interval)
if __name__ == "__main__":
main()