put in a generic exception handler

[uccvend-vendserver.git] / sql-edition / servers / VendServer.py

diff --git a/sql-edition/servers/VendServer.py b/sql-edition/servers/VendServer.py
index de763f2..4e0b16c 100755 (executable)
--- a/sql-edition/servers/VendServer.py
+++ b/sql-edition/servers/VendServer.py
@@ -3,6 +3,7 @@
 
 USE_DB = 0
 
+import ConfigParser
 import sys, os, string, re, pwd
 if USE_DB: import pg
 from time import time, sleep
@@ -197,11 +198,11 @@ class MessageKeeper:
        def done(self):
                return len(self.scrolling_message) == 0
 
-def run_forever(rfh, wfh):
+def run_forever(rfh, wfh, options, cf):
        v = VendingMachine(rfh, wfh)
        print 'PING is', v.ping()
 
-       if USE_DB: db = DispenseDatabase(v, DBServer, DBName, DBUser, DBPassword)
+       if USE_DB: db = DispenseDatabase(v, cf.DBServer, cf.DBName, cf.DBUser, cf.DBPassword)
        cur_user = ''
        cur_pin = ''
        cur_selection = ''
@@ -363,11 +364,10 @@ def run_forever(rfh, wfh):
                                        cur_selection = ''
                                        time_to_autologout = time() + 8
 
-# FIXME: a less ugly way of passing all these options would be nice
-def connect_to_vend(options, DBServer, DBName, DBUser, DBPassword, ServiceName, ServicePassword, ServerName, ConnectPassword, PrivPassword):
-       # Open vending machine via LAT
+def connect_to_vend(options, cf):
+       # Open vending machine via LAT?
        if options.use_lat:
-               latclient = LATClient(service = ServiceName, password = ServicePassword, server_name = ServerName, connect_password = ConnectPassword, priv_password = PrivPassword)
+               latclient = LATClient(service = cf.ServiceName, password = cf.ServicePassword, server_name = cf.ServerName, connect_password = cf.ConnectPassword, priv_password = cf.PrivPassword)
                rfh, wfh = latclient.get_fh()
        else:
                #(rfh, wfh) = popen2('../../virtualvend/vvend.py')
@@ -394,44 +394,75 @@ def parse_args():
 
        return options
 
-if __name__ == '__main__':
-       import ConfigParser
-
-       options = parse_args()
+config_options = {
+       'DBServer': ('Database', 'Server'),
+       'DBName': ('Database', 'Name'),
+       'DBUser': ('VendingMachine', 'DBUser'),
+       'DBPassword': ('VendingMachine', 'DBPassword'),
+       
+       'ServiceName': ('VendingMachine', 'ServiceName'),
+       'ServicePassword': ('VendingMachine', 'Password'),
+       
+       'ServerName': ('DecServer', 'Name'),
+       'ConnectPassword': ('DecServer', 'ConnectPassword'),
+       'PrivPassword': ('DecServer', 'PrivPassword'),
+       }
+
+class VendConfigFile:
+       def __init__(self, config_file, options):
+               try:
+                       cp = ConfigParser.ConfigParser()
+                       cp.read(config_file)
 
-       try:
-               cp = ConfigParser.ConfigParser()
-               cp.read(options.config_file)
-               DBServer = cp.get('Database', 'Server')
-               DBName = cp.get('Database', 'Name')
-               DBUser = cp.get('VendingMachine', 'DBUser')
-               DBPassword = cp.get('VendingMachine', 'DBPassword')
-               
-               ServiceName = cp.get('VendingMachine', 'ServiceName')
-               ServicePassword = cp.get('VendingMachine', 'Password')
+                       for option in options:
+                               section, name = options[option]
+                               value = cp.get(section, name)
+                               self.__dict__[option] = value
                
-               ServerName = cp.get('DecServer', 'Name')
-               ConnectPassword = cp.get('DecServer', 'ConnectPassword')
-               PrivPassword = cp.get('DecServer', 'PrivPassword')
-       except ConfigParser.Error, e:
-               print "Error reading config file: " + str(e)
-               sys.exit(1)
+               except ConfigParser.Error, e:
+                       print "Error reading config file "+config_file+": " + str(e)
+                       sys.exit(1)
+
+def do_vend_server():
+       options = parse_args()
+       config_opts = VendConfigFile(options.config_file, config_options)
 
        while True:
                try:
-                       rfh, wfh = connect_to_vend(options, DBServer, DBName, DBUser, DBPassword, ServiceName, ServicePassword, ServerName, ConnectPassword, PrivPassword)
+                       rfh, wfh = connect_to_vend(options, config_opts)
                except (LATClientException, socket.error), e:
                        (exc_type, exc_value, exc_traceback) = sys.exc_info()
+                       del exc_traceback
                        print
                        print "Connection error: "+str(exc_type)+" "+str(e)
-                       print_tb(exc_traceback)
-                       del exc_traceback
                        print "Trying again in 5 seconds."
                        sleep(5)
                        continue
                try:
-                       run_forever(rfh, wfh)
+                       run_forever(rfh, wfh, options, config_opts)
                except VendingException:
                        print
                        print "Connection died, trying again..."
 
+
+if __name__ == '__main__':
+       while True:
+               try:
+                       do_vend_server()
+               except KeyboardInterrupt:
+                       print "Killed by SIGINT."
+                       break
+               except:
+                       (exc_type, exc_value, exc_traceback) = sys.exc_info()
+                       print
+                       print "Uh-oh, unhandled " + str(exc_type) + " exception"
+                       print "Message: ", str(exc_value)
+                       print
+                       print_tb(exc_traceback)
+                       del exc_traceback
+                       print
+                       print "This message should be considered a bug in the Vend Server."
+                       print "Please report this to someone who can fix it."
+                       print
+                       print "Trying again anyway (might not help, but hey...)"
+