Tooltron

#!/usr/bin/python

################################################################################

## dustmite.py is the main tooltron server

# It connects to the civicrm database of mysql and sends commands to

# the cardbox and tools.  It must be connected over usb to the card

# reader, and the two arguments specify the devices to use for the

# serial communication to the card and tool boxes.

#

# This script requires the modules seen below, which may not be

# present by default

#

# Because of a problem with card scans buffering in stdio, there is a

# seperate thread which stores the last swiped id. To cleanly exit the

# program type C-d

#

# email bradneuman@gmail.com if you have any questions

################################################################################

import re

import sys

import serial

import MySQLdb

import getpass

#tools for messages are these + 1 but in decimal, not ascii

#the values are lists of numbers which get appended to the acl

tools = {

   'Bandsaw':['2','8'],

   'DrillPress':['1','9'],

   'Mill':['4'],

   'Lathe':['5'],

   'ChopMiterSaw':['3']

}

TT_GET_KEY = 'k'

TT_ACK     = 'a'

TT_NACK    = 'n'

TT_TO      = 'f'

TT_TIMEOUT = 't'

BAUD_RATE = 9600

#################################################

#this code spawns a thread which always puts the last card id in the

#lastid variable.

#################################################

import threading

import time

lastid = None

idready = False

cv = threading.Condition()

class idThread ( threading.Thread ):

   def run ( self ):

       global lastid

       global idready

       try:

           while True:

               l = raw_input()

               cv.acquire()

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

               if id != None:

                   lastid = id

                   print "!!! setting idready"

                   idready = True

                   cv.notify()

               cv.release()

       except EOFError:

           cv.acquire()

           idready = True

           lastid=None

           cv.notify()

           cv.release()

#end thread

###############################################

def get_last_id():

    global idready

    global lastid

    got = None

    print ">>> locking for last_id"

    cv.acquire()

    while not idready:

        print ">>> wait"

        cv.wait()

    print ">>> idready set"

    got = lastid

    idready = False

    cv.release()

    print ">>> done"

    return got

def clear_id():

    global idready

    global lastid

    print "=== locking to clear idready flag"

    cv.acquire()

    idready = False

    lastid = None

    cv.release()

    print "=== done"

    return

if len(sys.argv) < 3:

    print "usage: dustmite.py /path/to/keypad/device /path/to/tool/bus/device"

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"

    #start the id scan thread AFTER the getpass prompt

    idThread().start()

    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 = 30) 

    keypad.flushInput()

    print keypad

    bus = serial.Serial(sys.argv[2], BAUD_RATE, timeout = 15)

    bus.flushInput()

    print bus

    # ^ <src> <dest> <data>

    def sendTool(t):

        tn = int(t) + 1 

        msg = '^' + chr(1) + chr(tn) + 'O' + chr(1 ^ tn ^ ord('O'))

        bus.write(msg)

        return

    #returns [src, data] or [] on error

    def readTool():

        ret = [0,0,0]

        if bus.read(1) == '^':

            ret[0] = bus.read(1)

            ret[1] = bus.read(1)

            ret[2] = bus.read(1)

            x = bus.read(1)

            print "got packet",ret

            if chr(ord(ret[0]) ^ ord(ret[1]) ^ ord(ret[2])) == x:

                return ret

            else:

                print "xor fail. got", x, "should have been got",(ord(ret[0]) ^ ord(ret[1]) ^ ord(ret[2]))

        return []

    def checkAck(t):

        tn = int(t) + 1 

        m = readTool()

        if m== []:

            return False

        return m[0] == chr(tn) and m[1] == chr(1) and m[2] == 'A'

    while True:

        id = get_last_id()

        if id != None:

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

            print "sending key request"

            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.extendy (tools[t])

                        except KeyError:

                            #this doesn't really matter

                            pass

            print "user has access to:", acl

            resp = keypad.read(1)

            print "request:",resp

            if acl.count(resp) > 0:

                keypad.write(TT_ACK)

                sendTool(resp)

                print "ACCESS GRANTED"

                if checkAck(resp):

                    print "ACK"

                else:

                    print "NO ACK!!!!"

            else:

                keypad.write(TT_NACK)

                print "ACCESS DENIED!!"

        else:

            break

        clear_id()