1
0
Fork 0
mirror of https://github.com/roles-ansible/ansible_role_i3wm.git synced 2024-08-16 10:09:53 +02:00
ansible_role_i3wm/files/i3blocks/wlan.py

79 lines
2.1 KiB
Python
Raw Normal View History

2023-08-22 14:53:13 +02:00
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
import os
import re
import subprocess
import sys
import argparse
def find_interface(interfaces):
for iface in interfaces:
if os.path.exists(f"/sys/class/net/{iface}"):
return iface
return None
def get_signal_quality(interface):
try:
iw_output = subprocess.check_output(["/sbin/iw", "dev", interface, "link"], text=True)
signal_match = re.search(r'(\-\d{2}) dBm', iw_output)
return int(signal_match.group(1)) if signal_match else None
except subprocess.CalledProcessError:
return None
def get_ssid(interface):
try:
iw_output = subprocess.check_output(["/sbin/iw", "dev", interface, "link"], text=True)
ssid_match = re.search(r'SSID: (.+)', iw_output)
return ssid_match.group(1).encode("utf-8").decode("unicode-escape") if ssid_match else None
except subprocess.CalledProcessError:
return None
def print_color(quality):
if quality >= 70:
print("#00FF00")
elif quality >= 50:
print("#CCFF00")
elif quality >= 30:
print("#FFFF00")
elif quality >= 10:
print("#FFAA00")
else:
print("#FF0000")
def main():
interfaces = ["wlp3s0", "wlan0"]
parser = argparse.ArgumentParser(description="Display WiFi signal strength and SSID")
parser.add_argument("--interface", help="Specify the wireless interface")
args = parser.parse_args()
if args.interface:
interface = args.interface
else:
interface = find_interface(interfaces)
if not interface:
sys.exit(0)
signal_quality = get_signal_quality(interface)
ssid = get_ssid(interface)
if signal_quality is None:
sys.exit(1)
# quality = int((signal_quality / 70) * 100) # Adjust the calculation as needed
quality = int(signal_quality) # Adjust the calculation as needed
emoji_signal = "📶"
if ssid:
print(f"{emoji_signal} {ssid} {quality}dBm")
else:
print(f"{emoji_signal} {quality}dBm")
print_color(quality)
if __name__ == "__main__":
main()