Add initial scripts for LED control and network scanning

2025-11-11 19:30:43 -05:00
parent 13f5e6465a
commit 9d8df6bcbe
13 changed files with 288 additions and 0 deletions
--- a/network_scan.py
+++ b/network_scan.py
@@ -0,0 +1,75 @@
+import network
+import time
+
+def scan_networks(timeout=10):
+    """
+    Scan for available WiFi networks and return formatted results.
+    Returns list of dicts with network info.
+    """
+    wlan = network.WLAN(network.STA_IF)
+    wlan.active(True)
+    
+    print("Scanning for networks...")
+    networks = wlan.scan()
+    
+    # Format results
+    results = []
+    for net in networks:
+        ssid, bssid, channel, rssi, security, hidden = net
+        
+        # Decode SSID if bytes
+        if isinstance(ssid, bytes):
+            ssid = ssid.decode('utf-8')
+        
+        # Security type mapping
+        security_types = {
+            0: "Open",
+            1: "WEP",
+            2: "WPA-PSK",
+            3: "WPA2-PSK",
+            4: "WPA/WPA2-PSK"
+        }
+        
+        results.append({
+            "ssid": ssid,
+            "bssid": ":".join(["%02X" % b for b in bssid]),
+            "channel": channel,
+            "rssi": rssi,
+            "security": security_types.get(security, "Unknown"),
+            "hidden": bool(hidden)
+        })
+    
+    # Sort by signal strength (strongest first)
+    results.sort(key=lambda x: x["rssi"], reverse=True)
+    
+    return results
+
+def print_networks(networks):
+    """Pretty print network scan results."""
+    print("\n" + "="*70)
+    print(f"Found {len(networks)} networks:")
+    print("="*70)
+    
+    for i, net in enumerate(networks, 1):
+        print(f"\n{i}. {net['ssid']}")
+        print(f"   BSSID: {net['bssid']}")
+        print(f"   Channel: {net['channel']}")
+        print(f"   Signal: {net['rssi']} dBm")
+        print(f"   Security: {net['security']}")
+        if net['hidden']:
+            print(f"   (Hidden Network)")
+    
+    print("\n" + "="*70)
+
+def find_network(ssid):
+    """Find a specific network by SSID."""
+    networks = scan_networks()
+    for net in networks:
+        if net['ssid'] == ssid:
+            return net
+    return None
+
+# Run scan if executed directly
+if __name__ == "__main__":
+    networks = scan_networks()
+    print_networks(networks)