diff --git a/poller-wrapper.py b/poller-wrapper.py index d9b542011..d7b8f8beb 100755 --- a/poller-wrapper.py +++ b/poller-wrapper.py @@ -83,18 +83,30 @@ db_username = config['db_user'] db_password = config['db_pass'] if config['db_host'][:5].lower() == 'unix:': - db_server = config['db_host'] - db_port = 0 + db_server = config['db_host'] + db_port = 0 elif ':' in config['db_host']: - db_server = config['db_host'].rsplit(':')[0] - db_port = int(config['db_host'].rsplit(':')[1]) + db_server = config['db_host'].rsplit(':')[0] + db_port = int(config['db_host'].rsplit(':')[1]) else: - db_server = config['db_host'] - db_port =0 + db_server = config['db_host'] + db_port = 0 db_dbname = config['db_name'] +def db_open(): + try: + if db_port == 0: + db = MySQLdb.connect(host=db_server, user=db_username, passwd=db_password, db=db_dbname) + else: + db = MySQLdb.connect(host=db_server, port=db_port, user=db_username, passwd=db_password, db=db_dbname) + return db + except: + print "ERROR: Could not connect to MySQL database!" + sys.exit(2) + + # (c) 2015, GPLv3, Daniel Preussker <<