Updated snack config to use `dispense iteminfo` for prices (goodbye config file)

[uccvend-vendserver.git] / sql-edition / servers / VendingMachine.py

diff --git a/sql-edition/servers/VendingMachine.py b/sql-edition/servers/VendingMachine.py
index e242df8..e96832a 100644 (file)
--- a/sql-edition/servers/VendingMachine.py
+++ b/sql-edition/servers/VendingMachine.py
@@ -1,7 +1,10 @@
 # vim:ts=4
-import re, pg
+import re
 from CRC import do_crc
 from select import select
+import socket, logging
+from time import time, sleep
+from MIFAREClient import MIFAREClient
 
 asynchronous_responses = [     '400', '401', # door open/closed
                                '610',        # switches changed
@@ -11,11 +14,16 @@ asynchronous_responses = [  '400', '401', # door open/closed
 DOOR = 1
 SWITCH = 2
 KEY = 3
+TICK = 4
+MIFARE = 5
+
+class VendingException(Exception): pass
 
 class VendingMachine:
-       def __init__(self, rfh, wfh):
+       def __init__(self, rfh, wfh, use_mifare):
                self.events = []
-               self.secret = 'AAAAAAAAAAAAAAAA'
+               # Secret
+               self.secret = 'SN4CKZ0RZZZZZZZZ'
                self.rfh = rfh
                self.wfh = wfh
                self.challenge = None
@@ -27,17 +35,31 @@ class VendingMachine:
                self.wfh.write('PING\n')
                code = ''
                while code != '000':
-                       (code, _) = self.get_response()
+                       code = self.get_response()[0]
                self.get_switches()
+               if use_mifare:
+                       self.mifare = MIFAREClient()
+                       self.mifare_timeout = 0
+               else:
+                       self.mifare = None
 
        def await_prompt(self):
                self.wfh.flush()
                state = 1
+               timeout = 0.5
                prefix = ''
                s = ''
+               # mtearle - vending machine was dying wait for a response from
+               # the hardware, suspect it was missing characters
+               #
+               # fixed by migration to pyserial - but future good place to start
                while True:
-                       s = self.rfh.read(1)
-                       if s == '': raise Exception('nothing read!')
+                       try:
+                               s = self.rfh.read(1)
+                       except socket.error:
+                               raise VendingException('failed to read input from vending machine')
+                       if s == '': raise VendingException('nothing read!')
+                       if (s != '#' and s != '%') and state == 1: prefix += s
                        if s == '\n' or s == '\r':
                                state = 1
                                prefix = ''
@@ -56,7 +78,8 @@ class VendingMachine:
                        s = ''
                        while s == '':
                                s = self.rfh.readline()
-                               if s == '': return None
+                               if s == '':
+                                       raise VendingException('Input socket has closed!')
                                s = s.strip('\r\n')
                        code = s[0:3]
                        text = s[4:]
@@ -80,16 +103,17 @@ class VendingMachine:
 
        def handle_event(self, code, text):
                if code == '400':
-                       self.events.append((DOOR, 0))
-               elif code == '401':
                        self.events.append((DOOR, 1))
+               elif code == '401':
+                       self.events.append((DOOR, 0))
                elif code == '610':
-                       self.events.append((SWITCH, None))
+                       # NOP this. Nothing handles this yet.
+                       #self.events.append((SWITCH, None))
                        self.interpret_switches(text)
                elif code[0] == '2':
                        self.events.append((KEY, int(code[1:3])))
                else:
-                       sys.stderr.write('WARNING: Unhandled event! (%s %s)\n'%(code,text))
+                       logging.warning('Unhandled event! (%s %s)\n'%(code,text))
 
        def authed_message(self, message):
                if self.challenge == None:
@@ -107,7 +131,7 @@ class VendingMachine:
        def vend(self, item):
                if not re.search('^[0-9][0-9]$', item):
                        return (False, 'Invalid item requested (%s)'%item)
-               self.wfh.write(self.authed_message(('V%s\n'%item)+'\n'))
+               self.wfh.write(self.authed_message(('V%s'%item))+'\n')
                (code, string) = self.get_response()
                return (code == '100', code, string)
 
@@ -136,6 +160,7 @@ class VendingMachine:
        def display(self, string):
                if len(string) > 10:
                        string = string[0:10]
+               string = re.sub('(.)\.', lambda match: '.'+match.group(1), string)
                self.wfh.write('D'+string+'\n')
                (code, string) = self.get_response()
                return (code == '300', code, string)
@@ -143,15 +168,35 @@ class VendingMachine:
        def next_event(self, timeout = None):
                # we don't want to buffer in the serial port, so we get all the events
                # we can ASAP.
-               if len(self.events) > 0: timeout = 0
-               while True:
-                       (r, _, _) = select([self.rfh], [], [], timeout)
+
+               # Never have no timeout...
+               if timeout == None: timeout = 60*60*24*365
+
+               # Make sure we go through the loop at least once.
+               if timeout < 0: timeout = 0
+
+               while timeout >= 0:
+                       this_timeout = min(timeout, 0.2)
+                       timeout -= this_timeout
+
+                       (r, _, _) = select([self.rfh], [], [], this_timeout)
                        if r:
                                self.get_response(async = True)
                                timeout = 0
-                       else:
+
+                       if self.mifare:
+                               now = time()
+                               if now > self.mifare_timeout:
+                                       self.mifare_timeout = now + 0.5
+                                       mifare_uid = self.mifare.get_card_id()
+                                       if mifare_uid != None:
+                                               logging.info('Got MIFARE card id %s'%(repr(mifare_uid)))
+                                               self.events.append((MIFARE, mifare_uid))
+                                               timeout = 0
+                       if timeout == 0:
                                break
-               if len(self.events) == 0: return None
+
+               if len(self.events) == 0: return (TICK, time())
                ret = self.events[0]
                del self.events[0]
                return ret