pi_kismet.py

import sys, time
sys.path.append("/home/pi/Adafruit-Raspberry-Pi-Python-Code/Adafruit_CharLCDPlate")
sys.path.append("/home/pi/kismetclient")
from Adafruit_CharLCDPlate import Adafruit_CharLCDPlate
from pi_kismet_display import KismetDisplay
from kismetclient import Client
from kismetclient import handlers

HOLD_TIME = 3.0
REFRESH_TIME = 5.0
display = KismetDisplay()
lcd = display.lcd
prevCol = -1
prev = -1
lastTime = time.time()

def shutdown():
    lcd.clear()
    exit(0)

total = wpa = wep = none = wps = 0
aps = []
k = Client()

def count_crypts(client, name, macaddr, cryptstring):
    global aps, total, wpa, wep, none, wps
    pairing = (name, macaddr)
    if pairing not in aps:
        aps.append(pairing)
        if 'None' in cryptstring:
            none += 1
        elif 'WPA' in cryptstring:
            wpa += 1
        elif 'WEP' in cryptstring:
            wep += 1
        elif 'WPS' in cryptstring:
            wpa += 1
            wps += 1
        elif cryptstring == '':
            none += 1
        total = len(aps)

k.register_handler('DEVICE', count_crypts)

# Listen for button presses
while True:
    k.listen()
    display.update_screens(wpa,wep,wps,none,total)
    b = lcd.buttons()
    if b is not prev:
        if lcd.buttonPressed(lcd.SELECT):
            tt = time.time() # Start time of button press
            while lcd.buttonPressed(lcd.SELECT): # Wait for button release
                if (time.time() - tt) >= HOLD_TIME: # Extended hold?
                    shutdown() # We're outta here
            display.backlightStep()
        elif lcd.buttonPressed(lcd.LEFT):
                display.scrollRight()
        elif lcd.buttonPressed(lcd.RIGHT):
                display.scrollLeft()
        elif lcd.buttonPressed(lcd.UP):
                display.modeUp()
        elif lcd.buttonPressed(lcd.DOWN):
                display.modeDown()
        prev = b
        lastTime = time.time()
    else:
        now = time.time()
        since = now - lastTime
        if since > REFRESH_TIME or since < 0.0:
            display.update()
            lastTime = now