diff options
| -rw-r--r-- | gn3/db_utils.py | 10 |
1 files 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)) |