X-Git-Url: https://git.ucc.asn.au/?p=uccvend-vendserver.git;a=blobdiff_plain;f=sql-edition%2Fservers%2FLATClient.py;h=4d17643f225e4f33151802c926cce807ae2603c3;hp=ce77970f336f4a502522e884d6f401452352d5b9;hb=7c61f2e7d593f845697c84ba52f7dc58045c78c6;hpb=8bf3b178db0d2a8f3b8dcdfa02b6f8ca80cfad56

diff --git a/sql-edition/servers/LATClient.py b/sql-edition/servers/LATClient.py
index ce77970..4d17643 100644
--- a/sql-edition/servers/LATClient.py
+++ b/sql-edition/servers/LATClient.py
@@ -2,6 +2,7 @@ from socket import *
 from select import select
 from os import popen4
 from time import sleep
+import logging
 
 LATCP_SOCKET = '/var/run/latlogin'
 
@@ -26,14 +27,14 @@ def read_for_a_bit(rfh):
 			message = message + ch
 		else:
 			break
-	print "Received message: ", repr(message)
+	logging.debug("Received message: ", repr(message))
 	return message
 
 def write_and_get_response(rfh, wfh, message, expect_echo=True):
-	print "Writing message:", repr(message)
+	logging.debug("Writing message:", repr(message))
 	wfh.write(message+'\r\n')
 	wfh.flush()
-	print "  --> Sent"
+	logging.debug("  --> Sent")
 	response = read_for_a_bit(rfh)
 	if response.find(message) == -1 and expect_echo:
 		raise LATClientException("Talking to DEC server, expected to find original message in echo but didn't")
@@ -80,7 +81,7 @@ class LATClient:
 		if r:
 			l = self.rfh.readline()
 			if l.find('Service in use') >= 0:
-				print "Service in use, apparently: restarting DEC server"
+				logging.warning("Service in use, apparently: restarting DEC server")
 				self.reboot_server()
 
 	def __del__(self):
@@ -97,35 +98,47 @@ class LATClient:
 	def reboot_server(self):
 		self.sock.shutdown(2)
 		self.sock.close()
+		
+		logging.info('Logging into DEC server')
 		mopw, mopr = popen4('/usr/sbin/moprc '+self.server_name)
 		write_and_get_response(mopr, mopw, '')
+
+		logging.info('Sending password')
 		r = write_and_get_response(mopr, mopw, self.connect_password, False)
 		if r.find('Enter username> ') == -1:
-			print "Expected username prompt, got ", repr(r)
+			logging.warning("Expected username prompt, got " + repr(r))
 			raise LATClientException('failed to reboot server')
+
+		logging.info('Sending username')
 		r = write_and_get_response(mopr, mopw, 'grim reaper')
 		if r.find('Local> ') == -1:
-			print "Expected DEC server prompt, got ", repr(r)
+			logging.warning("Expected DEC server prompt, got " + repr(r))
 			raise LATClientException('failed to reboot server')
+
+		logging.info('Requesting privileges')
 		r = write_and_get_response(mopr, mopw, 'set priv')
 		if r.find('Password> ') == -1:
-			print "Expected priv password prompt, got ", repr(r)
+			logging.warning("Expected priv password prompt, got " + repr(r))
 			raise LATClientException('failed to reboot server')
+
+		logging.info('Sending password')
 		r = write_and_get_response(mopr, mopw, self.priv_password, False)
 		if r.find('Local> ') == -1:
-			print "Expected DEC server prompt, got ", repr(r)
+			logging.warning("Expected DEC server prompt, got " + repr(r))
 			raise LATClientException('failed to reboot server')
+
+		logging.info('Sending reboot request')
 		r = write_and_get_response(mopr, mopw, 'init del 0')
 		if r.find('Target does not respond') == -1:
-			print "Expected DEC server to die, got ", repr(r)
+			logging.warning("Expected DEC server to die, got " + repr(r))
 			raise LATClientException('failed to reboot server')
+
+		logging.info('Closed connection to server')
 		mopr.close()
 		mopw.close()
-		print
-		print "Waiting 10 seconds for DEC server to come back to life..."
+		logging.info("Waiting 10 seconds for DEC server to come back to life...")
 		sleep(10)
-		print "Rightyo, back to vending!"
-		print
+		logging.info("Rightyo, back to vending!")
 		raise LATClientException('needed to reboot server')
 
 	def read_reply(self):