23 lines
467 B
Python
23 lines
467 B
Python
import logging
|
|
import pymysql
|
|
import os
|
|
from dotenv import load_dotenv
|
|
|
|
|
|
def get_db_connection() -> pymysql.Connection:
|
|
conn = None
|
|
try:
|
|
load_dotenv(".env")
|
|
|
|
conn = pymysql.connect(
|
|
user=os.environ.get('DB_USER'),
|
|
password=os.environ.get('DB_PASSWORD'),
|
|
host=os.environ.get('DB_HOST'),
|
|
port=int(os.environ.get('DB_PORT')),
|
|
database=os.environ.get('DB_NAME')
|
|
)
|
|
except Exception as e:
|
|
logging.error("SQL Connection Error:%s", e)
|
|
|
|
return conn
|