Tooltron

#!/usr/bin/python

import re

import sys

import serial

import MySQLdb

import getpass

TT_GET_KEY = 'k'

TT_ACK     = 'a'

TT_NACK    = 'n'

TT_TO      = 'f'

TT_TIMEOUT = 't'

BAUD_RATE = 9600

tools = {

   'Bandsaw':'1',

   'DrillPress':'2',

   'Mill':'3',

   'Lathe':'4',

   'Welder':'5',

   'CircSaw':'6',

   'ChopMiterSaw':'7',

   'JigSaw':'8',

   'Router':'9'}

if len(sys.argv) < 2:

    print "usage: dustmite.py /path/to/keypad/dev"

else:

    pw = getpass.getpass("mysql password: ")

    db = MySQLdb.connect(host="roboclub8.frc.ri.cmu.edu", user="tooltron", passwd=pw, db="civicrm")

    print "connected, now accepting input"

#TODO: errors!

    cursor = db.cursor()

    qry = "SELECT tools_6 FROM civicrm_value_roboclub_info_2 WHERE card_number_1 = "

    keypad = serial.Serial(sys.argv[1], BAUD_RATE, timeout=10) 

    keypad.flushInput()

    while True:

#TODO: only consider the last valid id so only the last swipe since a button press will work

        s = raw_input()

        id = re.search('%([0-9]*)=.*', s)

        if id != None:

            print "-----------\nid# ", id.group(1)

            print "waiting for key..."

            keypad.write(TT_GET_KEY)

            cursor.execute(qry + id.group(1))

            result = cursor.fetchall()

            acl = []

            for r in result:

                tls = r[0].split("\x01")

                for t in tls:

                    if t != '':

                        try:

                            acl.append (tools[t])

                        except KeyError:

                            print "ERROR: could not find key!"

            print "user has access to:", acl

            resp = keypad.read(1)

            print "request:",resp

            if acl.count(resp) > 0:

                keypad.write(TT_ACK)

            else:

                keypad.write(TT_NACK)