X-Git-Url: https://git.ucc.asn.au/?p=uccvend-vendserver.git;a=blobdiff_plain;f=sql-edition%2Fservers%2FVendingMachine.py;fp=sql-edition%2Fservers%2FVendingMachine.py;h=0000000000000000000000000000000000000000;hp=e5a0251f0210b1f66f28e4b26b144dff2f68fc10;hb=6c130dc942392ccc8b7ffa0ec2ede5f9327f99e6;hpb=e423f11f190d685156cb557319b169d2b79b752f diff --git a/sql-edition/servers/VendingMachine.py b/sql-edition/servers/VendingMachine.py deleted file mode 100644 index e5a0251..0000000 --- a/sql-edition/servers/VendingMachine.py +++ /dev/null @@ -1,204 +0,0 @@ -# vim:ts=4 -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 - '200', '201', '202', '203', '204', '205', '206', - '207', '208', '209', '211', # key presses - ] -DOOR = 1 -SWITCH = 2 -KEY = 3 -TICK = 4 -MIFARE = 5 - -class VendingException(Exception): pass - -class VendingMachine: - def __init__(self, rfh, wfh, use_mifare): - self.events = [] - # Secret - self.secret = 'SN4CKZ0RZZZZZZZZ' - self.rfh = rfh - self.wfh = wfh - self.challenge = None - # Initialise ourselves into a known state - self.wfh.write('\n') - self.await_prompt() - self.wfh.write('echo off\n') - self.await_prompt() - self.wfh.write('PING\n') - code = '' - while code != '000': - 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: - 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 = '' - if (s == '#' or s == '%') and state == 1: state = 2 - if s == ' ' and state == 2: - if prefix == '': - self.challenge = None - return - if re.search('^[0-9a-fA-F]{4}$', prefix): - self.challenge = int(prefix, 16) - return - - def get_response(self, async = False): - self.wfh.flush() - while True: - s = '' - while s == '': - s = self.rfh.readline() - if s == '': - raise VendingException('Input socket has closed!') - s = s.strip('\r\n') - code = s[0:3] - text = s[4:] - if code in asynchronous_responses: - self.handle_event(code, text) - if async: return None - else: - self.await_prompt() - return (code, text) - - def get_switches(self): - self.wfh.write('S\n') - (code, text) = self.get_response() - if code != '600': - return (False, code, text) - self.interpret_switches(text) - return (True, code, text) - - def interpret_switches(self, text): - self.switches = (int(text[0:2], 16) << 8) | int(text[3:5], 16) - - def handle_event(self, code, text): - if code == '400': - self.events.append((DOOR, 1)) - elif code == '401': - self.events.append((DOOR, 0)) - elif code == '610': - # 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: - logging.warning('Unhandled event! (%s %s)\n'%(code,text)) - - def authed_message(self, message): - print 'self.challenge = %04x' % self.challenge - if self.challenge == None: - return message - crc = do_crc('%c%c'%(self.challenge >> 8, self.challenge & 0xff)) - crc = do_crc(self.secret, crc) - crc = do_crc(message, crc) - print 'output = "%s|%04x"' % (message, crc) - return message+'|'+('%04x'%crc) - - def ping(self): - self.wfh.write('PING\n') - (code, string) = self.get_response() - return (code == '000', code, string) - - 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'%item))+'\n') - (code, string) = self.get_response() - return (code == '100', code, string) - - def beep(self, duration = None, synchronous = True): - msg = 'B' - if synchronous: msg += 'S' - if duration != None: - if duration > 255: duration = 255 - if duration < 1: duration = 1 - msg += '%02x'%duration - self.wfh.write(msg+'\n') - (code, string) = self.get_response() - return (code == '500', code, string) - - def silence(self, duration = None, synchronous = True): - msg = 'C' - if synchronous: msg += 'S' - if duration != None: - if duration > 255: duration = 255 - if duration < 1: duration = 1 - msg += '%02x'%duration - self.wfh.write(msg+'\n') - (code, string) = self.get_response() - return (code == '501', code, string) - - 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) - - 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. - - # 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 - - 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 (TICK, time()) - ret = self.events[0] - del self.events[0] - return ret