X-Git-Url: https://git.ucc.asn.au/?a=blobdiff_plain;ds=sidebyside;f=sql-edition%2Fservers%2FVendingMachine.py;h=9506daf0599b8c2d0b44697c32e2c6466ea98f87;hb=66518c6317986e5f0d41559c75e4038faee5f411;hp=9fe524519e7666579c09245f6110fbcc0050fe24;hpb=a4f325c95da6ac6cf3f739172fdc0ce2d84b6f50;p=uccvend-vendserver.git

diff --git a/sql-edition/servers/VendingMachine.py b/sql-edition/servers/VendingMachine.py
index 9fe5245..9506daf 100644
--- a/sql-edition/servers/VendingMachine.py
+++ b/sql-edition/servers/VendingMachine.py
@@ -2,6 +2,7 @@
 import re
 from CRC import do_crc
 from select import select
+import socket
 
 asynchronous_responses = [	'400', '401', # door open/closed
 				'610',        # switches changed
@@ -62,7 +63,8 @@ class VendingMachine:
 			s = ''
 			while s == '':
 				s = self.rfh.readline()
-				if s == '': return None
+				if s == '':
+					raise VendingException('Input socket has closed!')
 				s = s.strip('\r\n')
 			code = s[0:3]
 			text = s[4:]