X-Git-Url: https://git.ucc.asn.au/?p=uccvend-vendserver.git;a=blobdiff_plain;f=sql-edition%2Fservers%2FVendingMachine.py;h=e96832a9f52b22c57c86c24a97d8afc8fcfd849f;hp=9506daf0599b8c2d0b44697c32e2c6466ea98f87;hb=08a392617c464e6336d47ba37168c3a3a2529123;hpb=37123733790f1525161256cfabd1da877bf37e74

diff --git a/sql-edition/servers/VendingMachine.py b/sql-edition/servers/VendingMachine.py
index 9506daf..e96832a 100644
--- a/sql-edition/servers/VendingMachine.py
+++ b/sql-edition/servers/VendingMachine.py
@@ -2,7 +2,9 @@
 import re
 from CRC import do_crc
 from select import select
-import socket
+import socket, logging
+from time import time, sleep
+from MIFAREClient import MIFAREClient
 
 asynchronous_responses = [	'400', '401', # door open/closed
 				'610',        # switches changed
@@ -12,13 +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
@@ -32,19 +37,29 @@ class VendingMachine:
 		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:
-				print "Blah, seems DEC server has fallen over"
 				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 = ''
@@ -88,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:
@@ -115,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)
 
@@ -144,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)
@@ -151,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