git.ucc.asn.au
/
uccvend-vendserver.git
/ blobdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
|
commitdiff
|
tree
raw
|
inline
| side by side
add sample servers.conf to make testing on random machines easier
[uccvend-vendserver.git]
/
sql-edition
/
servers
/
VendingMachine.py
diff --git
a/sql-edition/servers/VendingMachine.py
b/sql-edition/servers/VendingMachine.py
index
e242df8
..
9fe5245
100644
(file)
--- a/
sql-edition/servers/VendingMachine.py
+++ b/
sql-edition/servers/VendingMachine.py
@@
-1,5
+1,5
@@
# vim:ts=4
# vim:ts=4
-import re
, pg
+import re
from CRC import do_crc
from select import select
from CRC import do_crc
from select import select
@@
-12,6
+12,8
@@
DOOR = 1
SWITCH = 2
KEY = 3
SWITCH = 2
KEY = 3
+class VendingException(Exception): pass
+
class VendingMachine:
def __init__(self, rfh, wfh):
self.events = []
class VendingMachine:
def __init__(self, rfh, wfh):
self.events = []
@@
-27,7
+29,7
@@
class VendingMachine:
self.wfh.write('PING\n')
code = ''
while code != '000':
self.wfh.write('PING\n')
code = ''
while code != '000':
- (code, _) = self.get_response()
+ code = self.get_response()[0]
self.get_switches()
def await_prompt(self):
self.get_switches()
def await_prompt(self):
@@
-36,8
+38,12
@@
class VendingMachine:
prefix = ''
s = ''
while True:
prefix = ''
s = ''
while True:
- s = self.rfh.read(1)
- if s == '': raise Exception('nothing read!')
+ 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 == '\n' or s == '\r':
state = 1
prefix = ''
if s == '\n' or s == '\r':
state = 1
prefix = ''
UCC
git Repository :: git.ucc.asn.au