From 79c0e0933b4f252c3d76323891ae943a7ca6fd58 Mon Sep 17 00:00:00 2001 From: Frederick Muriuki Muriithi Date: Mon, 22 Aug 2022 07:16:44 +0300 Subject: Use non-default MySQL port if present By default, MySQL/Mariadb connects at port 3306, however, the configuration variable `SQL_URI` could contain a different port. In that case, this commit allows the system to connect to the non-default port. --- gn3/db_utils.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/gn3/db_utils.py b/gn3/db_utils.py index 3b72d28..4865131 100644 --- a/gn3/db_utils.py +++ b/gn3/db_utils.py @@ -10,11 +10,13 @@ def parse_db_url() -> Tuple: is a default value for SQL_URI so a tuple result is\ always expected""" parsed_db = urlparse(SQL_URI) - return (parsed_db.hostname, parsed_db.username, - parsed_db.password, parsed_db.path[1:]) + return ( + parsed_db.hostname, parsed_db.username, parsed_db.password, + parsed_db.path[1:], parsed_db.port) def database_connector() -> mdb.Connection: """function to create db connector""" - host, user, passwd, db_name = parse_db_url() - return mdb.connect(host, user, passwd, db_name) + host, user, passwd, db_name, db_port = parse_db_url() + + return mdb.connect(host, user, passwd, db_name, port=(db_port or 3306)) -- cgit v1.2.3