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

#

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

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

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 for messages are these + 1 but in decimal, not ascii

tools = {

   'Bandsaw':'2',

   'DrillPress':'1',

   'Mill':'4',

   'Lathe':'5',

   'ChopMiterSaw':'3'

}

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"

    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:

        s = raw_input()

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

        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.append (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!!"

        fflush(stdin)