[PATCH v1 2/2] Allow custom MySQL port from the env file

[Ammar Faizi <ammarfaizi2@gnuweeb.org>]

From: Ammar Faizi <ammarfaizi2@gnuweeb.org>

I don't always use 3306 as the MySQL server port. Allow custom port
to be configured from the env file!

Signed-off-by: Ammar Faizi <ammarfaizi2@gnuweeb.org>
---
 daemon/dc.py                | 7 +++++++
 daemon/discord.env.example  | 1 +
 daemon/telegram.env.example | 1 +
 daemon/tg.py                | 7 +++++++
 4 files changed, 16 insertions(+)

diff --git a/daemon/dc.py b/daemon/dc.py
index f9b57e5..a9d5ebd 100644
--- a/daemon/dc.py
+++ b/daemon/dc.py
@@ -29,10 +29,17 @@ def main():
 	logger = BotLogger(Platform.DISCORD)
 	logger.init()
 
+	port = os.getenv("DB_PORT")
+	if not port:
+		port = 3306
+	else:
+		port = int(port)
+
 	client = GWClient(
 		db_conn=connector.connect(
 			host=os.getenv("DB_HOST"),
 			user=os.getenv("DB_USER"),
+			port=port,
 			password=os.getenv("DB_PASS"),
 			database=os.getenv("DB_NAME")
 		),
diff --git a/daemon/discord.env.example b/daemon/discord.env.example
index 60fcfac..de7e9f4 100644
--- a/daemon/discord.env.example
+++ b/daemon/discord.env.example
@@ -4,6 +4,7 @@ DISCORD_TOKEN=
 # Input your MySQL connection below
 DB_HOST=
 DB_USER=
+DB_PORT=3306
 DB_PASS=
 DB_NAME=
 
diff --git a/daemon/telegram.env.example b/daemon/telegram.env.example
index a070dd0..a42d86c 100644
--- a/daemon/telegram.env.example
+++ b/daemon/telegram.env.example
@@ -9,6 +9,7 @@ BOT_TOKEN=
 # MySQL Config
 DB_HOST=
 DB_USER=
+DB_PORT=3306
 DB_PASS=
 DB_NAME=
 
diff --git a/daemon/tg.py b/daemon/tg.py
index 8382c29..73ff2b9 100644
--- a/daemon/tg.py
+++ b/daemon/tg.py
@@ -21,6 +21,12 @@ def main():
 	logger = BotLogger()
 	logger.init()
 
+	port = os.getenv("DB_PORT")
+	if not port:
+		port = 3306
+	else:
+		port = int(port)
+
 	client = DaemonClient(
 		"telegram/storage/EmailScraper",
 		api_id=int(os.getenv("API_ID")),
@@ -30,6 +36,7 @@ def main():
 		conn=connector.connect(
 			host=os.getenv("DB_HOST"),
 			user=os.getenv("DB_USER"),
+			port=port,
 			password=os.getenv("DB_PASS"),
 			database=os.getenv("DB_NAME")
 		),
-- 
Ammar Faizi