/[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.2 by dagobah, Fri Jun 25 16:40:56 2004 UTC revision 1.18 by dagobah, Fri Mar 2 12:33:13 2007 UTC
# Line 1  Line 1 
1  import re, pg  # vim:ts=4
2    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 10  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):
# Line 26  class VendingMachine: Line 32  class VendingMachine:
32                  self.wfh.write('PING\n')                  self.wfh.write('PING\n')
33                  code = ''                  code = ''
34                  while code != '000':                  while code != '000':
35                          (code, _) = self.get_response()                          code = self.get_response()[0]
36                  self.get_switches()                  self.get_switches()
37    
38          def await_prompt(self):          def await_prompt(self):
# Line 35  class VendingMachine: Line 41  class VendingMachine:
41                  prefix = ''                  prefix = ''
42                  s = ''                  s = ''
43                  while True:                  while True:
44                          s = self.rfh.read(1)                          try:
45                          if s == '': raise Exception                                  s = self.rfh.read(1)
46                            except socket.error:
47                                    raise VendingException('failed to read input from vending machine')
48                            if s == '': raise VendingException('nothing read!')
49                            if (s != '#' and s != '%') and state == 1: prefix += s
50                          if s == '\n' or s == '\r':                          if s == '\n' or s == '\r':
51                                  state = 1                                  state = 1
52                                  prefix = ''                                  prefix = ''
# Line 49  class VendingMachine: Line 59  class VendingMachine:
59                                          self.challenge = int(prefix, 16)                                          self.challenge = int(prefix, 16)
60                                          return                                          return
61    
62          def get_response(self):          def get_response(self, async = False):
63                  self.wfh.flush()                  self.wfh.flush()
64                  while True:                  while True:
65                          s = ''                          s = ''
66                          while s == '':                          while s == '':
67                                  s = self.rfh.readline()                                  s = self.rfh.readline()
68                                  if s == '': return None                                  if s == '':
69                                            raise VendingException('Input socket has closed!')
70                                  s = s.strip('\r\n')                                  s = s.strip('\r\n')
71                          code = s[0:3]                          code = s[0:3]
72                          text = s[4:]                          text = s[4:]
73                          if code in asynchronous_responses:                          if code in asynchronous_responses:
74                                  self.handle_event(code, text)                                  self.handle_event(code, text)
75                                    if async: return None
76                          else:                          else:
77                                  self.await_prompt()                                  self.await_prompt()
78                                  return (code, text)                                  return (code, text)
# Line 78  class VendingMachine: Line 90  class VendingMachine:
90    
91          def handle_event(self, code, text):          def handle_event(self, code, text):
92                  if code == '400':                  if code == '400':
                         self.events.append((DOOR, 0))  
                 elif code == '401':  
93                          self.events.append((DOOR, 1))                          self.events.append((DOOR, 1))
94                    elif code == '401':
95                            self.events.append((DOOR, 0))
96                  elif code == '610':                  elif code == '610':
97                          self.events_append((SWITCH, None))                          # NOP this. Nothing handles this yet.
98                            #self.events.append((SWITCH, None))
99                          self.interpret_switches(text)                          self.interpret_switches(text)
100                  elif code[0] == '2':                  elif code[0] == '2':
101                          self.events.append((KEY, int(code[1:3])))                          self.events.append((KEY, int(code[1:3])))
102                  else:                  else:
103                          sys.stderr.write('WARNING: Unhandled event! (%s %s)\n'%(code,text))                          logging.warning('Unhandled event! (%s %s)\n'%(code,text))
104    
105          def authed_message(self, message):          def authed_message(self, message):
106                  if self.challenge == None:                  if self.challenge == None:
# Line 105  class VendingMachine: Line 118  class VendingMachine:
118          def vend(self, item):          def vend(self, item):
119                  if not re.search('^[0-9][0-9]$', item):                  if not re.search('^[0-9][0-9]$', item):
120                          return (False, 'Invalid item requested (%s)'%item)                          return (False, 'Invalid item requested (%s)'%item)
121                  self.wfh.write(self.authed_message(('V%s\n'%item)+'\n'))                  self.wfh.write(self.authed_message(('V%s'%item))+'\n')
122                  (code, string) = self.get_response()                  (code, string) = self.get_response()
123                  return (code == '100', code, string)                  return (code == '100', code, string)
124    
# Line 129  class VendingMachine: Line 142  class VendingMachine:
142                          msg += '%02x'%duration                          msg += '%02x'%duration
143                  self.wfh.write(msg+'\n')                  self.wfh.write(msg+'\n')
144                  (code, string) = self.get_response()                  (code, string) = self.get_response()
145                  # FIXME: workaround a bug in rom W. should be just: return (code == '500', code, string)                  return (code == '501', code, string)
                 return (code == '500' or code == '501', code, string)  
146    
147          def display(self, string):          def display(self, string):
148                  if len(string) > 10:                  if len(string) > 10:
149                          string = string[0:10]                          string = string[0:10]
150                    string = re.sub('(.)\.', lambda match: '.'+match.group(1), string)
151                  self.wfh.write('D'+string+'\n')                  self.wfh.write('D'+string+'\n')
152                  (code, string) = self.get_response()                  (code, string) = self.get_response()
153                  return (code == '300', code, string)                  return (code == '300', code, string)
154    
155          def next_event(self):          def next_event(self, timeout = None):
156                  if len(self.events) > 0:                  # we don't want to buffer in the serial port, so we get all the events
157                          ret = self.events[0]                  # we can ASAP.
158                          del self.events[0]                  if timeout < 0: timeout = 0
159                          return ret                  if len(self.events) > 0: timeout = 0
                 return None  
   
         def wait_for_events(self, timeout = None):  
                 (r, _, _) = select([self.rfh], [], [], timeout)  
                 if not r: return False  
                 event_added = False  
160                  while True:                  while True:
161                      r.get_response()                          (r, _, _) = select([self.rfh], [], [], timeout)
162                      (r, _, _) = select([self.rfh], [], [], 0)                          if r:
163                      if not r: return event_added                                  self.get_response(async = True)
164                                    timeout = 0
165                            else:
166                                    break
167                    if len(self.events) == 0: return (TICK, time())
168                    ret = self.events[0]
169                    del self.events[0]
170                    return ret

Legend:
Removed from v.1.2  
changed lines
  Added in v.1.18

Managed by UCC Webmasters ViewVC Help
Powered by ViewVC 1.1.26