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


UCC Code Repository

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

Parent Directory Parent Directory | Revision Log Revision Log | View Patch Patch

revision 1.6 by dagobah, Sun Jun 27 11:11:00 2004 UTC revision 1.19 by dagobah, Mon Mar 5 07:06:48 2007 UTC
# Line 1  Line 1 
1  # vim:ts=4  # vim:ts=4
2  import re, pg  import re
3  from CRC import do_crc  from CRC import do_crc
4  from select import select  from select import select
5    import socket, logging
6    from time import time, sleep
7    
8  asynchronous_responses = [      '400', '401', # door open/closed  asynchronous_responses = [      '400', '401', # door open/closed
9                                  '610',        # switches changed                                  '610',        # switches changed
# Line 11  asynchronous_responses = [     '400', '401', Line 13  asynchronous_responses = [     '400', '401',
13  DOOR = 1  DOOR = 1
14  SWITCH = 2  SWITCH = 2
15  KEY = 3  KEY = 3
16    TICK = 4
17    
18    class VendingException(Exception): pass
19    
20  class VendingMachine:  class VendingMachine:
21          def __init__(self, rfh, wfh):          def __init__(self, rfh, wfh):
22                  self.events = []                  self.events = []
23                  self.secret = 'AAAAAAAAAAAAAAAA'                  # Secret
24                    self.secret = 'SN4CKZ0RZZZZZZZZ'
25                  self.rfh = rfh                  self.rfh = rfh
26                  self.wfh = wfh                  self.wfh = wfh
27                  self.challenge = None                  self.challenge = None
# Line 27  class VendingMachine: Line 33  class VendingMachine:
33                  self.wfh.write('PING\n')                  self.wfh.write('PING\n')
34                  code = ''                  code = ''
35                  while code != '000':                  while code != '000':
36                          (code, _) = self.get_response()                          code = self.get_response()[0]
37                  self.get_switches()                  self.get_switches()
38    
39          def await_prompt(self):          def await_prompt(self):
# Line 36  class VendingMachine: Line 42  class VendingMachine:
42                  prefix = ''                  prefix = ''
43                  s = ''                  s = ''
44                  while True:                  while True:
45                          s = self.rfh.read(1)                          try:
46                          if s == '': raise Exception('nothing read!')                                  s = self.rfh.read(1)
47                            except socket.error:
48                                    raise VendingException('failed to read input from vending machine')
49                            if s == '': raise VendingException('nothing read!')
50                            if (s != '#' and s != '%') and state == 1: prefix += s
51                          if s == '\n' or s == '\r':                          if s == '\n' or s == '\r':
52                                  state = 1                                  state = 1
53                                  prefix = ''                                  prefix = ''
# Line 56  class VendingMachine: Line 66  class VendingMachine:
66                          s = ''                          s = ''
67                          while s == '':                          while s == '':
68                                  s = self.rfh.readline()                                  s = self.rfh.readline()
69                                  if s == '': return None                                  if s == '':
70                                            raise VendingException('Input socket has closed!')
71                                  s = s.strip('\r\n')                                  s = s.strip('\r\n')
72                          code = s[0:3]                          code = s[0:3]
73                          text = s[4:]                          text = s[4:]
# Line 80  class VendingMachine: Line 91  class VendingMachine:
91    
92          def handle_event(self, code, text):          def handle_event(self, code, text):
93                  if code == '400':                  if code == '400':
                         self.events.append((DOOR, 0))  
                 elif code == '401':  
94                          self.events.append((DOOR, 1))                          self.events.append((DOOR, 1))
95                    elif code == '401':
96                            self.events.append((DOOR, 0))
97                  elif code == '610':                  elif code == '610':
98                          self.events.append((SWITCH, None))                          # NOP this. Nothing handles this yet.
99                            #self.events.append((SWITCH, None))
100                          self.interpret_switches(text)                          self.interpret_switches(text)
101                  elif code[0] == '2':                  elif code[0] == '2':
102                          self.events.append((KEY, int(code[1:3])))                          self.events.append((KEY, int(code[1:3])))
103                  else:                  else:
104                          sys.stderr.write('WARNING: Unhandled event! (%s %s)\n'%(code,text))                          logging.warning('Unhandled event! (%s %s)\n'%(code,text))
105    
106          def authed_message(self, message):          def authed_message(self, message):
107                  if self.challenge == None:                  if self.challenge == None:
# Line 107  class VendingMachine: Line 119  class VendingMachine:
119          def vend(self, item):          def vend(self, item):
120                  if not re.search('^[0-9][0-9]$', item):                  if not re.search('^[0-9][0-9]$', item):
121                          return (False, 'Invalid item requested (%s)'%item)                          return (False, 'Invalid item requested (%s)'%item)
122                  self.wfh.write(self.authed_message(('V%s\n'%item)+'\n'))                  self.wfh.write(self.authed_message(('V%s'%item))+'\n')
123                  (code, string) = self.get_response()                  (code, string) = self.get_response()
124                  return (code == '100', code, string)                  return (code == '100', code, string)
125    
# Line 136  class VendingMachine: Line 148  class VendingMachine:
148          def display(self, string):          def display(self, string):
149                  if len(string) > 10:                  if len(string) > 10:
150                          string = string[0:10]                          string = string[0:10]
151                    string = re.sub('(.)\.', lambda match: '.'+match.group(1), string)
152                  self.wfh.write('D'+string+'\n')                  self.wfh.write('D'+string+'\n')
153                  (code, string) = self.get_response()                  (code, string) = self.get_response()
154                  return (code == '300', code, string)                  return (code == '300', code, string)
# Line 143  class VendingMachine: Line 156  class VendingMachine:
156          def next_event(self, timeout = None):          def next_event(self, timeout = None):
157                  # we don't want to buffer in the serial port, so we get all the events                  # we don't want to buffer in the serial port, so we get all the events
158                  # we can ASAP.                  # we can ASAP.
159                    if timeout < 0: timeout = 0
160                  if len(self.events) > 0: timeout = 0                  if len(self.events) > 0: timeout = 0
161                  while True:                  while True:
162                          (r, _, _) = select([self.rfh], [], [], timeout)                          (r, _, _) = select([self.rfh], [], [], timeout)
# Line 151  class VendingMachine: Line 165  class VendingMachine:
165                                  timeout = 0                                  timeout = 0
166                          else:                          else:
167                                  break                                  break
168                  if len(self.events) == 0: return None                  if len(self.events) == 0: return (TICK, time())
169                  ret = self.events[0]                  ret = self.events[0]
170                  del self.events[0]                  del self.events[0]
171                  return ret                  return ret

Legend:
Removed from v.1.6  
changed lines
  Added in v.1.19

Managed by UCC Webmasters ViewVC Help
Powered by ViewVC 1.1.26