From 5d3ce472266fd51d50fef6a38ab614c7a9b8a94d Mon Sep 17 00:00:00 2001 From: Felix Hamme Date: Tue, 16 May 2023 19:29:07 +0200 Subject: [PATCH] connection arguments: fix KeyError --- plugins/module_utils/mysql.py | 12 +++++++----- 1 file changed, 7 insertions(+), 5 deletions(-) diff --git a/plugins/module_utils/mysql.py b/plugins/module_utils/mysql.py index cc60f68..371918b 100644 --- a/plugins/module_utils/mysql.py +++ b/plugins/module_utils/mysql.py @@ -145,15 +145,17 @@ def mysql_connect(module, login_user=None, login_password=None, config_file='', # In case of PyMySQL driver: if mysql_driver.version_info[0] < 1: # for PyMySQL < 1.0.0, use 'db' instead of 'database' - config['db'] = config.get('database') - del config['database'] + if 'database' in config: + config['db'] = config['database'] + del config['database'] db_connection = mysql_driver.connect(autocommit=autocommit, **config) else: # In case of MySQLdb driver - if mysql_driver.version_info[0] < 2 or mysql_driver.version_info[1] < 1: + if mysql_driver.version_info[0] < 2 and mysql_driver.version_info[1] < 1: # for MySQLdb < 2.1.0, use 'db' instead of 'database' - config['db'] = config.get('database') - del config['database'] + if 'database' in config: + config['db'] = config['database'] + del config['database'] db_connection = mysql_driver.connect(**config) if autocommit: db_connection.autocommit(True)