port sqlalchemy config

python/lib/db/connect.py

@@ -1,17 +1,14 @@
 from typing import Any
 from sqlalchemy import create_engine
 from sqlalchemy.orm import Session
+from lib.dataclass.config import DatabaseConfig
 
 
-default_port = 3306
-
-
-def connect_to_db(credentials: dict[str, Any]):
-    host     = credentials['host']
-    port     = credentials['port']
-    username = credentials['username']
-    password = credentials['passwd']
-    database = credentials['database']
-    port     = int(port) if port else default_port
+def connect_to_db(credentials: DatabaseConfig):
+    host     = credentials.host
+    port     = credentials.port
+    username = credentials.username
+    password = credentials.password
+    database = credentials.database
     engine = create_engine(f'mariadb+mysqlconnector://{username}:{password}@{host}:{port}/{database}')
     return Session(engine)