Open SSH tunnel to MySQL if specified.

This commit is contained in:
Pablo Martin 2022-07-21 18:31:14 +02:00
parent 6a99fff921
commit 104d95bc5a

View file

@ -115,8 +115,15 @@ def get_connection_to_mysql(
:param connection_config: specifies host, port, etc. :param connection_config: specifies host, port, etc.
:return: the connection object :return: the connection object
""" """
if connection_config["ssh_tunneling"]["use_tunnel"]:
open_ssh_tunnel(connection_config)
mysql_connection_host = "127.0.0.1"
if not connection_config["ssh_tunneling"]["use_tunnel"]:
mysql_connection_host = connection_config["host"]
connection = mysql.connector.connect( connection = mysql.connector.connect(
host=connection_config["host"], host=mysql_connection_host,
port=connection_config["port"], port=connection_config["port"],
user=connection_config["user"], user=connection_config["user"],
password=connection_config["password"], password=connection_config["password"],