/[anoncvs]/projects/dispense2/sql-edition/servers/VendingMachine.py


UCC Code Repository

Contents of /projects/dispense2/sql-edition/servers/VendingMachine.py

Parent Directory Parent Directory | Revision Log Revision Log


Revision 1.23 - (show annotations) (download) (as text)
Sat Apr 9 03:29:12 2011 UTC (10 years, 2 months ago) by zanchey
Branch: MAIN
Changes since 1.22: +1 -1 lines
File MIME type: text/x-python
MIFARE: better logging, thanks to [TRS] I think

1 # vim:ts=4
2 import re
3 from CRC import do_crc
4 from select import select
5 import socket, logging
6 from time import time, sleep
7 from MIFAREClient import MIFAREClient
8
9 asynchronous_responses = [ '400', '401', # door open/closed
10 '610', # switches changed
11 '200', '201', '202', '203', '204', '205', '206',
12 '207', '208', '209', '211', # key presses
13 ]
14 DOOR = 1
15 SWITCH = 2
16 KEY = 3
17 TICK = 4
18 MIFARE = 5
19
20 class VendingException(Exception): pass
21
22 class VendingMachine:
23 def __init__(self, rfh, wfh, use_mifare):
24 self.events = []
25 # Secret
26 self.secret = 'SN4CKZ0RZZZZZZZZ'
27 self.rfh = rfh
28 self.wfh = wfh
29 self.challenge = None
30 # Initialise ourselves into a known state
31 self.wfh.write('\n')
32 self.await_prompt()
33 self.wfh.write('echo off\n')
34 self.await_prompt()
35 self.wfh.write('PING\n')
36 code = ''
37 while code != '000':
38 code = self.get_response()[0]
39 self.get_switches()
40 if use_mifare:
41 self.mifare = MIFAREClient()
42 self.mifare_timeout = 0
43 else:
44 self.mifare = None
45
46 def await_prompt(self):
47 self.wfh.flush()
48 state = 1
49 timeout = 0.5
50 prefix = ''
51 s = ''
52 # mtearle - vending machine was dying wait for a response from
53 # the hardware, suspect it was missing characters
54 #
55 # fixed by migration to pyserial - but future good place to start
56 while True:
57 try:
58 s = self.rfh.read(1)
59 except socket.error:
60 raise VendingException('failed to read input from vending machine')
61 if s == '': raise VendingException('nothing read!')
62 if (s != '#' and s != '%') and state == 1: prefix += s
63 if s == '\n' or s == '\r':
64 state = 1
65 prefix = ''
66 if (s == '#' or s == '%') and state == 1: state = 2
67 if s == ' ' and state == 2:
68 if prefix == '':
69 self.challenge = None
70 return
71 if re.search('^[0-9a-fA-F]{4}$', prefix):
72 self.challenge = int(prefix, 16)
73 return
74
75 def get_response(self, async = False):
76 self.wfh.flush()
77 while True:
78 s = ''
79 while s == '':
80 s = self.rfh.readline()
81 if s == '':
82 raise VendingException('Input socket has closed!')
83 s = s.strip('\r\n')
84 code = s[0:3]
85 text = s[4:]
86 if code in asynchronous_responses:
87 self.handle_event(code, text)
88 if async: return None
89 else:
90 self.await_prompt()
91 return (code, text)
92
93 def get_switches(self):
94 self.wfh.write('S\n')
95 (code, text) = self.get_response()
96 if code != '600':
97 return (False, code, text)
98 self.interpret_switches(text)
99 return (True, code, text)
100
101 def interpret_switches(self, text):
102 self.switches = (int(text[0:2], 16) << 8) | int(text[3:5], 16)
103
104 def handle_event(self, code, text):
105 if code == '400':
106 self.events.append((DOOR, 1))
107 elif code == '401':
108 self.events.append((DOOR, 0))
109 elif code == '610':
110 # NOP this. Nothing handles this yet.
111 #self.events.append((SWITCH, None))
112 self.interpret_switches(text)
113 elif code[0] == '2':
114 self.events.append((KEY, int(code[1:3])))
115 else:
116 logging.warning('Unhandled event! (%s %s)\n'%(code,text))
117
118 def authed_message(self, message):
119 if self.challenge == None:
120 return message
121 crc = do_crc('%c%c'%(self.challenge >> 8, self.challenge & 0xff))
122 crc = do_crc(self.secret, crc)
123 crc = do_crc(message, crc)
124 return message+'|'+('%04x'%crc)
125
126 def ping(self):
127 self.wfh.write('PING\n')
128 (code, string) = self.get_response()
129 return (code == '000', code, string)
130
131 def vend(self, item):
132 if not re.search('^[0-9][0-9]$', item):
133 return (False, 'Invalid item requested (%s)'%item)
134 self.wfh.write(self.authed_message(('V%s'%item))+'\n')
135 (code, string) = self.get_response()
136 return (code == '100', code, string)
137
138 def beep(self, duration = None, synchronous = True):
139 msg = 'B'
140 if synchronous: msg += 'S'
141 if duration != None:
142 if duration > 255: duration = 255
143 if duration < 1: duration = 1
144 msg += '%02x'%duration
145 self.wfh.write(msg+'\n')
146 (code, string) = self.get_response()
147 return (code == '500', code, string)
148
149 def silence(self, duration = None, synchronous = True):
150 msg = 'C'
151 if synchronous: msg += 'S'
152 if duration != None:
153 if duration > 255: duration = 255
154 if duration < 1: duration = 1
155 msg += '%02x'%duration
156 self.wfh.write(msg+'\n')
157 (code, string) = self.get_response()
158 return (code == '501', code, string)
159
160 def display(self, string):
161 if len(string) > 10:
162 string = string[0:10]
163 string = re.sub('(.)\.', lambda match: '.'+match.group(1), string)
164 self.wfh.write('D'+string+'\n')
165 (code, string) = self.get_response()
166 return (code == '300', code, string)
167
168 def next_event(self, timeout = None):
169 # we don't want to buffer in the serial port, so we get all the events
170 # we can ASAP.
171
172 # Never have no timeout...
173 if timeout == None: timeout = 60*60*24*365
174
175 # Make sure we go through the loop at least once.
176 if timeout < 0: timeout = 0
177
178 while timeout >= 0:
179 this_timeout = min(timeout, 0.2)
180 timeout -= this_timeout
181
182 (r, _, _) = select([self.rfh], [], [], this_timeout)
183 if r:
184 self.get_response(async = True)
185 timeout = 0
186
187 if self.mifare:
188 now = time()
189 if now > self.mifare_timeout:
190 self.mifare_timeout = now + 0.5
191 mifare_uid = self.mifare.get_card_id()
192 if mifare_uid != None:
193 logging.info('Got MIFARE card id %s'%(repr(mifare_uid)))
194 self.events.append((MIFARE, mifare_uid))
195 timeout = 0
196 if timeout == 0:
197 break
198
199 if len(self.events) == 0: return (TICK, time())
200 ret = self.events[0]
201 del self.events[0]
202 return ret

Managed by UCC Webmasters ViewVC Help
Powered by ViewVC 1.1.26