aboutsummaryrefslogtreecommitdiff
path: root/gn3
diff options
context:
space:
mode:
Diffstat (limited to 'gn3')
-rw-r--r--gn3/db_utils.py10
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))